From 4a21ca730ff3a142f69fb2d524fa3377cdb94586 Mon Sep 17 00:00:00 2001
From: MarkusJoe <markushammered@gmail.com>
Date: Sun, 2 Oct 2022 17:30:12 +0800
Subject: [PATCH] =?UTF-8?q?update:=20=E6=9B=B4=E6=96=B0=E4=BA=86=E5=AF=B9R?=
 =?UTF-8?q?edis=E6=95=B0=E6=8D=AE=E5=BA=93=E7=9A=84=E6=94=AF=E6=8C=81,=20?=
 =?UTF-8?q?=E5=B9=B6=E5=8F=96=E6=B6=88=E4=BA=86=E5=AF=B9MySQL=E7=9A=84?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/db/db.py | 78 ++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 58 insertions(+), 20 deletions(-)

diff --git a/src/db/db.py b/src/db/db.py
index b881e3d..8a7dc6d 100644
--- a/src/db/db.py
+++ b/src/db/db.py
@@ -10,11 +10,10 @@ import os
 from src.db import operator
 
 
-class DatabaseSQL:
+class Base:
     def __init__(self):
         self.conn = None
         self.cursor = None
-        self.target = None
 
     def __del__(self):
         self.conn.commit()
@@ -31,7 +30,7 @@ class DatabaseSQL:
         return result
 
     def insert(self, _id: str) -> bool:
-        self.cursor.execute('insert into data (id, times) values (%s, 1);' % self.target % _id)
+        self.cursor.execute('insert into data (id, times) values (%s, 1);' % _id)
         return True
 
     def update(self, _id: str, times: int) -> bool:
@@ -45,32 +44,71 @@ class DatabaseSQL:
         return result
 
     def query_image(self, theme: str) -> list:
-        self.cursor.execute('select * from %s;' % self.target % theme)
+        self.cursor.execute('select * from %s;' % theme)
         result = self.cursor.fetchall()
         return result
 
 
-class SQLite(DatabaseSQL):
+class SQLite(Base):
     def __init__(self):
         super().__init__()
         self.conn = operator.connect('./src/db/data.sqlite')
         self.cursor = self.conn.cursor()
-        self.target = '"%s"'
 
 
-class MySQL(DatabaseSQL):
+class Redis(Base):
     def __init__(self):
         super().__init__()
-        user = os.getenv('m_user')
-        pwd = os.getenv('m_password')
-        host = os.getenv('m_host')
-        port = int(os.getenv('m_port'))
-        db = os.getenv('m_database')
-        self.conn = operator.connect(host=host,
-                                     user=user,
-                                     password=pwd,
-                                     port=port,
-                                     database=db
-                                     )
-        self.cursor = self.conn.cursor()
-        self.target = 'Counter.%s'
+        self.host: str = os.getenv('c_host')
+        self.port: int = int(os.getenv('c_port'))
+        self.password: str = os.getenv('c_password')  # if is None then it is None. lol
+        self.conn = operator.Redis(host=self.host,
+                                   port=self.port,
+                                   password=self.password,
+                                   decode_responses=True)
+
+    def __del__(self):
+        self.conn.close()
+
+    def query(self, _id: str) -> tuple:
+        return self._get(_id)
+
+    def insert(self, _id: str) -> bool:
+        return self._set(_id)
+
+    def update(self, _id: str, times: int) -> bool:
+        return self._set(_id, times)
+
+    def query_all(self) -> list:
+        return self._get_all()
+
+    def query_image(self, theme: str) -> list:
+        return self._get_image(theme)
+
+    def _get(self, _id: str) -> tuple:
+        result = self.conn.get(_id)
+        if result is None:
+            self._set(_id)
+            return tuple([_id, 0])
+        self._set(_id, int(result))
+        return tuple([_id, int(result)])
+
+    def _set(self, _id: str, times: int = 0) -> bool:
+        times += 1
+        return self.conn.set(_id, times)
+
+    def _get_all(self) -> list:
+        results = self.conn.keys()
+        return results
+
+    def _get_image(self, theme: str) -> list:
+        images = []
+        for i in range(10):
+            images.append([i, self.conn.get(f'{theme}-{i}')])
+        width = int(self.conn.get(f'{theme}-width'))
+        height = int(self.conn.get(f'{theme}-height'))
+        for j in range(10):
+            images[j].append(width)
+            images[j].append(height)
+
+        return images