StorageFactory: set default storage class as class variable
This commit is contained in:
parent
318a152e52
commit
dcbf231038
1 changed files with 5 additions and 2 deletions
|
@ -79,11 +79,14 @@ class StorageFactory(object):
|
||||||
IdType = Integer
|
IdType = Integer
|
||||||
JsonType = JSON
|
JsonType = JSON
|
||||||
|
|
||||||
def __init__(self, config, storageClass=Storage):
|
storageClass = Storage
|
||||||
|
|
||||||
|
def __init__(self, config, storageClass=None):
|
||||||
self.engine = self.getEngine(config.dbengine, config.dbname,
|
self.engine = self.getEngine(config.dbengine, config.dbname,
|
||||||
config.dbuser, config.dbpassword)
|
config.dbuser, config.dbpassword)
|
||||||
self.Session = self.sessionFactory()
|
self.Session = self.sessionFactory()
|
||||||
self.storageClass = storageClass
|
if storageClass is not None:
|
||||||
|
self.storageClass = storageClass
|
||||||
|
|
||||||
def __call__(self, schema=None):
|
def __call__(self, schema=None):
|
||||||
return self.storageClass(self, schema=schema)
|
return self.storageClass(self, schema=schema)
|
||||||
|
|
Loading…
Add table
Reference in a new issue