Skip to content

Commit fb346de

Browse files
committed
refactor: database context manager dependency injected in populate function
1 parent 5cd7291 commit fb346de

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/infra/database/populate.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,16 @@
11
import os
2-
from src.infra.database.database import PgDatabase
2+
from src.infra.database.database import PgDatabase, Database
33

44
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
55

66

7-
def populate() -> None:
7+
def populate(database: Database) -> None:
88
def insert(filePath: str) -> None:
99
sql_file = open(BASE_DIR + filePath, "r")
1010
sql = sql_file.read()
1111
lines = sql.split(";")
1212

13-
with PgDatabase() as db:
13+
with database as db:
1414
for line in lines:
1515
print(line)
1616
if line.strip():
@@ -23,4 +23,4 @@ def insert(filePath: str) -> None:
2323

2424

2525
if __name__ == "__main__":
26-
populate()
26+
populate(PgDatabase())

0 commit comments

Comments
 (0)