From 840ffd8a4ae12c8f8ab964aa1997bf8e15b22bfb Mon Sep 17 00:00:00 2001 From: 100gle Date: Mon, 3 Jul 2023 21:41:58 +0800 Subject: [PATCH] refactor: clean left `os.path` code --- backend/src/module/database/orm/connector.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/backend/src/module/database/orm/connector.py b/backend/src/module/database/orm/connector.py index 05be0df4..ef6e972a 100644 --- a/backend/src/module/database/orm/connector.py +++ b/backend/src/module/database/orm/connector.py @@ -1,4 +1,5 @@ -import os +from os import PathLike +from pathlib import Path import sqlite3 from .delete import Delete @@ -10,10 +11,13 @@ from module.conf import DATA_PATH class Connector: - def __init__(self, table_name: str, data: dict, database: str = DATA_PATH): + def __init__( + self, table_name: str, data: dict, database: PathLike[str] | Path = DATA_PATH + ): # Create folder if not exists - if not os.path.exists(os.path.dirname(DATA_PATH)): - os.makedirs(os.path.dirname(DATA_PATH)) + if isinstance(database, PathLike): + database = Path(database) + database.parent.mkdir(parents=True, exist_ok=True) self._conn = sqlite3.connect(database) self._cursor = self._conn.cursor()