PEP 8
eric
parent
1554300013
commit
08ea4a11d7
252
CherryPyApp.py
252
CherryPyApp.py
|
|
@ -32,8 +32,8 @@ from libgutenberg import GutenbergDatabase
|
|||
import i18n_tool
|
||||
# Make translator functions available everywhere. Do this early, at
|
||||
# least before Genshi starts loading templates.
|
||||
builtins._ = i18n_tool.ugettext
|
||||
builtins.__ = i18n_tool.ungettext
|
||||
builtins._ = i18n_tool.ugettext
|
||||
builtins.__ = i18n_tool.ungettext
|
||||
|
||||
import ConnectionPool
|
||||
import Page
|
||||
|
|
@ -50,12 +50,12 @@ import Formatters
|
|||
import Timer
|
||||
|
||||
plugins.Timer = Timer.TimerPlugin
|
||||
install_dir = os.path.dirname (os.path.abspath (__file__))
|
||||
install_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
|
||||
CHERRYPY_CONFIG = os.path.join(install_dir, 'CherryPy.conf')
|
||||
LOCAL_CONFIG = [os.path.expanduser('~/.autocat3'), '/etc/autocat3.conf']
|
||||
|
||||
class MyRoutesDispatcher (cherrypy.dispatch.RoutesDispatcher):
|
||||
class MyRoutesDispatcher(cherrypy.dispatch.RoutesDispatcher):
|
||||
""" Dispatcher that tells us the matched route.
|
||||
|
||||
CherryPy makes it hard for us by forgetting the matched route object.
|
||||
|
|
@ -63,22 +63,22 @@ class MyRoutesDispatcher (cherrypy.dispatch.RoutesDispatcher):
|
|||
|
||||
"""
|
||||
|
||||
def connect (self, name, route, controller, **kwargs):
|
||||
def connect(self, name, route, controller, **kwargs):
|
||||
""" Add a 'route_name' parameter that will tell us the matched route. """
|
||||
kwargs['route_name'] = name
|
||||
kwargs.setdefault ('action', 'index')
|
||||
cherrypy.dispatch.RoutesDispatcher.connect (self, name, route, controller, **kwargs)
|
||||
kwargs.setdefault('action', 'index')
|
||||
cherrypy.dispatch.RoutesDispatcher.connect(self, name, route, controller, **kwargs)
|
||||
|
||||
|
||||
def main ():
|
||||
def main():
|
||||
""" Main function. """
|
||||
|
||||
# default config
|
||||
cherrypy.config.update ({
|
||||
cherrypy.config.update({
|
||||
'uid': 0,
|
||||
'gid': 0,
|
||||
'server_name': 'localhost',
|
||||
'genshi.template_dir': os.path.join (install_dir, 'templates'),
|
||||
'genshi.template_dir': os.path.join(install_dir, 'templates'),
|
||||
'daemonize': False,
|
||||
'pidfile': None,
|
||||
'host': 'localhost',
|
||||
|
|
@ -87,7 +87,7 @@ def main ():
|
|||
})
|
||||
|
||||
cherrypy.config.update(CHERRYPY_CONFIG)
|
||||
|
||||
|
||||
extra_config = ''
|
||||
for config_filename in LOCAL_CONFIG:
|
||||
try:
|
||||
|
|
@ -108,217 +108,217 @@ def main ():
|
|||
cherrypy.log.error_file = ""
|
||||
cherrypy.log.access_file = ""
|
||||
|
||||
max_bytes = getattr (cherrypy.log, "rot_max_bytes", 100 * 1024 * 1024)
|
||||
backup_count = getattr (cherrypy.log, "rot_backup_count", 2)
|
||||
#print(os.path.abspath(error_file)+": Filehandler cherrypy")
|
||||
h = logging.handlers.RotatingFileHandler (error_file, 'a', max_bytes, backup_count, 'utf-8')
|
||||
h.setLevel (logging.DEBUG)
|
||||
h.setFormatter (cherrypy._cplogging.logfmt)
|
||||
cherrypy.log.error_log.addHandler (h)
|
||||
|
||||
h = logging.handlers.RotatingFileHandler (access_file, 'a', max_bytes, backup_count, 'utf-8')
|
||||
h.setLevel (logging.DEBUG)
|
||||
h.setFormatter (cherrypy._cplogging.logfmt)
|
||||
cherrypy.log.access_log.addHandler (h)
|
||||
# set up python logging
|
||||
|
||||
max_bytes = getattr(cherrypy.log, "rot_max_bytes", 100 * 1024 * 1024)
|
||||
backup_count = getattr(cherrypy.log, "rot_backup_count", 2)
|
||||
|
||||
h = logging.handlers.RotatingFileHandler(error_file, 'a', max_bytes, backup_count, 'utf-8')
|
||||
h.setLevel(logging.INFO)
|
||||
h.setFormatter(cherrypy._cplogging.logfmt)
|
||||
cherrypy.log.error_log.addHandler(h)
|
||||
|
||||
h = logging.handlers.RotatingFileHandler(access_file, 'a', max_bytes, backup_count, 'utf-8')
|
||||
h.setLevel(logging.INFO)
|
||||
h.setFormatter(cherrypy._cplogging.logfmt)
|
||||
cherrypy.log.access_log.addHandler(h)
|
||||
|
||||
|
||||
|
||||
if not cherrypy.config['daemonize']:
|
||||
ch = logging.StreamHandler ()
|
||||
ch.setLevel (logging.DEBUG)
|
||||
ch.setFormatter (cherrypy._cplogging.logfmt)
|
||||
cherrypy.log.error_log.addHandler (ch)
|
||||
ch = logging.StreamHandler()
|
||||
ch.setLevel(logging.DEBUG)
|
||||
ch.setFormatter(cherrypy._cplogging.logfmt)
|
||||
cherrypy.log.error_log.addHandler(ch)
|
||||
|
||||
# continue app init
|
||||
#
|
||||
|
||||
cherrypy.log ('*' * 80, context = 'ENGINE', severity = logging.INFO)
|
||||
cherrypy.log ("Using config file '%s'." % CHERRYPY_CONFIG,
|
||||
context = 'ENGINE', severity = logging.INFO)
|
||||
cherrypy.log('*' * 80, context='ENGINE', severity=logging.INFO)
|
||||
cherrypy.log("Using config file '%s'." % CHERRYPY_CONFIG,
|
||||
context='ENGINE', severity=logging.INFO)
|
||||
if extra_config:
|
||||
cherrypy.log('extra_config: %s' % extra_config, context='ENGINE', severity=logging.INFO)
|
||||
|
||||
# after cherrypy.config is parsed
|
||||
Formatters.init ()
|
||||
cherrypy.log ("Continuing App Init", context = 'ENGINE', severity = logging.INFO)
|
||||
Formatters.init()
|
||||
cherrypy.log("Continuing App Init", context='ENGINE', severity=logging.INFO)
|
||||
|
||||
cherrypy.log ("Continuing App Init", context = 'ENGINE', severity = logging.INFO)
|
||||
cherrypy.tools.I18nTool = i18n_tool.I18nTool ()
|
||||
cherrypy.log("Continuing App Init", context='ENGINE', severity=logging.INFO)
|
||||
cherrypy.tools.I18nTool = i18n_tool.I18nTool()
|
||||
|
||||
cherrypy.log ("Continuing App Init", context = 'ENGINE', severity = logging.INFO)
|
||||
cherrypy.log("Continuing App Init", context='ENGINE', severity=logging.INFO)
|
||||
|
||||
# Used to bust the cache on js and css files. This should be the
|
||||
# files' mtime, but the files are not stored on the app server.
|
||||
# This is a `good enough´ replacement though.
|
||||
t = str (int (time.time ()))
|
||||
t = str(int(time.time()))
|
||||
cherrypy.config['css_mtime'] = t
|
||||
cherrypy.config['js_mtime'] = t
|
||||
|
||||
cherrypy.config['all_hosts'] = (
|
||||
cherrypy.config['host'], cherrypy.config['host_mobile'], cherrypy.config['file_host'])
|
||||
|
||||
if hasattr (cherrypy.engine, 'signal_handler'):
|
||||
cherrypy.engine.signal_handler.subscribe ()
|
||||
if hasattr(cherrypy.engine, 'signal_handler'):
|
||||
cherrypy.engine.signal_handler.subscribe()
|
||||
|
||||
GutenbergDatabase.options.update(cherrypy.config)
|
||||
cherrypy.engine.pool = plugins.ConnectionPool (
|
||||
cherrypy.engine, params = GutenbergDatabase.get_connection_params (cherrypy.config))
|
||||
cherrypy.engine.pool.subscribe ()
|
||||
cherrypy.engine.pool = plugins.ConnectionPool(
|
||||
cherrypy.engine, params=GutenbergDatabase.get_connection_params(cherrypy.config))
|
||||
cherrypy.engine.pool.subscribe()
|
||||
|
||||
plugins.Timer (cherrypy.engine).subscribe ()
|
||||
plugins.Timer(cherrypy.engine).subscribe()
|
||||
|
||||
cherrypy.log ("Daemonizing", context = 'ENGINE', severity = logging.INFO)
|
||||
cherrypy.log("Daemonizing", context='ENGINE', severity=logging.INFO)
|
||||
|
||||
if cherrypy.config['daemonize']:
|
||||
plugins.Daemonizer (cherrypy.engine).subscribe ()
|
||||
plugins.Daemonizer(cherrypy.engine).subscribe()
|
||||
|
||||
uid = cherrypy.config['uid']
|
||||
gid = cherrypy.config['gid']
|
||||
if uid > 0 or gid > 0:
|
||||
plugins.DropPrivileges (cherrypy.engine, uid = uid, gid = gid, umask = 0o22).subscribe ()
|
||||
plugins.DropPrivileges(cherrypy.engine, uid=uid, gid=gid, umask=0o22).subscribe()
|
||||
|
||||
if cherrypy.config['pidfile']:
|
||||
pid = plugins.PIDFile (cherrypy.engine, cherrypy.config['pidfile'])
|
||||
# Write pidfile after privileges are dropped (prio == 77)
|
||||
pid = plugins.PIDFile(cherrypy.engine, cherrypy.config['pidfile'])
|
||||
# Write pidfile after privileges are dropped(prio == 77)
|
||||
# or we will not be able to remove it.
|
||||
cherrypy.engine.subscribe ('start', pid.start, 78)
|
||||
cherrypy.engine.subscribe ('exit', pid.exit, 78)
|
||||
cherrypy.engine.subscribe('start', pid.start, 78)
|
||||
cherrypy.engine.subscribe('exit', pid.exit, 78)
|
||||
|
||||
|
||||
cherrypy.log ("Setting up routes", context = 'ENGINE', severity = logging.INFO)
|
||||
cherrypy.log("Setting up routes", context='ENGINE', severity=logging.INFO)
|
||||
|
||||
# setup 'routes' dispatcher
|
||||
#
|
||||
# d = cherrypy.dispatch.RoutesDispatcher (full_result = True)
|
||||
d = MyRoutesDispatcher (full_result = True)
|
||||
# d = cherrypy.dispatch.RoutesDispatcher(full_result=True)
|
||||
d = MyRoutesDispatcher(full_result=True)
|
||||
cherrypy.routes_mapper = d.mapper
|
||||
|
||||
def check_id (environ, result):
|
||||
def check_id(environ, result):
|
||||
""" Check if id is a valid number. """
|
||||
try:
|
||||
return str (int (result['id'])) == result['id']
|
||||
return str(int(result['id'])) == result['id']
|
||||
except:
|
||||
return False
|
||||
|
||||
d.connect ('start', r'/ebooks{.format}/',
|
||||
controller = StartPage.Start ())
|
||||
d.connect('start', r'/ebooks{.format}/',
|
||||
controller=StartPage.Start())
|
||||
|
||||
d.connect ('suggest', r'/ebooks/suggest{.format}/',
|
||||
controller = SuggestionsPage.Suggestions ())
|
||||
d.connect('suggest', r'/ebooks/suggest{.format}/',
|
||||
controller=SuggestionsPage.Suggestions())
|
||||
|
||||
# search pages
|
||||
|
||||
d.connect ('search', r'/ebooks/search{.format}/',
|
||||
controller = BookSearchPage ())
|
||||
d.connect('search', r'/ebooks/search{.format}/',
|
||||
controller=BookSearchPage())
|
||||
|
||||
d.connect ('author_search', r'/ebooks/authors/search{.format}/',
|
||||
controller = AuthorSearchPage ())
|
||||
d.connect('author_search', r'/ebooks/authors/search{.format}/',
|
||||
controller=AuthorSearchPage())
|
||||
|
||||
d.connect ('subject_search', r'/ebooks/subjects/search{.format}/',
|
||||
controller = SubjectSearchPage ())
|
||||
d.connect('subject_search', r'/ebooks/subjects/search{.format}/',
|
||||
controller=SubjectSearchPage())
|
||||
|
||||
d.connect ('bookshelf_search', r'/ebooks/bookshelves/search{.format}/',
|
||||
controller = BookshelfSearchPage ())
|
||||
d.connect('bookshelf_search', r'/ebooks/bookshelves/search{.format}/',
|
||||
controller=BookshelfSearchPage())
|
||||
|
||||
# 'id' pages
|
||||
|
||||
d.connect ('author', r'/ebooks/author/{id:\d+}{.format}',
|
||||
controller = AuthorPage (), conditions = dict (function = check_id))
|
||||
d.connect('author', r'/ebooks/author/{id:\d+}{.format}',
|
||||
controller=AuthorPage(), conditions=dict(function=check_id))
|
||||
|
||||
d.connect ('subject', r'/ebooks/subject/{id:\d+}{.format}',
|
||||
controller = SubjectPage (), conditions = dict (function = check_id))
|
||||
d.connect('subject', r'/ebooks/subject/{id:\d+}{.format}',
|
||||
controller=SubjectPage(), conditions=dict(function=check_id))
|
||||
|
||||
d.connect ('bookshelf', r'/ebooks/bookshelf/{id:\d+}{.format}',
|
||||
controller = BookshelfPage (), conditions = dict (function = check_id))
|
||||
d.connect('bookshelf', r'/ebooks/bookshelf/{id:\d+}{.format}',
|
||||
controller=BookshelfPage(), conditions=dict(function=check_id))
|
||||
|
||||
d.connect ('also', r'/ebooks/{id:\d+}/also/{.format}',
|
||||
controller = AlsoDownloadedPage (), conditions = dict (function = check_id))
|
||||
d.connect('also', r'/ebooks/{id:\d+}/also/{.format}',
|
||||
controller=AlsoDownloadedPage(), conditions=dict(function=check_id))
|
||||
|
||||
# bibrec pages
|
||||
|
||||
d.connect ('download', r'/ebooks/{id:\d+}/download{.format}',
|
||||
controller = Page.NullPage (), _static = True)
|
||||
d.connect('download', r'/ebooks/{id:\d+}/download{.format}',
|
||||
controller=Page.NullPage(), _static=True)
|
||||
|
||||
d.connect ('bibrec', r'/ebooks/{id:\d+}{.format}',
|
||||
controller = BibrecPage (), conditions = dict (function = check_id))
|
||||
d.connect('bibrec', r'/ebooks/{id:\d+}{.format}',
|
||||
controller=BibrecPage(), conditions=dict(function=check_id))
|
||||
|
||||
|
||||
# legacy compatibility with /ebooks/123.bibrec
|
||||
d.connect ('bibrec2', r'/ebooks/{id:\d+}.bibrec{.format}',
|
||||
controller = BibrecPage (), conditions = dict (function = check_id))
|
||||
d.connect('bibrec2', r'/ebooks/{id:\d+}.bibrec{.format}',
|
||||
controller=BibrecPage(), conditions=dict(function=check_id))
|
||||
|
||||
d.connect ('cover', r'/covers/{size:small|medium}/{order:latest|popular}/{count}',
|
||||
controller = CoverPages.CoverPages ())
|
||||
d.connect('cover', r'/covers/{size:small|medium}/{order:latest|popular}/{count}',
|
||||
controller=CoverPages.CoverPages())
|
||||
|
||||
d.connect ('qrcode', r'/qrcode/',
|
||||
controller = QRCodePage.QRCodePage ())
|
||||
d.connect('qrcode', r'/qrcode/',
|
||||
controller=QRCodePage.QRCodePage())
|
||||
|
||||
d.connect ('iplimit', r'/iplimit/',
|
||||
controller = Page.NullPage ())
|
||||
d.connect('iplimit', r'/iplimit/',
|
||||
controller=Page.NullPage())
|
||||
|
||||
d.connect ('stats', r'/stats/',
|
||||
controller = Page.NullPage (), _static = True)
|
||||
d.connect('stats', r'/stats/',
|
||||
controller=Page.NullPage(), _static=True)
|
||||
|
||||
d.connect ('honeypot_send', r'/ebooks/send/megaupload/{id:\d+}.{filetype}',
|
||||
controller = Page.NullPage (), _static = True)
|
||||
d.connect('honeypot_send', r'/ebooks/send/megaupload/{id:\d+}.{filetype}',
|
||||
controller=Page.NullPage(), _static=True)
|
||||
|
||||
# /w/captcha/question/ so varnish will cache it
|
||||
d.connect ('captcha.question', r'/w/captcha/question/',
|
||||
controller = Page.GoHomePage ())
|
||||
d.connect('captcha.question', r'/w/captcha/question/',
|
||||
controller=Page.GoHomePage())
|
||||
|
||||
d.connect ('captcha.answer', r'/w/captcha/answer/',
|
||||
controller = Page.GoHomePage ())
|
||||
d.connect('captcha.answer', r'/w/captcha/answer/',
|
||||
controller=Page.GoHomePage())
|
||||
|
||||
# sitemap protocol access control requires us to place sitemaps in /ebooks/
|
||||
d.connect ('sitemap', r'/ebooks/sitemaps/',
|
||||
controller = Sitemap.SitemapIndex ())
|
||||
d.connect('sitemap', r'/ebooks/sitemaps/',
|
||||
controller=Sitemap.SitemapIndex())
|
||||
|
||||
d.connect ('sitemap_index', r'/ebooks/sitemaps/{page:\d+}',
|
||||
controller = Sitemap.Sitemap ())
|
||||
d.connect('sitemap_index', r'/ebooks/sitemaps/{page:\d+}',
|
||||
controller=Sitemap.Sitemap())
|
||||
|
||||
if 'dropbox_client_id' in cherrypy.config:
|
||||
import Dropbox
|
||||
dropbox = Dropbox.Dropbox ()
|
||||
cherrypy.log ("Dropbox Client Id: %s" % cherrypy.config['dropbox_client_id'],
|
||||
context = 'ENGINE', severity = logging.INFO)
|
||||
d.connect ('dropbox_send', r'/ebooks/send/dropbox/{id:\d+}.{filetype}',
|
||||
controller = dropbox, conditions = dict (function = check_id))
|
||||
d.connect ('dropbox_callback', r'/ebooks/send/dropbox/',
|
||||
controller = dropbox)
|
||||
dropbox = Dropbox.Dropbox()
|
||||
cherrypy.log("Dropbox Client Id: %s" % cherrypy.config['dropbox_client_id'],
|
||||
context='ENGINE', severity=logging.INFO)
|
||||
d.connect('dropbox_send', r'/ebooks/send/dropbox/{id:\d+}.{filetype}',
|
||||
controller=dropbox, conditions=dict(function=check_id))
|
||||
d.connect('dropbox_callback', r'/ebooks/send/dropbox/',
|
||||
controller=dropbox)
|
||||
|
||||
if 'gdrive_client_id' in cherrypy.config:
|
||||
import GDrive
|
||||
gdrive = GDrive.GDrive ()
|
||||
cherrypy.log ("GDrive Client Id: %s" % cherrypy.config['gdrive_client_id'],
|
||||
context = 'ENGINE', severity = logging.INFO)
|
||||
d.connect ('gdrive_send', r'/ebooks/send/gdrive/{id:\d+}.{filetype}',
|
||||
controller = gdrive, conditions = dict (function = check_id))
|
||||
d.connect ('gdrive_callback', r'/ebooks/send/gdrive/',
|
||||
controller = gdrive)
|
||||
gdrive = GDrive.GDrive()
|
||||
cherrypy.log("GDrive Client Id: %s" % cherrypy.config['gdrive_client_id'],
|
||||
context='ENGINE', severity=logging.INFO)
|
||||
d.connect('gdrive_send', r'/ebooks/send/gdrive/{id:\d+}.{filetype}',
|
||||
controller=gdrive, conditions=dict(function=check_id))
|
||||
d.connect('gdrive_callback', r'/ebooks/send/gdrive/',
|
||||
controller=gdrive)
|
||||
|
||||
if 'msdrive_client_id' in cherrypy.config:
|
||||
import MSDrive
|
||||
msdrive = MSDrive.MSDrive ()
|
||||
cherrypy.log ("MSDrive Client Id: %s" % cherrypy.config['msdrive_client_id'],
|
||||
context = 'ENGINE', severity = logging.INFO)
|
||||
d.connect ('msdrive_send', r'/ebooks/send/msdrive/{id:\d+}.{filetype}',
|
||||
controller = msdrive, conditions = dict (function = check_id))
|
||||
d.connect ('msdrive_callback', r'/ebooks/send/msdrive/',
|
||||
controller = msdrive)
|
||||
msdrive = MSDrive.MSDrive()
|
||||
cherrypy.log("MSDrive Client Id: %s" % cherrypy.config['msdrive_client_id'],
|
||||
context='ENGINE', severity=logging.INFO)
|
||||
d.connect('msdrive_send', r'/ebooks/send/msdrive/{id:\d+}.{filetype}',
|
||||
controller=msdrive, conditions=dict(function=check_id))
|
||||
d.connect('msdrive_callback', r'/ebooks/send/msdrive/',
|
||||
controller=msdrive)
|
||||
|
||||
# start http server
|
||||
#
|
||||
|
||||
cherrypy.log ("Mounting root", context = 'ENGINE', severity = logging.INFO)
|
||||
cherrypy.log("Mounting root", context='ENGINE', severity=logging.INFO)
|
||||
|
||||
app = cherrypy.tree.mount (root = None, config = CHERRYPY_CONFIG)
|
||||
app = cherrypy.tree.mount(root=None, config=CHERRYPY_CONFIG)
|
||||
|
||||
app.merge ({'/': {'request.dispatch': d}})
|
||||
app.merge({'/': {'request.dispatch': d}})
|
||||
return app
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main ()
|
||||
cherrypy.engine.start ()
|
||||
cherrypy.engine.block ()
|
||||
main()
|
||||
cherrypy.engine.start()
|
||||
cherrypy.engine.block()
|
||||
|
|
|
|||
Loading…
Reference in New Issue