isso/isso/__init__.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
#
# Copyright 2012, Martin Zimmermann <info@posativ.org>.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# 1. Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
#
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# The views and conclusions contained in the software and documentation are
# those of the authors and should not be interpreted as representing official
# policies, either expressed or implied, of posativ <info@posativ.org>.
#
# Isso – a lightweight Disqus alternative

__version__ = '0.2'

import sys; reload(sys)
sys.setdefaultencoding('utf-8')  # we only support UTF-8 and python 2.X :-)

import io
import json

from os.path import join, dirname
from optparse import OptionParser, make_option, SUPPRESS_HELP

from itsdangerous import URLSafeTimedSerializer

from werkzeug.wsgi import SharedDataMiddleware
from werkzeug.routing import Map, Rule
from werkzeug.serving import run_simple
from werkzeug.wrappers import Request, Response
from werkzeug.exceptions import HTTPException, NotFound, InternalServerError

from isso import admin, comment, db, migrate
from isso.utils import determine, import_object, RegexConverter, IssoEncoder

# override default json :func:`dumps`.
_dumps = json.dumps
setattr(json, 'dumps', lambda obj, **kw: _dumps(obj, cls=IssoEncoder, **kw))

# yep. lazy.
url = lambda path, endpoint, methods: Rule(path, endpoint=endpoint, methods=methods)

url_map = Map([
    # moderation panel
    url('/', 'admin.login', ['GET', 'POST']),
    url('/admin/', 'admin.index', ['GET', 'POST']),

    # comment API, note that the client side quotes the URL, but this is
    # actually unnecessary. PEP 333 aka WSGI always unquotes PATH_INFO.
    url('/1.0/<re(".+"):path>/', 'comment.get', ['GET']),
    url('/1.0/<re(".+"):path>/new', 'comment.create', ['POST']),
    url('/1.0/<re(".+"):path>/<int:id>', 'comment.get', ['GET']),
    url('/1.0/<re(".+"):path>/<int:id>', 'comment.modify', ['PUT', 'DELETE']),
    url('/1.0/<re(".+"):path>/<int:id>/approve', 'comment.approve', ['PUT'])
], converters={'re': RegexConverter})


class Isso:

    PRODUCTION = True
    SECRET = 'secret'
    SECRET_KEY = ',\x1e\xbaY\xbb\xdf\xe7@\x85\xe3\xd9\xb4A9\xe4G\xa6O'
    MODERATION = False
    SQLITE = None

    HOST = 'http://localhost:8000/'
    MAX_AGE = 15 * 60

    def __init__(self, conf):

        self.__dict__.update(dict((k, v) for k, v in conf.iteritems() if k.isupper()))
        self.signer = URLSafeTimedSerializer(self.SECRET_KEY)
        self.HOST = determine(self.HOST)

        if self.SQLITE:
            self.db = db.SQLite(self)

        self.markup = import_object(conf.get('MARKUP', 'isso.markup.Markdown'))(conf)

    def sign(self, obj):
        return self.signer.dumps(obj)

    def unsign(self, obj):
        return self.signer.loads(obj, max_age=self.MAX_AGE)

    def dispatch(self, request, start_response):
        adapter = url_map.bind_to_environ(request.environ)
        try:
            endpoint, values = adapter.match()
            module, function = endpoint.split('.', 1)
            handler = getattr(globals()[module], function)
            return handler(self, request.environ, request, **values)
        except NotFound, e:
            return Response('Not Found', 404)
        except HTTPException, e:
            return e
        except InternalServerError, e:
            return Response(e, 500)

    def wsgi_app(self, environ, start_response):
        request = Request(environ)
        response = self.dispatch(request, start_response)
        return response(environ, start_response)

    def __call__(self, environ, start_response):
        return self.wsgi_app(environ, start_response)


def main():

    options = [
        make_option("--version", action="store_true", help="print version info and exit"),
        make_option("--sqlite", dest="sqlite", metavar='FILE', default="/tmp/sqlite.db",
            help="use SQLite3 database"),
        make_option("--port", dest="port", default=8000, help="webserver port"),
        make_option("--test", dest="production", action="store_false", default=True,
            help=SUPPRESS_HELP),
    ]

    parser = OptionParser(option_list=options)
    options, args = parser.parse_args()

    if options.version:
        print 'isso', __version__
        sys.exit(0)

    app = Isso({'SQLITE': options.sqlite, 'PRODUCTION': options.production})

    if len(args) > 0 and args[0] == 'import':
        if len(args) < 2:
            print 'Usage: isso import FILE'
            sys.exit(2)

        with io.open(args[1], encoding='utf-8') as fp:
            migrate.disqus(app.db, fp.read())
    else:
        app = SharedDataMiddleware(app, {
            '/static': join(dirname(__file__), 'static'),
            '/js': join(dirname(__file__), 'js')})
        run_simple('127.0.0.1', 8000, app, use_reloader=True)