def Acryl():
"""The main function that dispatches the CLI. We use :class:`AcrylFormatter`
as custom help formatter that ommits the useless list of available subcommands
and their aliases.
All flags from acrylamid --help are also available in subcommands altough not
explicitely printed in their help."""
parser = argparse.ArgumentParser(
parents=[], formatter_class=AcrylFormatter
)
parser.add_argument("-v", "--verbose", action="store_const", dest="verbosity",
help="more verbose", const=log.SKIP, default=log.INFO)
parser.add_argument("-q", "--quiet", action="store_const", dest="verbosity",
help="less verbose", const=log.WARN)
parser.add_argument("-C", "--no-color", action="store_false", dest="colors",
help="disable color", default=True)
parser.add_argument("--version", action="version",
version=colors.blue('acrylamid ') + __version__)
subparsers = parser.add_subparsers(dest="parser")
# a repeat yourself of default arguments but not visible on subcommand --help
default = argparse.ArgumentParser(add_help=False)
default.add_argument("-v", "--verbose", action="store_const", dest="verbosity",
help=argparse.SUPPRESS, const=log.SKIP, default=log.INFO)
default.add_argument("-q", "--quiet", action="store_const", dest="verbosity",
help=argparse.SUPPRESS, const=log.WARN)
default.add_argument("-C", "--no-color", action="store_false", dest="colors",
help=argparse.SUPPRESS, default=True)
# --- gen params --- #
generate = subparsers.add_parser('compile', help='compile blog', parents=[default])
generate.add_argument("-f", "--force", action="store_true", dest="force",
help="clear cache before compilation", default=False)
generate.add_argument("-n", "--dry-run", dest="dryrun", action='store_true',
help="show what would have been compiled", default=False)
generate.add_argument("-i", "--ignore", dest="ignore", action="store_true",
help="ignore critical errors", default=False)
# --- webserver params --- #
view = subparsers.add_parser('view', help="fire up built-in webserver", parents=[default])
view.add_argument("-p", "--port", dest="port", type=int, default=8000,
help="webserver port")
# --- aco params --- #
autocompile = subparsers.add_parser('autocompile', help="automatic compilation and serving",
parents=[default])
autocompile.add_argument("-f", "--force", action="store_true", dest="force",
help="clear cache before compilation", default=False)
autocompile.add_argument("-n", "--dry-run", dest="dryrun", action='store_true',
help="show what would have been compiled", default=False)
autocompile.add_argument("-i", "--ignore", dest="ignore", action="store_true",
help="ignore critical errors", default=False)
autocompile.add_argument("-p", "--port", dest="port", type=int, default=8000,
help="webserver port")
for alias in ('co', 'gen', 'generate'):
subparsers._name_parser_map[alias] = generate
for alias in ('serve', 'srv'):
subparsers._name_parser_map[alias] = view
subparsers._name_parser_map['aco'] = autocompile
new = subparsers.add_parser('new', help="create a new entry", parents=[default],
epilog=("Takes all leading [args] as title or prompt if none given. creates "
"a new entry based on your PERMALINK_FORMAT and opens it with your "
"favourite $EDITOR."))
new.add_argument("title", nargs="*", default='')
# initialize other tasks
tasks.initialize(subparsers, default)
# parse args
options = parser.parse_args()
# initialize colored logger
log.init('acrylamid', level=options.verbosity, colors=options.colors)
env = Struct({'version': __version__, 'author': __author__, 'url': __url__})
env['options'] = options
env['globals'] = Struct()
# -- init -- #
# TODO: acrylamid init --layout_dir=somedir to overwrite defaults
if options.parser in ('init', ):
tasks.collected[options.parser](env, options)
sys.exit(0)
# -- teh real thing -- #
conf = Struct(defaults.conf)
try:
ns = dict([(k.upper(), v) for k, v in defaults.conf.iteritems()])
os.chdir(os.path.dirname(find('conf.py', os.getcwd())))
execfile('conf.py', ns)
conf.update(dict([(k.lower(), ns[k]) for k in ns if k.upper() == k]))
except IOError:
log.critical('no conf.py found. Try "acrylamid init".')
sys.exit(1)
except Exception as e:
log.critical("%s in `conf.py`" % e.__class__.__name__)
traceback.print_exc(file=sys.stdout)
sys.exit(1)
conf['output_dir'] = conf.get('output_dir', 'output/')
conf['content_dir'] = conf.get('content_dir', 'content/')
conf['layout_dir'] = conf.get('layout_dir', 'layouts/')
# -- run -- #
if options.parser in ('gen', 'generate', 'co', 'compile'):
log.setLevel(options.verbosity)
try:
commands.compile(conf, env, **options.__dict__)
except AcrylamidException as e:
log.fatal(e.args[0])
sys.exit(1)
elif options.parser in ('new', 'create'):
try:
commands.new(conf, env, title=' '.join(options.title), prompt=log.level()<log.WARN)
except AcrylamidException as e:
log.fatal(e.args[0])
sys.exit(1)
elif options.parser in ('srv', 'serve', 'view'):
from acrylamid.lib.httpd import Webserver
ws = Webserver(options.port, conf['output_dir']); ws.start()
log.info(' * Running on http://127.0.0.1:%i/' % options.port)
try:
while True:
time.sleep(1)
except (SystemExit, KeyboardInterrupt, Exception) as e:
ws.kill_received = True
sys.exit(0)
elif options.parser in ('aco', 'autocompile'):
from acrylamid.lib.httpd import Webserver
# XXX compile on request _or_ use inotify/fsevent
ws = Webserver(options.port, conf['output_dir']); ws.start()
log.info(' * Running on http://127.0.0.1:%i/' % options.port)
try:
commands.autocompile(ws, conf, env, **options.__dict__)
except (SystemExit, KeyboardInterrupt, Exception) as e:
ws.kill_received = True
log.error(e.args[0])
traceback.print_exc(file=sys.stdout)
sys.exit(0)
elif options.parser in tasks.collected:
try:
tasks.collected[options.parser](conf, env, options)
except AcrylamidException as e:
log.critical(e.args[0])
sys.exit(1)
else:
log.critical('No such command!')
sys.exit(2)
sys.exit(0)
assert defaults.check_conf(conf)
conf.update(dict((k, v) for k, v in options.__dict__.iteritems() if v != None))
# -- run -- #
if options.parser in ('gen', 'generate', 'co', 'compile'):
log.setLevel(options.verbosity)
try:
commands.compile(conf, env, **options.__dict__)
except AcrylamidException as e:
log.fatal(e.args[0])
sys.exit(1)
elif options.parser in ('new', 'create'):
try:
commands.new(conf, env, title=' '.join(options.title), prompt=log.level()<log.WARN)
except AcrylamidException as e:
log.fatal(e.args[0])
sys.exit(1)
elif options.parser in ('srv', 'serve', 'view'):
from acrylamid.lib.httpd import Webserver
ws = Webserver(options.port, conf['output_dir']); ws.start()
log.info(' * Running on http://127.0.0.1:%i/' % options.port)
try:
while True:
time.sleep(1)
except (SystemExit, KeyboardInterrupt, Exception) as e:
ws.kill_received = True
sys.exit(0)