更新代码

bin/db/db.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/2/24
+# @File Name: db.py

bin/db/sqlite.py

@@ -1,114 +0,0 @@
-#!/usr/bin/env python3
-# -- coding:utf-8 --
-# @Author: markushammered@gmail.com
-# @Development Tool: PyCharm
-# @Create Time: 2022/2/4
-# @File Name: sqlite.py
-
-import sqlite3
-from typing import Any
-from bin.utils.settings import Settings
-
-
-class Database:
-    def __init__(self):
-        self.sqlite_path = Settings().path
-
-    def insert_data(self, name: str) -> None:
-        """
-        插入数据
-        :param name:
-        :return:
-        """
-        conn = sqlite3.connect(self.sqlite_path, check_same_thread=False)
-        cursor = conn.cursor()
-        try:
-            cursor.execute('insert into ReqCount values(?, ?)', (name, 1))
-            conn.commit()
-        finally:
-            cursor.close()
-            conn.close()
-
-    def fetch_data(self, name: str) -> int:
-        """
-        获取数据
-        :param name:
-        :return:
-        """
-        conn = sqlite3.connect(self.sqlite_path, check_same_thread=False)
-        cursor = conn.cursor()
-        try:
-            cursor.execute('select * from ReqCount')
-            conn.commit()
-            data = cursor.fetchall()
-
-            temp_dict = {}
-            for k, v in data:  # 遍历数据将元组数据转换为字典类型
-                temp_dict.setdefault(k, []).append(v)
-            for i, c in zip(temp_dict.keys(), temp_dict.values()):
-                temp_dict[i] = c[0]
-
-            if name in temp_dict.keys():
-                count = temp_dict[name]  # 获取原数字 为 整型
-                self.update_data(name, count)
-                return count
-            else:
-                # 新建用户数据
-                self.insert_data(name)
-                return 0
-        finally:
-            cursor.close()
-            conn.close()
-
-    def update_data(self, name: str, times: int) -> None:
-        """
-        更新数据
-        :param name:
-        :param times:
-        :return:
-        """
-        conn = sqlite3.connect(self.sqlite_path, check_same_thread=False)
-        cursor = conn.cursor()
-        try:
-            times += 1
-            cursor.execute('update ReqCount set times=? where name=?', (times, name))
-            conn.commit()
-        finally:
-            cursor.close()
-            conn.close()
-
-    def fetch_table(self) -> list:
-        """
-        返回已有主题列表
-        :return:
-        """
-        conn_temp = sqlite3.connect(self.sqlite_path, check_same_thread=False)
-        cursor_temp = conn_temp.cursor()
-        try:
-            lst = []
-            cursor_temp.execute("select * from sqlite_master where type='table'")
-            for i in cursor_temp.fetchall():
-                lst.append(i[1])
-            return lst
-        finally:
-            cursor_temp.close()
-            conn_temp.close()
-
-    def fetch_style_data(self, style: str) -> list[dict[str, Any]]:
-        """
-        获取主题数据库内的数据
-        :param style:
-        :return:
-        """
-        conn = sqlite3.connect(self.sqlite_path, check_same_thread=False)
-        cursor = conn.cursor()
-        try:
-            cursor.execute('select * from %(style_name)s' % {'style_name': style})
-            data = cursor.fetchall()
-            data_set = []
-            for i in data:
-                data_set.append({'index': i[0], 'base64': i[1], 'width': i[2], 'height': i[3]})
-            return data_set
-        finally:
-            cursor.close()
-            conn.close()

bin/tests/MySQL_.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/2/25
+# @File Name: MySQL_.py