From 2a47b24867c0456c29a98344d78b5c8d62a163ba Mon Sep 17 00:00:00 2001 From: RTAkland Date: Sat, 7 Jan 2023 20:56:43 +0800 Subject: [PATCH] =?UTF-8?q?update:=20=E4=BF=AE=E6=94=B9=E4=BA=86=E7=B1=BB?= =?UTF-8?q?=E7=9A=84=E5=91=BD=E5=90=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/config.py | 4 ++-- src/db/db.py | 12 ++++++------ 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/config.py b/src/config.py index 8c4b9de..97ed29f 100644 --- a/src/config.py +++ b/src/config.py @@ -9,9 +9,9 @@ import os class Config: """ - Available database: sqlite3, mysql + Available database: sqlite3, mysql, detaBase sqlite3 -> sqlite3 (default) - mysql -> user:pwd@host:port/db + mysql -> mysql://user:pwd@host:port/db deta -> deta """ database = os.getenv("COUNTER_DB") or "sqlite3" # 数据库类型 diff --git a/src/db/db.py b/src/db/db.py index 53e277d..ac77672 100644 --- a/src/db/db.py +++ b/src/db/db.py @@ -9,15 +9,15 @@ import os from src.config import Config -if Config.database == 'sqlite3': +if Config.database in ["sqlite3", "sqlite"]: import sqlite3 as operator -elif Config.database == 'deta': +elif Config.database == "deta": from deta import Deta else: import pymysql as operator -class BaseSQL: +class SQL: def __init__(self): self.conn = None self.cursor = None @@ -56,17 +56,17 @@ class BaseSQL: return result -class SQLite(BaseSQL): +class SQLite(SQL): def __init__(self): super().__init__() self.conn = operator.connect('./src/db/data.sqlite') self.cursor = self.conn.cursor() -class MySQL(BaseSQL): +class MySQL(SQL): def __init__(self): super().__init__() - _CONFIG = Config.database.split('@') + _CONFIG = Config.database.replace("mysql://", "").split("@") user = _CONFIG[0].split(":")[0] pwd = _CONFIG[0].split(":")[1] host = _CONFIG[1].split(":")[0]