initializedb.py 1.2KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  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. from ..models import MyModel
  17. def usage(argv):
  18. cmd = os.path.basename(argv[0])
  19. print('usage: %s <config_uri> [var=value]\n'
  20. '(example: "%s development.ini")' % (cmd, cmd))
  21. sys.exit(1)
  22. def main(argv=sys.argv):
  23. if len(argv) < 2:
  24. usage(argv)
  25. config_uri = argv[1]
  26. options = parse_vars(argv[2:])
  27. setup_logging(config_uri)
  28. settings = get_appsettings(config_uri, options=options)
  29. engine = get_engine(settings)
  30. DeclarativeBase.metadata.create_all(engine)
  31. session_factory = get_session_factory(engine)
  32. # TODO - G.M - 28-03-2018 - [Cleanup] Remove code related to example
  33. with transaction.manager:
  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