pch_xyz/relay
1
0
Fork 0
mirror of https://git.pleroma.social/pleroma/relay.git synced 2025-04-20 01:26:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

raise error on invalid database backend

Browse source
This commit is contained in:
Izalia Mae 2025-04-19 11:09:26 -04:00
parent 5870f3a96b
commit 6e5f0f237a
1 changed files with 15 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
relay/database/__init__.py
View file

@ -30,18 +30,22 @@ def get_database(state: State, migrate: bool = True) -> Database[Connection]:
db: Database[Connection]
if state.config.db_type == "sqlite":
db = Database.sqlite(state.config.sqlite_path, **options)
match state.config.db_type:
case "sqlite" | "sqlite3":
db = Database.sqlite(state.config.sqlite_path, **options)
elif state.config.db_type == "postgres":
db = Database.postgresql(
state.config.pg_name,
state.config.pg_host,
state.config.pg_port,
state.config.pg_user,
state.config.pg_pass,
**options
)
case "postgres" | "postgresql":
db = Database.postgresql(
state.config.pg_name,
state.config.pg_host,
state.config.pg_port,
state.config.pg_user,
state.config.pg_pass,
**options
)
case _:
raise RuntimeError(f"Invalid database backend: {state.config.db_type}")
db.load_prepared_statements(File.from_resource("relay", "data/statements.sql"))
db.connect()