piecrust2: piecrust/main.py annotate

annotate piecrust/main.py @ 219:d7a548ebcd58

serve: Add server sent events for showing pipeline errors in the debug window. The server can now run an endpoint that streams pipeline status, including errors or "fixed" statuses. As a result, I had to investigate using event-loop based server alternatives, before I figured out the correct flag to set in Werkzeug. Support for Gunicorn is therefore now possible, although disabled by default. I will come in handy though when proper support for CMS-mode is enabled.

author	Ludovic Chabant <ludovic@chabant.com>
date	Mon, 02 Feb 2015 08:34:44 -0800
parents	0a1736ef740d
children	869a206facd5

rev	line source
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	1 import io
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	2 import sys
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	3 import time
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	4 import os.path
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	5 import logging
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	6 import argparse
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 import colorama
69 cb1ed436642c Always use version generated by `setup.py`. Better version generation. Ludovic Chabant <ludovic@chabant.com> parents: 61 diff changeset	8 from piecrust import APP_VERSION
cb1ed436642c Always use version generated by `setup.py`. Better version generation. Ludovic Chabant <ludovic@chabant.com> parents: 61 diff changeset	9 from piecrust.app import PieCrust, PieCrustConfiguration
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	10 from piecrust.chefutil import (format_timed, log_friendly_exception,
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	11 print_help_item)
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	12 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	13 from piecrust.environment import StandardEnvironment
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	14 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	15 from piecrust.plugins.base import PluginLoader
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	16
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	17
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	18 logger = logging.getLogger(__name__)
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
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 class ColoredFormatter(logging.Formatter):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22 COLORS = {
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23 '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	24 'INFO': '',
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	25 'WARNING': colorama.Fore.YELLOW,
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	26 'ERROR': colorama.Fore.RED,
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	27 '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	28 }
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	29
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	30 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	31 super(ColoredFormatter, self).__init__(fmt, datefmt)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	32
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	33 def format(self, record):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	34 color = self.COLORS.get(record.levelname)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	35 res = super(ColoredFormatter, self).format(record)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	36 if color:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	37 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	38 return res
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	39
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	40
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	41 class NullPieCrust:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	42 def __init__(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	43 self.root_dir = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	44 self.debug = False
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	45 self.templates_dirs = []
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	46 self.plugins_dirs = []
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	47 self.theme_dir = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	48 self.cache_dir = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	49 self.config = PieCrustConfiguration()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	50 self.plugin_loader = PluginLoader(self)
103 028df35a690e Fix using `chef` outside of a website. Ludovic Chabant <ludovic@chabant.com> parents: 99 diff changeset	51 self.env = None
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	52
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	53
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	54 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	55 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	56 # 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	57 # 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	58 # 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	59 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	60 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	61 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	62 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	63 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	64
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	65 argv = sys.argv
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	66 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	67 try:
98 5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	68 exit_code = _run_chef(pre_args)
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	69 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	70 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	71 logger.exception(ex)
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	72 else:
39 2f717f961996 Better error reporting and cache validation. Ludovic Chabant <ludovic@chabant.com> parents: 38 diff changeset	73 log_friendly_exception(logger, ex)
98 5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	74 exit_code = 1
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	75 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	76
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	77
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	78 class PreParsedChefArgs(object):
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	79 def __init__(self, root=None, cache=True, debug=False, quiet=False,
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	80 log_file=None, log_debug=False, config_variant=None):
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	81 self.root = root
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	82 self.cache = cache
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	83 self.debug = debug
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	84 self.quiet = quiet
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	85 self.log_file = log_file
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	86 self.log_debug = log_debug
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	87 self.config_variant = config_variant
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	88
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	89
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	90 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	91 # 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	92 # 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	93 # 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	94 # from the beginning.
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	95 res = PreParsedChefArgs()
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	96 i = 1
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	97 while i < len(argv):
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	98 arg = argv[i]
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	99 if arg.startswith('--root='):
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	100 res.root = os.path.expanduser(arg[len('--root='):])
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	101 elif arg == '--root':
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	102 res.root = argv[i + 1]
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	103 ++i
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	104 elif arg.startswith('--config='):
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	105 res.config_variant = arg[len('--config='):]
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	106 elif arg == '--config':
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	107 res.config_variant = argv[i + 1]
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	108 ++i
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	109 elif arg == '--log':
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	110 res.log_file = argv[i + 1]
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	111 ++i
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	112 elif arg == '--log-debug':
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	113 res.log_debug = True
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	114 elif arg == '--no-cache':
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	115 res.cache = False
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	116 elif arg == '--debug':
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	117 res.debug = True
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	118 elif arg == '--quiet':
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	119 res.quiet = True
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	120
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	121 if arg[0] != '-':
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	122 break
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	123
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	124 i = i + 1
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	125
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	126 # 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	127 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	128 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	129
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	130 colorama.init()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	131 root_logger = logging.getLogger()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	132 root_logger.setLevel(logging.INFO)
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	133 if res.debug or res.log_debug:
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	134 root_logger.setLevel(logging.DEBUG)
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	135
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	136 log_handler = logging.StreamHandler(sys.stdout)
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	137 if res.debug:
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	138 log_handler.setLevel(logging.DEBUG)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	139 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	140 else:
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	141 if res.quiet:
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	142 log_handler.setLevel(logging.WARNING)
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	143 else:
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	144 log_handler.setLevel(logging.INFO)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	145 log_handler.setFormatter(ColoredFormatter("%(message)s"))
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	146 root_logger.addHandler(log_handler)
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	147
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	148 if res.log_file:
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	149 file_handler = logging.FileHandler(res.log_file, mode='w')
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	150 root_logger.addHandler(file_handler)
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	151 if res.log_debug:
00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	152 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	153
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	154 return res
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	155
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	156
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	157 def _run_chef(pre_args):
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	158 # Setup the app.
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	159 start_time = time.clock()
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	160 root = pre_args.root
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	161 if root is None:
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	162 try:
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	163 root = find_app_root()
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	164 except SiteNotFoundError:
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	165 root = None
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	166
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	167 if not root:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	168 app = NullPieCrust()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	169 else:
44 a62452ab5080 Correctly set the `debug` flag on the app. Ludovic Chabant <ludovic@chabant.com> parents: 39 diff changeset	170 app = PieCrust(root, cache=pre_args.cache, debug=pre_args.debug)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	171
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	172 # Handle a configuration variant.
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	173 if pre_args.config_variant is not None:
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	174 if not root:
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	175 raise SiteNotFoundError("Can't apply any variant.")
091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	176 app.config.applyVariant('variants/' + pre_args.config_variant)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	177
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	178 # Setup the arg parser.
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	179 parser = argparse.ArgumentParser(
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	180 prog='chef',
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	181 description="The PieCrust chef manages your website.",
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	182 formatter_class=argparse.RawDescriptionHelpFormatter)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	183 parser.add_argument('--version', action='version', version=('%(prog)s ' + APP_VERSION))
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	184 parser.add_argument('--root', help="The root directory of the website.")
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	185 parser.add_argument('--config', help="The configuration variant to use for this command.")
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	186 parser.add_argument('--debug', help="Show debug information.", action='store_true')
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	187 parser.add_argument('--no-cache', help="When applicable, disable caching.", action='store_true')
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	188 parser.add_argument('--quiet', help="Print only important information.", action='store_true')
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	189 parser.add_argument('--log', help="Send log messages to the specified file.")
97 00a9b24ca944 Add `--log-debug` option. Ludovic Chabant <ludovic@chabant.com> parents: 69 diff changeset	190 parser.add_argument('--log-debug', help="Log debug messages to the log file.", action='store_true')
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	191
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	192 commands = sorted(app.plugin_loader.getCommands(),
7 343d08ef5668 More PieCrust 3 fixes, and a couple of miscellaneous bug fixes. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	193 key=lambda c: c.name)
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	194 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	195 for c in commands:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	196 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	197 c.setupParser(p, app)
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	198 p.set_defaults(func=c.checkedRun)
99 8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	199
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	200 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	201 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	202 with io.StringIO() as epilog:
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	203 epilog.write("additional help topics:\n")
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	204 for name, desc in help_cmd.getTopics():
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	205 print_help_item(epilog, name, desc)
8703be118430 Changes to `help` command and extendable commands: Ludovic Chabant <ludovic@chabant.com> parents: 98 diff changeset	206 parser.epilog = epilog.getvalue()
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	207
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	208
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	209 # Parse the command line.
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	210 result = parser.parse_args()
38 091f99bfbe44 Fix running `chef` outside of a website. Slightly better error reporting. Ludovic Chabant <ludovic@chabant.com> parents: 7 diff changeset	211 logger.debug(format_timed(start_time, 'initialized PieCrust', colored=False))
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	212
61 64f37c4cce68 Print the help by default when running `chef` with no command. Ludovic Chabant <ludovic@chabant.com> parents: 57 diff changeset	213 # 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	214 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	215 parser.print_help()
64f37c4cce68 Print the help by default when running `chef` with no command. Ludovic Chabant <ludovic@chabant.com> parents: 57 diff changeset	216 return 0
64f37c4cce68 Print the help by default when running `chef` with no command. Ludovic Chabant <ludovic@chabant.com> parents: 57 diff changeset	217
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	218 # Run the command!
57 c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	219 ctx = CommandContext(app, parser, result)
c8c522dacfea Add `help` function, cleanup argument handling. Ludovic Chabant <ludovic@chabant.com> parents: 44 diff changeset	220 exit_code = result.func(ctx)
98 5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	221 if exit_code is None:
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	222 return 0
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	223 if not isinstance(exit_code, int):
5959a117a943 Exit with the proper code. Ludovic Chabant <ludovic@chabant.com> parents: 97 diff changeset	224 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	225 return -1
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	226 return exit_code
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	227

Mercurial > piecrust2

annotate piecrust/main.py @ 219:d7a548ebcd58