146 lines
4.9 KiB
Python
146 lines
4.9 KiB
Python
#!/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.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']),
|
||
], converters={'re': RegexConverter})
|
||
|
||
|
||
class Isso:
|
||
|
||
PRODUCTION = True
|
||
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')})
|
||
run_simple('127.0.0.1', 8000, app, use_reloader=True)
|