def autocompile(ws, conf, env):
"""Subcommand: autocompile -- automatically re-compiles when something in
content-dir has changed and parallel serving files."""
mtime = -1
cmtime = getmtime('conf.py')
# config content_extension originally defined as string, not a list
exts = conf.get('content_extension', ['.txt', '.rst', '.md'])
if isinstance(exts, string_types):
whitelist = (exts, )
else:
whitelist = tuple(exts)
while True:
ws.wait = True
ntime = max(
max(
getmtime(e) for e in readers.filelist(conf['content_dir'],
conf['content_ignore'])
if e.endswith(whitelist)),
max(
getmtime(p) for p in chain([
f for theme in conf['theme']
for f in readers.filelist(theme, conf['theme_ignore'])
], readers.filelist(conf['static'], conf['static_ignore']))))
if mtime != ntime:
try:
compile(conf, env)
except (SystemExit, KeyboardInterrupt):
raise
except Exception:
log.exception("uncaught exception during auto-compilation")
else:
conf = load(env.options.conf)
env = Environment.new(env)
event.reset()
mtime = ntime
ws.wait = False
if cmtime != getmtime('conf.py'):
log.info(' * Restarting due to change in conf.py')
# Kill the webserver
ws.shutdown()
# Restart acrylamid
os.execvp(sys.argv[0], sys.argv)
time.sleep(1)
def autocompile(ws, conf, env):
"""Subcommand: autocompile -- automatically re-compiles when something in
content-dir has changed and parallel serving files."""
mtime = -1
cmtime = getmtime('conf.py')
# config content_extension originally defined as string, not a list
exts = conf.get('content_extension',['.txt', '.rst', '.md'])
if isinstance(exts, string_types):
whitelist = (exts,)
else:
whitelist = tuple(exts)
while True:
ws.wait = True
ntime = max(
max(getmtime(e) for e in readers.filelist(
conf['content_dir'], conf['content_ignore']) if e.endswith(whitelist)),
max(getmtime(p) for p in chain(
readers.filelist(conf['theme'], conf['theme_ignore']),
readers.filelist(conf['static'], conf['static_ignore']))))
if mtime != ntime:
try:
compile(conf, env)
except (SystemExit, KeyboardInterrupt):
raise
except Exception:
log.exception("uncaught exception during auto-compilation")
else:
conf = load(env.options.conf)
env = Environment.new(env)
event.reset()
mtime = ntime
ws.wait = False
if cmtime != getmtime('conf.py'):
log.info(' * Restarting due to change in conf.py')
# Kill the webserver
ws.shutdown()
# Restart acrylamid
os.execvp(sys.argv[0], sys.argv)
time.sleep(1)
def autocompile(ws, conf, env):
"""Subcommand: autocompile -- automatically re-compiles when something in
content-dir has changed and parallel serving files."""
mtime = -1
cmtime = getmtime("conf.py")
while True:
ws.wait = True
ntime = max(
max(getmtime(e) for e in readers.filelist(conf["content_dir"], conf["content_ignore"]) if istext(e)),
max(
getmtime(p)
for p in chain(
readers.filelist(conf["theme"], conf["theme_ignore"]),
readers.filelist(conf["static"], conf["static_ignore"]),
)
),
)
if mtime != ntime:
try:
compile(conf, env)
except (SystemExit, KeyboardInterrupt):
raise
except Exception:
log.exception("uncaught exception during auto-compilation")
else:
conf = load(env.options.conf)
env = Environment.new(env)
event.reset()
mtime = ntime
ws.wait = False
if cmtime != getmtime("conf.py"):
log.info(" * Restarting due to change in conf.py")
# Kill the webserver
ws.shutdown()
# Restart acrylamid
os.execvp(sys.argv[0], sys.argv)
time.sleep(1)
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("--conf", dest="conf", help="alternate conf.py",
default="conf.py", metavar="/path/to/conf")
parser.add_argument("--version", action="version",
version=colors.blue('Acrylamid ') + dist.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("--ignore", dest="ignore", action="store_true",
help="ignore critical errors", default=False)
generate.add_argument("--search", dest="search", action="store_true",
help="build search index", 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("--ignore", dest="ignore", action="store_true",
help="ignore critical errors", default=False)
autocompile.add_argument("--search", dest="search", action="store_true",
help="build search index", 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
# temporary log to catch issues during task initialization
log.init('temporary', level=log.WARN, colors=False)
# 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 = core.Environment({'author': __author__, 'url': __url__,
'options': options, 'globals': Struct()})
try:
conf = core.load(options.conf)
except IOError:
log.critical('no conf.py found. Are you inside your blog?')
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)
# -- run -- #
if options.parser in ('gen', 'generate', 'co', 'compile'):
log.setLevel(options.verbosity)
try:
commands.compile(conf, env)
except AcrylamidException as e:
log.exception(e.args[0])
sys.exit(1)
elif options.parser in ('srv', 'serve', 'view'):
from acrylamid.lib.httpd import Webserver
ws = partial(Webserver, options.port, conf['output_dir'])
ws = ws(log.info) if options.verbosity < 20 else ws(); ws.start()
log.info(' * Running on http://127.0.0.1:%i/' % options.port)
try:
while True:
time.sleep(1)
except (SystemExit, KeyboardInterrupt) 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)
except (SystemExit, KeyboardInterrupt) 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.exception('uncaught exception')
sys.exit(1)
else:
log.critical('No such command!')
sys.exit(2)
sys.exit(0)