Skip to content

Commit 1248d64

Browse files
author
Ke Wu
committed
add types
1 parent 6fcd836 commit 1248d64

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/qpydao/database_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ class DatabaseClient:
3636
"""Database client, both synchronous and asynchronous."""
3737

3838
def __init__(self, config: DatabaseConfig = None):
39-
self.config = config
39+
self.config:DatabaseConfig = config
4040
self._engine = None
4141
self._async_engine = None
4242

@@ -249,7 +249,7 @@ def get_db(self, db_name: str = "default") -> DatabaseClient:
249249

250250

251251
databases: Databases = Databases()
252-
db = databases.default_client()
252+
db:DatabaseConfig = databases.default_client()
253253

254254
def init_database(database: DatabaseClient, schema_name: str = ""):
255255
"""Init postgresql database

src/qpydao/models.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@ def check_db_url(self):
4141
_validate_param("port", self.host)
4242
_validate_param("database", self.host)
4343

44-
4544
class SqlRequestModel(BaseModel):
4645
"""SqlRequestModel: Model for SQLRequest."""
4746

0 commit comments

Comments
 (0)