Setup to handle book objects from sqlalchemy.

This commit is contained in:
th3root
2023-03-11 19:00:17 -05:00
parent 1e975405c8
commit b9b34831b6
10 changed files with 118 additions and 23 deletions

View File

@@ -164,3 +164,27 @@ class Storage:
f"Collection {_s} failed: {e}")
_collections.append(_p)
self.config.logger.info("Finished making collections.")
def get_books(self, collection=None):
"""Get books from database.
Parameters
----------
collection : str
Collection to filter by.
Returns
-------
_result : ScalarResult Object
"""
session = Session(self.engine)
if collection:
_result = session.execute(
select(Book).join(Collection).where(
Collection.collection == collection
)
).all()
else:
_result = session.execute(select(Book)).all()
session.close()
return _result