use example imports for tests and eventual demo
This commit is contained in:
+2
-1
@@ -52,6 +52,7 @@ class GrungDB(TinyDB):
|
||||
|
||||
def create_table(self, table_class):
|
||||
name = table_class.__name__
|
||||
print(name)
|
||||
if name not in self._tables:
|
||||
self._tables[name] = RecordTable(self.storage, name, document_class=table_class)
|
||||
return self.table(name)
|
||||
@@ -77,6 +78,6 @@ class GrungDB(TinyDB):
|
||||
def with_schema(cls, schema_module, *args, **kwargs):
|
||||
db = GrungDB(*args, **kwargs)
|
||||
for name, obj in inspect.getmembers(schema_module):
|
||||
if type(obj) == object and issubclass(Record, obj):
|
||||
if type(obj) == type and issubclass(obj, Record):
|
||||
db.create_table(obj)
|
||||
return db
|
||||
|
||||
Reference in New Issue
Block a user