| #!/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) |