feishu.py 优化代码

mysql.py 增加了select功能
test.py 写表格测试代码

applications/mysql.py

@@ -51,6 +51,21 @@ class Mysql(object):
         else:
             return True
 
+    def select(self, sql):
+        """
+        从mysql中读取数据
+        :param sql: 查询sql语句
+        """
+        cursor = self.connection.cursor()
+        cursor.execute(sql)
+        data = cursor.fetchall()
+        if data:
+            return data
+        else:
+            return []
+
+
+
 
 
 

feishu/feishu.py

@@ -2,7 +2,6 @@
 feishu python方法
 """
 import json
-import random
 
 import requests
 import datetime
@@ -156,8 +155,8 @@ class Feishu(object):
         print(w.json())
 
 
-if __name__ == "__main__":
-    F = Feishu()
-    for i in range(100):
-        value = [[i + 1] + [random.randint(0, 100) for item in range(7)]]
-        F.insert_value(sheet_id="gwzBOM", values=value, ranges="A2:G2")
+# if __name__ == "__main__":
+#     F = Feishu()
+#     for i in range(100):
+#         value = [[i + 1] + [random.randint(0, 100) for item in range(7)]]
+#         F.insert_value(sheet_id="gwzBOM", values=value, ranges="A2:G2")

test.py

@@ -0,0 +1,33 @@
+"""
+测试代码
+"""
+
+from feishu.feishu import Feishu
+from applications.mysql import Mysql
+
+
+F = Feishu()
+M = Mysql()
+
+sql = "SELECT * FROM we_analysis_results where mini_program = '票圈视频+' and date = \'2024-01-15\';"
+
+if __name__ == '__main__':
+    data_lines = M.select(sql)
+    for line in data_lines[:1]:
+        for info in line:
+            print(info)
+
+    # print(data)
+# value = [
+#     [1, 2, 3],
+#     [4, 5, 6],
+#     [7, 8, 9],
+#     [10, 11, 12],
+#     [13, 14, 15],
+#     [16, 17, 18]
+# ]
+# F.insert_value(
+#     sheet_id="gwzBOM",
+#     values=value,
+#     ranges="A2:C7"
+# )