diff --git a/IzzyLib/database/sql.py b/IzzyLib/database/sql.py index 0957a2c..4184c1a 100644 --- a/IzzyLib/database/sql.py +++ b/IzzyLib/database/sql.py @@ -123,11 +123,11 @@ class Session(object): def __init__(self, db): self.closed = False - self.db = db - self.classes = self.db.classes + self.database = db + self.classes = db.classes self.session = sessionmaker(bind=db.db)() - self.table = self.db.table - self.cache = self.db.cache + self.table = db.table + self.cache = db.cache # session aliases self.s = self.session @@ -137,6 +137,9 @@ class Session(object): self.query = self.s.query self.execute = self.s.execute + # remove in the future + self.db = db + self._setup()