use example imports for tests and eventual demo

This commit is contained in:
evilchili
2025-09-24 21:34:32 -07:00
parent 235d26ff1e
commit e5c0d66bde
3 changed files with 7 additions and 19 deletions
+2 -1
View File
@@ -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