evilchili/tabletop-frog
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

default attributes on db object are now queries

Browse Source
This commit is contained in:
evilchili 2024-04-20 23:40:17 -07:00
parent 12c643c542
commit 5ec27e9344
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/ttfrog/db/manager.py
View File

@ -85,10 +85,7 @@ class SQLDatabaseManager:
return json.dumps(results, indent=2, cls=AlchemyEncoder) return json.dumps(results, indent=2, cls=AlchemyEncoder)
def __getattr__(self, name: str): def __getattr__(self, name: str):
try: return self.query(getattr(ttfrog.db.schema, name))
return self.tables[name]
except KeyError:
raise AttributeError(f"{self} does not contain the attribute '{name}'.")
db = SQLDatabaseManager() db = SQLDatabaseManager()

2
test/test_schema.py
View File

@ -8,7 +8,7 @@ def test_manage_character(db, classes_factory, ancestries_factory):
# load the fixtures so they are bound to the current session # load the fixtures so they are bound to the current session
classes = classes_factory() classes = classes_factory()
ancestries = ancestries_factory() ancestries = ancestries_factory()
darkvision = db.session.query(schema.AncestryTrait).filter_by(name="Darkvision")[0] darkvision = db.AncestryTrait.filter_by(name="Darkvision")[0]
# create a human character (the default) # create a human character (the default)
char = schema.Character(name="Test Character") char = schema.Character(name="Test Character")