piecrust2: piecrust/main.py annotate

annotate piecrust/main.py @ 683:ec384174b8b2

internal: More work/fixes on how default/theme/user configs are merged. Change how the code is organized to have better data flow. Add some tests.

author	Ludovic Chabant <ludovic@chabant.com>
date	Wed, 09 Mar 2016 00:23:51 -0800
parents	81d9c3a3a0b5
children	cd3a00455b87

rev	line source
612 2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	1 import os
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	2 import os.path
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	3 import io
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	4 import sys
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	5 import time
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	6 import hashlib
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 import logging
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8 import argparse
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	9 import colorama
69 cb1ed436642c Always use version generated by `setup.py`. Better version generation. Ludovic Chabant <ludovic@chabant.com> parents: 61 diff changeset	10 from piecrust import APP_VERSION
466 456db44dcc53 bake: Pass the config variants and values from the CLI to the baker. Ludovic Chabant <ludovic@chabant.com> parents: 384 diff changeset	11 from piecrust.app import (
456db44dcc53 bake: Pass the config variants and values from the CLI to the baker. Ludovic Chabant <ludovic@chabant.com> parents: 384 diff changeset	12 PieCrust, PieCrustConfiguration, apply_variant_and_values)
341 bd726306d4dc chef: Add a `--config-set` option to set ad-hoc site configuration settings. Ludovic Chabant <ludovic@chabant.com> parents: 307 diff changeset	13 from piecrust.chefutil import (
bd726306d4dc chef: Add a `--config-set` option to set ad-hoc site configuration settings. Ludovic Chabant <ludovic@chabant.com> parents: 307 diff changeset	14 format_timed, log_friendly_exception, print_help_item)
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	15 from piecrust.commands.base import CommandContext
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	16 from piecrust.pathutil import SiteNotFoundError, find_app_root
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	17 from piecrust.plugins.base import PluginLoader
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	18
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	19
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	20 logger = logging.getLogger(__name__)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23 class ColoredFormatter(logging.Formatter):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	24 COLORS = {
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	25 'DEBUG': colorama.Fore.BLACK + colorama.Style.BRIGHT,
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	26 'INFO': '',
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	27 'WARNING': colorama.Fore.YELLOW,
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	28 'ERROR': colorama.Fore.RED,
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	29 'CRITICAL': colorama.Back.RED + colorama.Fore.WHITE
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	30 }
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	31
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	32 def __init__(self, fmt=None, datefmt=None):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	33 super(ColoredFormatter, self).__init__(fmt, datefmt)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	34
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	35 def format(self, record):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	36 color = self.COLORS.get(record.levelname)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	37 res = super(ColoredFormatter, self).format(record)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	38 if color:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	39 res = color + res + colorama.Style.RESET_ALL
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	40 return res
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	41
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	42
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	43 class NullPieCrust:
663 3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	44 def __init__(self, theme_site=False):
3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	45 self.theme_site = theme_site
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	46 self.root_dir = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	47 self.debug = False
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	48 self.templates_dirs = []
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	49 self.theme_dir = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	50 self.cache_dir = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	51 self.config = PieCrustConfiguration()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	52 self.plugin_loader = PluginLoader(self)
103 028df35a690e Fix using `chef` outside of a website. Ludovic Chabant <ludovic@chabant.com> parents: 99 diff changeset	53 self.env = None
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	54
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	55
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	56 def main():
167 0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	57 if sys.platform == 'darwin':
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	58 # There's a bug on MacOSX that can cause Python to be confused
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	59 # about the locale. Let's try to fix that.
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	60 # See: http://bugs.python.org/issue18378
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	61 import locale
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	62 try:
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	63 locale.getdefaultlocale()
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	64 except ValueError:
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	65 locale.setlocale(locale.LC_ALL, 'en_US.UTF-8')
0a1736ef740d chef: Work around a bug in MacOSX where the default locale doesn't work. Ludovic Chabant <ludovic@chabant.com> parents: 103 diff changeset	66
384 d241585412ad internal: Make it possible to pass `argv` to the main Chef function. Ludovic Chabant <ludovic@chabant.com> parents: 371 diff changeset	67 argv = sys.argv[1:]
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	68 pre_args = _pre_parse_chef_args(argv)
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	69 try:
384 d241585412ad internal: Make it possible to pass `argv` to the main Chef function. Ludovic Chabant <ludovic@chabant.com> parents: 371 diff changeset	70 exit_code = _run_chef(pre_args, argv)
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	71 except Exception as ex:
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	72 if pre_args.debug:
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	73 logger.exception(ex)
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	74 else:
39 2f717f961996 Better error reporting and cache validation. Ludovic Chabant <ludovic@chabant.com> parents: 38 diff changeset	75 log_friendly_exception(logger, ex)
98 5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	76 exit_code = 1
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	77 sys.exit(exit_code)
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	78
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	79
583 1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	80 def _setup_main_parser_arguments(parser):
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	81 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	82 '--version',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	83 action='version',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	84 version=('%(prog)s ' + APP_VERSION))
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	85 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	86 '--root',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	87 help="The root directory of the website.")
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	88 parser.add_argument(
663 3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	89 '--theme',
3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	90 action='store_true',
3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	91 help="Makes the current command apply to a theme website.")
3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	92 parser.add_argument(
583 1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	93 '--config',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	94 dest='config_variant',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	95 help="The configuration variant to use for this command.")
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	96 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	97 '--config-set',
632 9dd2f68f243b chef: Fix the `--config-set` option. Ludovic Chabant <ludovic@chabant.com> parents: 612 diff changeset	98 nargs=2,
9dd2f68f243b chef: Fix the `--config-set` option. Ludovic Chabant <ludovic@chabant.com> parents: 612 diff changeset	99 metavar=('NAME', 'VALUE'),
9dd2f68f243b chef: Fix the `--config-set` option. Ludovic Chabant <ludovic@chabant.com> parents: 612 diff changeset	100 action='append',
583 1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	101 dest='config_values',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	102 help="Sets a specific site configuration setting.")
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	103 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	104 '--debug',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	105 help="Show debug information.", action='store_true')
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	106 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	107 '--debug-only',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	108 nargs='*',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	109 help="Only show debug information for the given categories.")
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	110 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	111 '--no-cache',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	112 help="When applicable, disable caching.",
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	113 action='store_true')
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	114 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	115 '--quiet',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	116 help="Print only important information.",
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	117 action='store_true')
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	118 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	119 '--log',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	120 dest='log_file',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	121 help="Send log messages to the specified file.")
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	122 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	123 '--log-debug',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	124 help="Log debug messages to the log file.",
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	125 action='store_true')
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	126 parser.add_argument(
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	127 '--no-color',
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	128 help="Don't use colorized output.",
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	129 action='store_true')
612 2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	130 parser.add_argument(
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	131 '--pid-file',
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	132 dest='pid_file',
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	133 help="Write a PID file for the current process.")
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	134
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	135
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	136 """ Kinda hacky, but we want the `serve` command to use a different cache
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	137 so that PieCrust doesn't need to re-render all the pages when going
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	138 between `serve` and `bake` (or, worse, not re-render them all correctly
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	139 and end up serving or baking the wrong version).
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	140 """
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	141 _command_caches = {
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	142 'serve': 'server'}
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	143
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	144
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	145 def _pre_parse_chef_args(argv):
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	146 # We need to parse some arguments before we can build the actual argument
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	147 # parser, because it can affect which plugins will be loaded. Also, log-
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	148 # related arguments must be parsed first because we want to log everything
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	149 # from the beginning.
583 1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	150 parser = argparse.ArgumentParser()
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	151 _setup_main_parser_arguments(parser)
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	152 parser.add_argument('extra_args', nargs=argparse.REMAINDER)
583 1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	153 res, _ = parser.parse_known_args(argv)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	154
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	155 # Setup the logger.
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	156 if res.debug and res.quiet:
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	157 raise Exception("You can't specify both --debug and --quiet.")
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	158
579 42a5e78e782a cli: Add `--no-color` option. Ludovic Chabant <ludovic@chabant.com> parents: 564 diff changeset	159 strip_colors = None
42a5e78e782a cli: Add `--no-color` option. Ludovic Chabant <ludovic@chabant.com> parents: 564 diff changeset	160 if res.no_color:
42a5e78e782a cli: Add `--no-color` option. Ludovic Chabant <ludovic@chabant.com> parents: 564 diff changeset	161 strip_colors = True
42a5e78e782a cli: Add `--no-color` option. Ludovic Chabant <ludovic@chabant.com> parents: 564 diff changeset	162
42a5e78e782a cli: Add `--no-color` option. Ludovic Chabant <ludovic@chabant.com> parents: 564 diff changeset	163 colorama.init(strip=strip_colors)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	164 root_logger = logging.getLogger()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	165 root_logger.setLevel(logging.INFO)
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	166 if res.debug or res.log_debug:
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	167 root_logger.setLevel(logging.DEBUG)
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	168
583 1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	169 if res.debug_only:
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	170 for n in res.debug_only:
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	171 logging.getLogger(n).setLevel(logging.DEBUG)
564 353a0b30f412 chef: Add `--debug-only` option to only show debug logging for a given logger. Ludovic Chabant <ludovic@chabant.com> parents: 489 diff changeset	172
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	173 log_handler = logging.StreamHandler(sys.stdout)
564 353a0b30f412 chef: Add `--debug-only` option to only show debug logging for a given logger. Ludovic Chabant <ludovic@chabant.com> parents: 489 diff changeset	174 if res.debug or res.debug_only:
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	175 log_handler.setLevel(logging.DEBUG)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	176 log_handler.setFormatter(ColoredFormatter("[%(name)s] %(message)s"))
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	177 else:
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	178 if res.quiet:
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	179 log_handler.setLevel(logging.WARNING)
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	180 else:
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	181 log_handler.setLevel(logging.INFO)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	182 log_handler.setFormatter(ColoredFormatter("%(message)s"))
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	183 root_logger.addHandler(log_handler)
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	184
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	185 if res.log_file:
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	186 file_handler = logging.FileHandler(res.log_file, mode='w')
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	187 root_logger.addHandler(file_handler)
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	188 if res.log_debug:
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	189 file_handler.setLevel(logging.DEBUG)
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	190
612 2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	191 # PID file.
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	192 if res.pid_file:
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	193 try:
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	194 pid_file_dir = os.path.dirname(res.pid_file)
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	195 if pid_file_dir and not os.path.isdir(pid_file_dir):
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	196 os.makedirs(pid_file_dir)
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	197
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	198 with open(res.pid_file, 'w') as fp:
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	199 fp.write(str(os.getpid()))
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	200 except OSError as ex:
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	201 raise Exception("Can't write PID file: %s" % res.pid_file) from ex
2edaefcb82cd chef: Add `--pid-file` option. Ludovic Chabant <ludovic@chabant.com> parents: 583 diff changeset	202
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	203 return res
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	204
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	205
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	206 def _build_cache_key(pre_args):
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	207 cache_key_str = 'default'
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	208 if pre_args.extra_args:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	209 cmd_name = pre_args.extra_args[0]
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	210 if cmd_name in _command_caches:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	211 cache_key_str = _command_caches[cmd_name]
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	212 if pre_args.config_variant is not None:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	213 cache_key_str += ',variant=%s' % pre_args.config_variant
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	214 if pre_args.config_values:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	215 for name, value in pre_args.config_values:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	216 cache_key_str += ',%s=%s' % (name, value)
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	217
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	218 logger.debug("Using cache key: %s" % cache_key_str)
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	219 cache_key = hashlib.md5(cache_key_str.encode('utf8')).hexdigest()
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	220 return cache_key
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	221
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	222
384 d241585412ad internal: Make it possible to pass `argv` to the main Chef function. Ludovic Chabant <ludovic@chabant.com> parents: 371 diff changeset	223 def _run_chef(pre_args, argv):
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	224 # Setup the app.
489 186a29f61ddc internal: Fix timing info. Ludovic Chabant <ludovic@chabant.com> parents: 467 diff changeset	225 start_time = time.perf_counter()
583 1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	226 root = None
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	227 if pre_args.root:
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	228 root = os.path.expanduser(pre_args.root)
1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	229 else:
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	230 try:
663 3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	231 root = find_app_root(theme=pre_args.theme)
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	232 except SiteNotFoundError:
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	233 root = None
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	234
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	235 # Can't apply custom configuration stuff if there's no website.
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	236 if (pre_args.config_variant or pre_args.config_values) and not root:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	237 raise SiteNotFoundError(
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	238 "Can't apply any configuration variant or value overrides, "
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	239 "there is no website here.")
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	240
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	241 if root:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	242 cache_key = None
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	243 if not pre_args.no_cache:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	244 cache_key = _build_cache_key(pre_args)
663 3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	245 app = PieCrust(
3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	246 root,
3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	247 theme_site=pre_args.theme,
3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	248 cache=(not pre_args.no_cache),
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	249 cache_key=cache_key,
663 3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 632 diff changeset	250 debug=pre_args.debug)
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	251 apply_variant_and_values(
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	252 app, pre_args.config_variant, pre_args.config_values)
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	253 else:
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	254 app = NullPieCrust(
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	255 theme_site=pre_args.theme)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	256
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	257 # Setup the arg parser.
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	258 parser = argparse.ArgumentParser(
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	259 prog='chef',
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	260 description="The PieCrust chef manages your website.",
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	261 formatter_class=argparse.RawDescriptionHelpFormatter)
583 1eda551ee681 cli: More proper argument parsing for the main/root arguments. Ludovic Chabant <ludovic@chabant.com> parents: 579 diff changeset	262 _setup_main_parser_arguments(parser)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	263
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	264 commands = sorted(app.plugin_loader.getCommands(),
341 bd726306d4dc chef: Add a `--config-set` option to set ad-hoc site configuration settings. Ludovic Chabant <ludovic@chabant.com> parents: 307 diff changeset	265 key=lambda c: c.name)
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	266 subparsers = parser.add_subparsers(title='list of commands')
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	267 for c in commands:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	268 p = subparsers.add_parser(c.name, help=c.description)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	269 c.setupParser(p, app)
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	270 p.set_defaults(func=c.checkedRun)
371 c2ca72fb7f0b caching: Use separate caches for config variants and other contexts. Ludovic Chabant <ludovic@chabant.com> parents: 342 diff changeset	271 p.set_defaults(cache_name=c.cache_name)
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	272
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	273 help_cmd = next(filter(lambda c: c.name == 'help', commands), None)
c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	274 if help_cmd and help_cmd.has_topics:
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	275 with io.StringIO() as epilog:
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	276 epilog.write("additional help topics:\n")
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	277 for name, desc in help_cmd.getTopics():
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	278 print_help_item(epilog, name, desc)
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	279 parser.epilog = epilog.getvalue()
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	280
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	281 # Parse the command line.
384 d241585412ad internal: Make it possible to pass `argv` to the main Chef function. Ludovic Chabant <ludovic@chabant.com> parents: 371 diff changeset	282 result = parser.parse_args(argv)
341 bd726306d4dc chef: Add a `--config-set` option to set ad-hoc site configuration settings. Ludovic Chabant <ludovic@chabant.com> parents: 307 diff changeset	283 logger.debug(format_timed(start_time, 'initialized PieCrust',
bd726306d4dc chef: Add a `--config-set` option to set ad-hoc site configuration settings. Ludovic Chabant <ludovic@chabant.com> parents: 307 diff changeset	284 colored=False))
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	285
61 64f37c4cce68 Print the help by default when running `chef` with no command. Ludovic Chabant <ludovic@chabant.com> parents: 57 diff changeset	286 # Print the help if no command was specified.
64f37c4cce68 Print the help by default when running `chef` with no command. Ludovic Chabant <ludovic@chabant.com> parents: 57 diff changeset	287 if not hasattr(result, 'func'):
64f37c4cce68 Print the help by default when running `chef` with no command. Ludovic Chabant <ludovic@chabant.com> parents: 57 diff changeset	288 parser.print_help()
64f37c4cce68 Print the help by default when running `chef` with no command. Ludovic Chabant <ludovic@chabant.com> parents: 57 diff changeset	289 return 0
64f37c4cce68 Print the help by default when running `chef` with no command. Ludovic Chabant <ludovic@chabant.com> parents: 57 diff changeset	290
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	291 # Run the command!
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	292 ctx = CommandContext(app, parser, result)
466 456db44dcc53 bake: Pass the config variants and values from the CLI to the baker. Ludovic Chabant <ludovic@chabant.com> parents: 384 diff changeset	293 ctx.config_variant = pre_args.config_variant
456db44dcc53 bake: Pass the config variants and values from the CLI to the baker. Ludovic Chabant <ludovic@chabant.com> parents: 384 diff changeset	294 ctx.config_values = pre_args.config_values
456db44dcc53 bake: Pass the config variants and values from the CLI to the baker. Ludovic Chabant <ludovic@chabant.com> parents: 384 diff changeset	295
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	296 exit_code = result.func(ctx)
98 5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	297 if exit_code is None:
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	298 return 0
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	299 if not isinstance(exit_code, int):
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	300 logger.error("Got non-integer exit code: %s" % exit_code)
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	301 return -1
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	302 return exit_code
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	303

Mercurial > piecrust2

annotate piecrust/main.py @ 683:ec384174b8b2