pagure/createdb.py

#!/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)
    if not os.path.exists(config):
        print("The file `{0}` could not be found".format(config))
        sys.exit(2)
    os.environ["PAGURE_CONFIG"] = config

if args.alembic_cfg:
    alembic_config = args.alembic_cfg
    if not alembic_config.startswith("/"):
        here = os.path.join(os.path.dirname(os.path.abspath(__file__)))
        alembic_config = os.path.join(here, alembic_config)
    if not alembic_config.endswith("alembic.ini"):
        print("--initial should point to the alembic.ini file to use.")
        sys.exit(1)
    if not os.path.exists(alembic_config):
        print("The file `{0}` could not be found".format(alembic_config))
        sys.exit(2)
    os.environ["ALEMBIC_CONFIG"] = alembic_config

import pagure.config
from pagure.lib import model

_config = pagure.config.reload_config()

model.create_tables(
    _config["DB_URL"],
    _config.get("PATH_ALEMBIC_INI", os.environ["ALEMBIC_CONFIG"]),
    acls=_config.get("ACLS", {}),
    debug=True,
)