Fix unknown options for psycopg2 startup
Some checks failed
ci/woodpecker/push/check Pipeline failed
Some checks failed
ci/woodpecker/push/check Pipeline failed
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
import os
|
||||
from typing import Annotated, Self, cast
|
||||
from typing import Annotated, Any, Self, cast
|
||||
import logging
|
||||
|
||||
from pydantic import BaseModel, Field, model_validator
|
||||
@@ -53,6 +53,9 @@ class _DatabaseConfig(BaseModel):
|
||||
# The file to read the database URI from
|
||||
uri_file: str | None = Field(default=None)
|
||||
|
||||
# sqlalchemy options for the database connection
|
||||
connect_args: dict[str, Any] = Field(default_factory=dict)
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_secret(self) -> Self:
|
||||
if self.uri_plain is None and self.uri_file is None:
|
||||
|
||||
@@ -12,9 +12,7 @@ from xmpp_api.db.bot import Bot, AllowedJid
|
||||
def get_engine(config: ConfigDep) -> Engine:
|
||||
return create_engine(
|
||||
config.database.uri,
|
||||
connect_args={
|
||||
"check_same_thread": False,
|
||||
},
|
||||
connect_args=config.database.connect_args,
|
||||
)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user