initializedb.py 1.1KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import sys
  4. import transaction
  5. from pyramid.paster import (
  6. get_appsettings,
  7. setup_logging,
  8. )
  9. from pyramid.scripts.common import parse_vars
  10. from ..models.meta import DeclarativeBase
  11. from ..models import (
  12. get_engine,
  13. get_session_factory,
  14. get_tm_session,
  15. )
  16. def usage(argv):
  17. cmd = os.path.basename(argv[0])
  18. print('usage: %s <config_uri> [var=value]\n'
  19. '(example: "%s development.ini")' % (cmd, cmd))
  20. sys.exit(1)
  21. def main(argv=sys.argv):
  22. if len(argv) < 2:
  23. usage(argv)
  24. config_uri = argv[1]
  25. options = parse_vars(argv[2:])
  26. setup_logging(config_uri)
  27. settings = get_appsettings(config_uri, options=options)
  28. engine = get_engine(settings)
  29. DeclarativeBase.metadata.create_all(engine)
  30. session_factory = get_session_factory(engine)
  31. # TODO - G.M - 28-03-2018 - [Cleanup] Remove code related to example
  32. with transaction.manager:
  33. pass
  34. # dbsession = get_tm_session(session_factory, transaction.manager)
  35. # model = MyModel(name='one', value=1)
  36. # dbsession.add(model)
  37. # Add global manager data, just for test