Optional create_objects keyword argument for install and initialize functions

Author: pylover

tests/test_cli.py

@@ -1,6 +1,7 @@
 from bddcli import Given, Application as CLIApplication, status, stderr, \
     when, stdout
 import easycli
+from pony.orm import PrimaryKey, Required
 from yhttp import Application
 from yhttp.ext.pony import install
 
@@ -20,16 +21,21 @@ def __call__(self, args):
 install(app, cliarguments=[Bar])
 
 
+class Foo(app.db.Entity):
+    id = PrimaryKey(int, auto=True)
+    title = Required(str)
+
+
 def test_applicationcli():
     cliapp = CLIApplication('example', 'tests.test_cli:app.climain')
-    with Given(cliapp, 'db --help'):
+    with Given(cliapp, 'db'):
         assert stderr == ''
         assert status == 0
 
         when('db drop')
         when('db create')
-        assert status == 0
         assert stderr == ''
+        assert status == 0
 
         when('db drop')
         assert status == 0
@@ -40,3 +46,12 @@ def test_applicationcli():
         assert status == 0
         assert stderr == ''
         assert stdout == 'bar\n'
+
+        when('db c')
+        assert status == 0
+        assert stderr == ''
+        assert stdout == '''Following objects has been created successfully:
+S foo_id_seq
+r foo
+i foo_pkey
+'''

tests/test_extension.py

@@ -11,18 +11,20 @@ def test_extension(app, Given, freshdb):
       db:
         url: {freshdb}
     ''')
-    dbsession = install(app)
+
+    dbsession = install(app, create_objects=True)
 
     class Foo(app.db.Entity):
         id = PrimaryKey(int, auto=True)
         title = Required(str)
 
+    app.ready()
+
     @dbsession
     def mockup():
         Foo(title='foo 1')
         Foo(title='foo 2')
 
-    app.ready()
     mockup()
 
     @app.route()

yhttp/ext/pony/__init__.py

@@ -5,4 +5,4 @@
 from .orm import initialize, deinitialize
 
 
-__version__ = '2.3.1'
+__version__ = '2.4.0'

yhttp/ext/pony/cli.py

@@ -2,8 +2,9 @@
 import getpass
 
 from easycli import SubCommand, Argument
+from pony.orm import db_session
 
-from . import dbmanager, uri
+from . import dbmanager, uri, orm
 
 
 getdbpass = functools.partial(getpass.getpass, 'Enter db password: ')
@@ -47,6 +48,25 @@ def __call__(self, args):
         self.getdbmanager(args) \
             .create(info['database'], owner=info.get('user'))
 
+        self.create_objects(args)
+        self.report_objects(args)
+
+    def create_objects(self, args):
+        app = args.application
+        orm.initialize(app.db, app.settings.db.url, create_objects=True)
+
+    @db_session
+    def report_objects(self, args):
+        app = args.application
+        result = app.db.execute('''
+            SELECT relname, relkind
+            FROM pg_class
+            WHERE relname !~ '^(pg|sql)_' AND relkind != 'v';
+        ''')
+        print('Following objects has been created successfully:')
+        for name, kind in result.fetchall():
+            print(kind, name)
+
 
 class DropDatabase(DatabaseAdministrativeCommand):
     __command__ = 'drop'

yhttp/ext/pony/install.py

@@ -31,7 +31,7 @@ def wrapper(*a, **kw):
     return outter
 
 
-def install(app, db=None, cliarguments=None):
+def install(app, db=None, cliarguments=None, create_objects=False):
     app.cliarguments.append(DatabaseCLI)
     if cliarguments:
         DatabaseCLI.__arguments__.extend(cliarguments)
@@ -49,7 +49,7 @@ def ready(app):
                 'postgres://:@/dbname'
             )
 
-        orm.initialize(db, app.settings.db.url)
+        orm.initialize(db, app.settings.db.url, create_objects=create_objects)
 
     @app.when
     def shutdown(app):

yhttp/ext/pony/orm.py

@@ -1,10 +1,13 @@
 from . import uri
 
 
-def initialize(db, url):
+def initialize(db, url, create_objects=False):
     url = uri.parse(url)
     db.bind(**url)
-    db.generate_mapping(create_tables=True)
+    db.generate_mapping(
+        check_tables=create_objects,
+        create_tables=create_objects
+    )
 
 
 def deinitialize(db):