def __init__(self, conf=None):
"""Sets configuration and environment and creates the Request
object"""
global sys
usage = "usage: %prog <subcommand> [options] [args]"
epilog = None
options, constants = (
[],
[
make_option("-h", "--help", action="store_true", help=SUPPRESS_HELP),
make_option(
"-v", "--verbose", action="store_const", dest="verbosity", help=SUPPRESS_HELP, const=log.SKIP
),
make_option(
"-q",
"--quit",
action="store_const",
dest="verbosity",
help=SUPPRESS_HELP,
const=log.WARN,
default=log.INFO,
),
make_option("-C", "--no-color", action="store_false", dest="colors", default=True, help=SUPPRESS_HELP),
make_option("--version", action="store_true", dest="version", help=SUPPRESS_HELP, default=False),
],
)
# reorder `prog help foo` and `prog --help foo`
if len(sys.argv) > 2 and sys.argv[1] in ("--help", "help"):
sys.argv[1], sys.argv[2] = sys.argv[2], "--help"
if len(sys.argv) <= 1 or sys.argv[1] in ("-h", "--help", "help"):
options, constants = (
[],
[
make_option(
"-q",
"--quiet",
action="store_const",
dest="verbosity",
help="less verbose",
const=log.WARN,
default=log.INFO,
),
make_option(
"-v", "--verbose", action="store_const", dest="verbosity", help="more verbose", const=log.SKIP
),
make_option(
"-C", "--no-color", action="store_false", dest="colors", default=True, help="disable color"
),
make_option("-h", "--help", action="store_true", help="show this help message and exit"),
make_option(
"--version", action="store_true", dest="version", help="print version details", default=False
),
],
)
epilog = (
"Commands:\n"
" init initializes base structure in DIR\n"
" create (new) creates a new entry\n"
" compile (co) compile blog\n"
" view fire up built-in webserver\n"
" autocompile automatic compilation and serving (short aco)\n"
" clean (rm) remove abandoned files\n"
" import import content from URL\n"
" deploy run a given TASK\n"
"\nAll subcommands except `init` require a conf.py file.\n"
)
# --- init params --- #
elif sys.argv[1] in ("init", "initialize"):
usage = "%prog " + sys.argv[1] + " [DEST|FILE] [-p]"
options = [
make_option(
"-f", "--force", action="store_true", dest="force", help="don't ask, just overwrite", default=False
),
make_option(
"--xhtml",
action="store_const",
dest="theme",
const="xhtml",
help="use XHTML theme",
default="html5",
),
make_option("--html5", action="store_const", dest="theme", const="html5", help="use HTML5 theme"),
]
# --- init params --- #
elif sys.argv[1] in ("new",):
usage = "%prog " + sys.argv[1] + " [args]"
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."
)
epilog = fill(epilog) + "\n"
# --- gen params --- #
elif sys.argv[1] in ("compile", "co", "generate", "gen"):
usage = "%prog " + sys.argv[1] + " [-fn]"
options = [
make_option(
"-f",
"--force",
action="store_true",
dest="force",
help="clear cache before compilation",
default=False,
),
make_option(
"-n",
"--dry-run",
dest="dryrun",
action="store_true",
default=False,
help="show what would have been compiled",
),
make_option(
"-i", "--ignore", dest="ignore", action="store_true", default=False, help="ignore critical errors"
),
]
# --- webserver params --- #
elif sys.argv[1] in ("view", "serve", "srv"):
usage = "%prog " + sys.argv[1] + " [-p]"
options = [make_option("-p", "--port", dest="port", type=int, default=8000, help="webserver port")]
# --- autocompile params --- #
elif sys.argv[1] in ("autocompile", "aco"):
usage = "%prog " + sys.argv[1] + " [-pf]"
options = [
make_option(
"-f",
"--force",
action="store_true",
dest="force",
help="clear cache before compilation",
default=False,
),
make_option(
"-i", "--ignore", dest="ignore", action="store_true", default=False, help="ignore critical errors"
),
make_option("-p", "--port", dest="port", type=int, default=8000, help="webserver port"),
]
# --- import params --- #
elif sys.argv[1] in ("import",):
usage = "%prog " + sys.argv[1] + " [-mk]"
options = [
make_option("-m", dest="import_fmt", default="Markdown", help="reconvert HTML to MARKUP"),
make_option("-k", "--keep-links", dest="keep_links", action="store_true", help="keep permanent links"),
]
else:
tasks = get_tasks()
if sys.argv[1] in tasks:
usage = tasks[sys.argv[1]].usage
options = tasks[sys.argv[1]].options
class AcrylParser(OptionParser):
# -- http://stackoverflow.com/q/1857346
def format_epilog(self, formatter):
if epilog is None:
return ""
return "\n" + self.epilog
parser = AcrylParser(option_list=options + constants, usage=usage, add_help_option=False, epilog=epilog)
(options, args) = parser.parse_args()
if len(sys.argv) <= 1 or sys.argv[1] == "help" or options.help:
parser.print_help()
sys.exit(0)
# initialize colored logger
log.init("acrylamid", level=options.verbosity, colors=options.colors)
env = Environment({"version": __version__, "author": __author__, "url": __url__})
env["options"] = options
if options.version:
print "acrylamid " + env.version
sys.exit(0)
# -- init -- #
# TODO: acrylamid init --layout_dir=somedir to overwrite defaults
if "init" in args:
if len(args) == 2:
defaults.init(args[1], options.theme, options.force)
else:
defaults.init(".", options.theme, options.force)
sys.exit(0)
# -- teh real thing -- #
conf = defaults.conf
try:
ns = dict([(k.upper(), v) for k, v in defaults.conf.iteritems()])
execfile("conf.py", ns)
conf.update(dict([(k.lower(), ns[k]) for k in ns if k.upper() == k]))
except OSError:
log.critical('no config file found: %s. Try "acrylamid init".', options.conf)
sys.exit(1)
except Exception, e:
log.critical("%s in `conf.py`" % e.__class__.__name__)
traceback.print_exc(file=sys.stdout)
sys.exit(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("--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)
# --- init params --- #
init = subparsers.add_parser('init', help='initializes base structure in DIR',
parents=[default, ])
init.add_argument("dest", metavar="DEST|FILE", nargs="?")
init.add_argument("-f", "--force", action="store_true", dest="force",
help="don't ask, just overwrite", default=False)
init.add_argument("--xhtml", action="store_const", dest="theme", const="xhtml",
help="use XHTML theme", default="html5")
init.add_argument("--html5", action="store_const", dest="theme", const="html5",
help="use HTML5 theme (default)")
init.add_argument("--mako", action="store_const", dest="engine", const="mako",
help="use the Mako template engine", default="jinja2")
init.add_argument("--jinja2", action="store_const", dest="engine", const="jinja2",
help="use the Jinja2 template engine (default)")
# --- 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")
# --- import params --- #
importer = subparsers.add_parser('import', help="import content from URL or FILE",
parents=[default])
importer.add_argument("src", metavar="FILE|URL")
importer.add_argument("-f", "--force", dest="force", action="store_true",
help="overwrite existing entries", default=False)
importer.add_argument("-m", dest="fmt", default="Markdown", help="reconvert HTML to FMT")
importer.add_argument("-k", "--keep-links", dest="keep", action="store_true",
help="keep permanent links", default=False)
importer.add_argument("-p", "--pandoc", dest="pandoc", action="store_true",
help="use pandoc first", default=False)
importer.add_argument("-a", dest="args", nargs="+", action="store", type=str,
help="add argument to header section", default=[])
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', ):
defaults.init(options.dest if options.dest else '.', options.theme, options.engine, options.force)
sys.exit(0)
# -- teh real thing -- #
conf = Struct(defaults.conf)
try:
ns = dict([(k.upper(), v) for k, v in defaults.conf.iteritems()])
execfile('conf.py', ns)
conf.update(dict([(k.lower(), ns[k]) for k in ns if k.upper() == k]))
except OSError:
log.critical('no config file found: %s. Try "acrylamid init".', options.conf)
sys.exit(1)
except Exception, e:
log.critical("%s in `conf.py`" % e.__class__.__name__)
traceback.print_exc(file=sys.stdout)
sys.exit(1)
