From ae21e0eb31825867b7b2ce2a2e2b9972164940df Mon Sep 17 00:00:00 2001
From: BennyThink <benny.think@gmail.com>
Date: Sun, 20 Feb 2022 22:44:44 +0800
Subject: [PATCH] add fakemysql

---
 ytdlbot/db.py        |  5 ++--
 ytdlbot/fakemysql.py | 58 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 60 insertions(+), 3 deletions(-)
 create mode 100644 ytdlbot/fakemysql.py

diff --git a/ytdlbot/db.py b/ytdlbot/db.py
index 2e37dd3..6390e2f 100644
--- a/ytdlbot/db.py
+++ b/ytdlbot/db.py
@@ -10,14 +10,12 @@ __author__ = "Benny <benny.think@gmail.com>"
 import base64
 import contextlib
 import datetime
-import json
 import logging
 import os
 import re
 import subprocess
 import time
 from io import BytesIO
-from unittest.mock import MagicMock
 
 import fakeredis
 import pymysql
@@ -27,6 +25,7 @@ from beautifultable import BeautifulTable
 from influxdb import InfluxDBClient
 
 from config import MYSQL_HOST, MYSQL_PASS, MYSQL_USER, QUOTA, REDIS
+from fakemysql import FakeMySQL
 
 
 class Redis:
@@ -205,7 +204,7 @@ class MySQL:
             self.con = pymysql.connect(host=MYSQL_HOST, user=MYSQL_USER, passwd=MYSQL_PASS, db="ytdl",
                                        charset="utf8mb4")
         else:
-            self.con = MagicMock()
+            self.con = FakeMySQL()
 
         self.cur = self.con.cursor()
         self.init_db()
diff --git a/ytdlbot/fakemysql.py b/ytdlbot/fakemysql.py
new file mode 100644
index 0000000..eb96a34
--- /dev/null
+++ b/ytdlbot/fakemysql.py
@@ -0,0 +1,58 @@
+#!/usr/local/bin/python3
+# coding: utf-8
+
+# ytdlbot - fakemysql.py
+# 2/20/22 20:08
+#
+
+__author__ = "Benny <benny.think@gmail.com>"
+
+import re
+import sqlite3
+
+init_con = sqlite3.connect(":memory:", check_same_thread=False)
+
+
+class FakeMySQL:
+    @staticmethod
+    def cursor() -> "Cursor":
+        return Cursor()
+
+    def commit(self):
+        pass
+
+    def close(self):
+        pass
+
+
+class Cursor:
+    def __init__(self):
+        self.con = init_con
+        self.cur = self.con.cursor()
+
+    def execute(self, *args, **kwargs):
+        sql = self.sub(args[0])
+        new_args = (sql,) + args[1:]
+        return self.cur.execute(*new_args, **kwargs)
+
+    def fetchall(self):
+        return self.cur.fetchall()
+
+    def fetchone(self):
+        return self.cur.fetchone()
+
+    @staticmethod
+    def sub(sql):
+        sql = re.sub(r"CHARSET.*|charset.*", "", sql, re.IGNORECASE)
+        sql = sql.replace("%s", "?")
+        return sql
+
+
+if __name__ == '__main__':
+    con = FakeMySQL()
+    cur = con.cursor()
+    cur.execute("create table user(id int, name varchar(20))")
+    cur.execute("insert into user values(%s,%s)", (1, "benny"))
+    cur.execute("select * from user")
+    data = cur.fetchall()
+    print(data)