Mercurial > piecrust2
diff piecrust/commands/builtin/admin.py @ 588:b884bef3e611
admin: New `admin` command to manage FoodTruck-related things.
Remove old FoodTruck command and packaging.
author | Ludovic Chabant <ludovic@chabant.com> |
---|---|
date | Sun, 17 Jan 2016 21:44:41 -0800 |
parents | |
children | 79a31a3c947b |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/piecrust/commands/builtin/admin.py Sun Jan 17 21:44:41 2016 -0800 @@ -0,0 +1,84 @@ +import os +import logging +from piecrust.commands.base import ChefCommand + + +logger = logging.getLogger(__name__) + + +class AdministrationPanelCommand(ChefCommand): + def __init__(self): + super(AdministrationPanelCommand, self).__init__() + self.name = 'admin' + self.description = "Manages the PieCrust administration panel." + self.requires_website = False + + def setupParser(self, parser, app): + subparsers = parser.add_subparsers() + + p = subparsers.add_parser( + 'init', + help="Creates a new administration panel website.") + p.set_defaults(sub_func=self._initFoodTruck) + + p = subparsers.add_parser( + 'genpass', + help=("Generates the hashed password for use as an " + "admin password")) + p.add_argument('password', help="The password to hash.") + p.set_defaults(sub_func=self._generatePassword) + + p = subparsers.add_parser( + 'run', + help="Runs the administrative panel website.") + p.set_defaults(sub_func=self._runFoodTruck) + + def checkedRun(self, ctx): + if not hasattr(ctx.args, 'sub_func'): + return self._runFoodTruck(ctx) + return ctx.args.sub_func(ctx) + + def _runFoodTruck(self, ctx): + from foodtruck.main import run_foodtruck + run_foodtruck(debug=ctx.args.debug) + + def _initFoodTruck(self, ctx): + import getpass + import bcrypt + + secret_key = os.urandom(22) + admin_username = input("Admin username (admin): ") or 'admin' + admin_password = getpass.getpass("Admin password: ") + if not admin_password: + logger.warning("No administration password set!") + logger.warning("Don't make this instance of FoodTruck public.") + logger.info("You can later set an admin password by editing " + "the `foodtruck.yml` file and using the " + "`chef admin genpass` command.") + else: + binpw = admin_password.encode('utf8') + hashpw = bcrypt.hashpw(binpw, bcrypt.gensalt()).decode('utf8') + admin_password = hashpw + + ft_config = """ +foodtruck: + secret_key: %(secret_key)s +security: + username: %(username)s + # You can generate another hashed password with `chef admin genpass`. + password: %(password)s +""" + ft_config = ft_config % { + 'secret_key': secret_key, + 'username': admin_username, + 'password': admin_password + } + with open('foodtruck.yml', 'w', encoding='utf8') as fp: + fp.write(ft_config) + + def _generatePassword(self, ctx): + import bcrypt + binpw = ctx.args.password.encode('utf8') + hashpw = bcrypt.hashpw(binpw, bcrypt.gensalt()).decode('utf8') + logger.info(hashpw) +