barker/barker/barker/scripts/initializedb.py

68 lines
1.2 KiB
Python

import os
import sys
from pyramid.paster import (
get_appsettings,
setup_logging,
)
from pyramid.scripts.common import parse_vars
from barker.models.meta import Base
from barker.models import (
get_engine,
get_session_factory,
get_tm_session,
)
from barker.models import (
Client,
Customer,
DbSetting,
Device,
FoodTable,
GuestBook,
Inventory,
InventoryModifier,
Kot,
LoginHistory,
MenuCategory,
Modifier,
ModifierCategory,
Overview,
Permission,
Printer,
Product,
Reprint,
Role,
role_permissions,
SaleCategory,
Section,
SectionPrinter,
SettleOption,
Settlement,
Tax,
User,
user_roles,
Voucher,
VoucherType,
)
def usage(argv):
cmd = os.path.basename(argv[0])
print("usage: %s <config_uri> [var=value]\n" '(example: "%s development.ini")' % (cmd, cmd))
sys.exit(1)
def main(argv=sys.argv):
if len(argv) < 2:
usage(argv)
config_uri = argv[1]
options = parse_vars(argv[2:])
setup_logging(config_uri)
settings = get_appsettings(config_uri, options=options)
engine = get_engine(settings)
Base.metadata.create_all(engine)