From 39742d9aa5e95ed10dba5c2660d661392cb30740 Mon Sep 17 00:00:00 2001
From: lwl0608 <lwl0608@foxmail.com>
Date: Wed, 10 Jul 2024 14:16:15 +0800
Subject: [PATCH] add

---
 vas/calc_diamonds.py | 73 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 vas/calc_diamonds.py

diff --git a/vas/calc_diamonds.py b/vas/calc_diamonds.py
new file mode 100644
index 0000000..75c5f25
--- /dev/null
+++ b/vas/calc_diamonds.py
@@ -0,0 +1,73 @@
+from lib.all import *
+
+
+class S:
+    def __init__(self):
+        self.mysql_db_vas = Mysql(
+            "rm-bp11t1616a1kjvmx5.mysql.rds.aliyuncs.com", 3306, "vas", "root", "Wishpal2024"
+        )
+
+    def __del__(self):
+        self.mysql_db_vas.close()
+
+    def get_ch_income(self, mid):
+        sql = "select * from vas_ch_income where mid={} order by order_id".format(mid)
+        rows = self.mysql_db_vas.query(sql)
+        return rows
+
+    def get_withdraw_his(self, order_id):
+        sql = "select * from vas_withdraw_diamonds_his where order_id='{}'".format(order_id)
+        rows = self.mysql_db_vas.query(sql)
+        return rows
+
+    def get_order(self, order_id):
+        sql = "select * from vas_order where id='{}'".format(order_id)
+        rows = self.mysql_db_vas.query(sql)
+        if not rows:
+            return None
+        if len(rows) > 0:
+            return rows[0]
+
+    def get_coin_order(self, order_id):
+        sql = "select * from vas_coin_order where id='{}'".format(order_id)
+        rows = self.mysql_db_vas.query(sql)
+        if not rows:
+            return None
+        if len(rows) > 0:
+            return rows[0]
+
+    def proc(self, mid):
+        hds = [
+            "order_id",
+            "change",
+            "typ",
+            "order_status",
+            "结算状态"
+        ]
+        csv_ = Csv("{}_incom.csv".format(mid), hds)
+        rows = self.get_ch_income(mid)
+        for row in rows:
+            order_id = safe_get_str(row, "order_id")
+            typ = "money"
+            o = self.get_order(order_id)
+            if not o:
+                typ = "coin"
+                o = self.get_coin_order(order_id)
+            if not o:
+                typ = "error"
+            order_stats = safe_get_int(o, "order_status")
+
+            data = [
+                order_id, safe_get_int(row, "change"), typ, order_stats
+            ]
+            w_his = self.get_withdraw_his(order_id)
+            if w_his:
+                pass
+            else:
+                data.append("未结算")
+
+            csv_.append([data])
+
+
+s = S()
+s.proc(74)