tonym/spoolman2
1
0
Fork 0
Code Issues 19 Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Improved compatibility with postgres

Browse Source
This commit is contained in:
Donkie
2023-05-09 20:03:19 +02:00
parent d8fc5fc37b
commit ed3e30ea75
3 changed files with 50 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
spoolman/database/database.py
View File

@@ -20,7 +20,13 @@ class Database:
def connect(self: "Database") -> None:
"""Connect to the database."""
self.engine = create_async_engine(self.connection_url, connect_args={"check_same_thread": False})
connect_args = {}
if self.connection_url.drivername == "sqlite":
connect_args["check_same_thread"] = False
elif self.connection_url.drivername == "postgresql":
connect_args["options"] = "-c timezone=utc"
self.engine = create_async_engine(self.connection_url, connect_args=connect_args)
self.session_maker = async_sessionmaker(self.engine, autocommit=False, autoflush=True)
async def create_tables(self: "Database") -> None: