From 510e9224ae541322a693ed3eee420c9c35bf1bef Mon Sep 17 00:00:00 2001 From: lwl0608 Date: Wed, 8 May 2024 22:40:09 +0800 Subject: [PATCH] day flow --- bi/vas_day_flow.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100644 bi/vas_day_flow.py diff --git a/bi/vas_day_flow.py b/bi/vas_day_flow.py new file mode 100644 index 0000000..feb800e --- /dev/null +++ b/bi/vas_day_flow.py @@ -0,0 +1,52 @@ +import time + +from lib.all import * +import datetime + + +# 日流水 + +class S: + def __init__(self): + self.mysql_db_vas = Mysql( + "rm-bp11t1616a1kjvmx5.mysql.rds.aliyuncs.com", 3306, "vas", "root", "Wishpal2024" + ) + self.mysql_db_bi = Mysql( + "127.0.0.1", 3306, "metabase_bi", "metabase_bi", "Wishpal_2024" + ) + + def __del__(self): + self.mysql_db_vas.close() + self.mysql_db_bi.close() + + def get_day_flow(self): + st = get_today_zero_time() + et = st + 86400 + sql = '''select sum(pay_amount) flow from vas_order where ct>={} and ct<{} and order_status in (1,2);'''.format(st, et) + rows = self.mysql_db_vas.query(sql) + if len(rows) > 0: + return safe_get_int(rows[0], "flow") + return 0 + + def update_day_flow(self, flow): + t = get_today_zero_time() + dt = datetime.datetime.fromtimestamp(t) + formatted_dt = dt.strftime('%Y-%m-%d 00:00:00') + + sql_q = "select * from vas_day_flow where pdate='{}'".format(get_time_str_by_ts(t)) + rows = self.mysql_db_bi.query(sql_q) + if len(rows) > 0: + sql_u = "update vas_day_flow set flow=%s where pdate=%s" + return self.mysql_db_bi.exec(sql_u, (flow, formatted_dt)) + else: + sql_i = "insert into vas_day_flow (pdate, flow) values (%s,%s)" + return self.mysql_db_bi.exec(sql_i, (formatted_dt, flow)) + + def proc(self): + flow = self.get_day_flow() + self.update_day_flow(flow) + print(flow) + + +s = S() +s.proc()