From 5e4cfc891060e576469a71c642f8c06a6eefe1bc Mon Sep 17 00:00:00 2001 From: MarkusJoe Date: Sun, 18 Sep 2022 09:19:39 +0800 Subject: [PATCH] =?UTF-8?q?update:=20=E6=9B=B4=E6=96=B0=E4=BA=86=E4=B8=8D?= =?UTF-8?q?=E5=90=8C=E6=95=B0=E6=8D=AE=E5=BA=93=E7=9A=84sql=E8=AF=AD?= =?UTF-8?q?=E5=8F=A5=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/db/db.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/db/db.py b/src/db/db.py index e70e690..500d91f 100644 --- a/src/db/db.py +++ b/src/db/db.py @@ -14,6 +14,7 @@ class DatabaseSQL: def __init__(self): self.conn = None self.cursor = None + self.target = None def __del__(self): self.conn.commit() @@ -30,7 +31,7 @@ class DatabaseSQL: return result def insert(self, _id: str) -> bool: - self.cursor.execute('insert into data (id, times) values ("%s", 1);' % _id) + self.cursor.execute('insert into data (id, times) values (%s, 1);' % self.target % _id) return True def update(self, _id: str, times: int) -> bool: @@ -43,8 +44,8 @@ class DatabaseSQL: result = self.cursor.fetchall() return result - def query_image(self, _id: str) -> list: - self.cursor.execute('select * from image where id="%s";' % _id) + def query_image(self, theme: str) -> list: + self.cursor.execute('select * from %s;' % self.target % theme) result = self.cursor.fetchall() return result @@ -54,6 +55,8 @@ class SQLite(DatabaseSQL): super().__init__() self.conn = operator.connect('./src/db/data.sqlite') self.cursor = self.conn.cursor() + self.target = '"%s"' + print(self.target) class MySQL(DatabaseSQL): @@ -71,3 +74,4 @@ class MySQL(DatabaseSQL): database=db ) self.cursor = self.conn.cursor() + self.target = 'Counter.%s'