#!/usr/bin/env python from __future__ import print_function, unicode_literals, absolute_import import argparse import sys import os parser = argparse.ArgumentParser( description='Create/Update the Pagure database') parser.add_argument( '--config', '-c', dest='config', help='Configuration file to use for pagure.') parser.add_argument( '--initial', '-i', dest='alembic_cfg', help='With this option, the database will be automatically stamped to ' 'the latest version according to alembic. Point to the alembic.ini ' 'file to use.') args = parser.parse_args() if args.config: config = args.config if not config.startswith('/'): here = os.path.join(os.path.dirname(os.path.abspath(__file__))) config = os.path.join(here, config) os.environ['PAGURE_CONFIG'] = config if args.alembic_cfg: if not args.alembic_cfg.endswith('alembic.ini'): print('--initial should point to the alembic.ini file to use.') sys.exit(1) if not os.path.exists(args.alembic_cfg): print('The file `{0}` could not be found'.format(args.alembic_cfg)) sys.exit(2) import pagure.config from pagure.lib import model _config = pagure.config.reload_config() model.create_tables( _config['DB_URL'], _config.get('PATH_ALEMBIC_INI', args.alembic_cfg), acls=_config.get('ACLS', {}), debug=True)