Browse Source

PEP8 for cli.py

pull/1/merge
alshain 14 years ago
parent
commit
89b783c3d1
  1. 55
      couchpotato/cli.py

55
couchpotato/cli.py

@ -1,8 +1,3 @@
from couchpotato import web
from couchpotato.api import api
from couchpotato.core.logger import CPLog
from couchpotato.core.settings import settings
from libs.daemon import createDaemon
from logging import handlers from logging import handlers
import argparse import argparse
import logging import logging
@ -10,15 +5,30 @@ import os.path
import sys import sys
def cmd_couchpotato(base_path, argv = None): from libs.daemon import createDaemon
from couchpotato import web
from couchpotato.api import api
from couchpotato.core.logger import CPLog
from couchpotato.core.settings import settings
def cmd_couchpotato(base_path, argv=None):
'''Commandline entry point.''' '''Commandline entry point.'''
# Options # Options
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('-s', '--datadir', dest = 'data_dir', default = base_path, help = 'Absolute or ~/ path, where settings/logs/database data is saved (default ./)') parser.add_argument('-s', '--datadir', dest='data_dir',
parser.add_argument('-t', '--test', '--debug', action = 'store_true', dest = 'debug', help = 'Debug mode') default=base_path,
parser.add_argument('-q', '--quiet', action = 'store_true', dest = 'quiet', help = "Don't log to console") help="Absolute or ~/ path, where "
parser.add_argument('-d', '--daemon', action = 'store_true', dest = 'daemonize', help = 'Daemonize the app') "settings/logs/database data is saved (default ./)")
parser.add_argument('-t', '--test', '--debug', action='store_true',
dest='debug', help='Debug mode')
parser.add_argument('-q', '--quiet', action='store_true', dest='quiet',
help="Don't log to console")
parser.add_argument('-d', '--daemon', action='store_true',
dest='daemonize', help='Daemonize the app')
args = argv if argv else sys.argv[1:] args = argv if argv else sys.argv[1:]
options = parser.parse_args(args) options = parser.parse_args(args)
@ -37,7 +47,7 @@ def cmd_couchpotato(base_path, argv = None):
# Register settings # Register settings
settings.setFile(os.path.join(options.data_dir, 'settings.conf')) settings.setFile(os.path.join(options.data_dir, 'settings.conf'))
debug = options.debug or settings.get('debug', default = False) debug = options.debug or settings.get('debug', default=False)
# Daemonize app # Daemonize app
if options.daemonize: if options.daemonize:
@ -45,7 +55,8 @@ def cmd_couchpotato(base_path, argv = None):
# Logger # Logger
logger = logging.getLogger() logger = logging.getLogger()
formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s', '%H:%M:%S') formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s',
'%H:%M:%S')
level = logging.DEBUG if debug else logging.INFO level = logging.DEBUG if debug else logging.INFO
logger.setLevel(level) logger.setLevel(level)
@ -56,7 +67,9 @@ def cmd_couchpotato(base_path, argv = None):
logger.addHandler(hdlr) logger.addHandler(hdlr)
# To file # To file
hdlr2 = handlers.RotatingFileHandler(os.path.join(log_dir, 'CouchPotato.log'), 'a', 5000000, 4) hdlr2 = handlers.RotatingFileHandler(os.path.join(log_dir,
'CouchPotato.log'),
'a', 5000000, 4)
hdlr2.setFormatter(formatter) hdlr2.setFormatter(formatter)
logger.addHandler(hdlr2) logger.addHandler(hdlr2)
@ -71,7 +84,7 @@ def cmd_couchpotato(base_path, argv = None):
# Load configs # Load configs
from couchpotato.core.settings.loader import SettingsLoader from couchpotato.core.settings.loader import SettingsLoader
sl = SettingsLoader(root = base_path) sl = SettingsLoader(root=base_path)
sl.addConfig('couchpotato', 'core') sl.addConfig('couchpotato', 'core')
sl.run() sl.run()
@ -83,20 +96,20 @@ def cmd_couchpotato(base_path, argv = None):
reloader = debug and not options.daemonize reloader = debug and not options.daemonize
# Basic config # Basic config
app.host = settings.get('host', default = '0.0.0.0') app.host = settings.get('host', default='0.0.0.0')
app.port = settings.get('port', default = 5000) app.port = settings.get('port', default=5000)
app.debug = debug app.debug = debug
app.secret_key = api_key app.secret_key = api_key
app.static_path = url_base + 'static' app.static_path = url_base + 'static'
# Add static url with url_base # Add static url with url_base
app.add_url_rule(app.static_path + '/<path:filename>', app.add_url_rule(app.static_path + '/<path:filename>',
endpoint = 'static', endpoint='static',
view_func = app.send_static_file) view_func=app.send_static_file)
# Register modules # Register modules
app.register_module(web, url_prefix = url_base) app.register_module(web, url_prefix=url_base)
app.register_module(api, url_prefix = '%s/%s' % (url_base + 'api', api_key)) app.register_module(api, url_prefix='%s/%s' % (url_base + 'api', api_key))
# Go go go! # Go go go!
app.run(use_reloader = reloader) app.run(use_reloader=reloader)

Loading…
Cancel
Save