From 438882572611aa12182685c150ac9f128354c95e Mon Sep 17 00:00:00 2001 From: lwl0608 Date: Tue, 3 Sep 2024 22:38:45 +0800 Subject: [PATCH] fix --- vas/calc_third_partner.py | 64 ++++++++++++++++++++++++++++++++++++ zone/get_streamer_moments.py | 50 ++++++++++++++++++++++++++++ 2 files changed, 114 insertions(+) create mode 100644 vas/calc_third_partner.py create mode 100644 zone/get_streamer_moments.py diff --git a/vas/calc_third_partner.py b/vas/calc_third_partner.py new file mode 100644 index 0000000..ba0ab86 --- /dev/null +++ b/vas/calc_third_partner.py @@ -0,0 +1,64 @@ +from lib.all import * + + +class S: + def __init__(self): + self.mysql_db_vas = Mysql( + "rm-bp11t1616a1kjvmx5.mysql.rds.aliyuncs.com", 3306, "vas", "root", "Wishpal2024" + ) + self.col_ztp = MongoDB( + host="mongodb://root:Wishpal2024@dds-bp1da1ddd62bede41.mongodb.rds.aliyuncs.com:3717,dds-bp1da1ddd62bede42.mongodb.rds.aliyuncs.com:3717/admin?replicaSet=mgset-77304659", + port=3717, + db="zone_third_partner", + collection="zone_third_partner" + ) + self.col_account = MongoDB( + host="mongodb://root:Wishpal2024@dds-bp1da1ddd62bede41.mongodb.rds.aliyuncs.com:3717,dds-bp1da1ddd62bede42.mongodb.rds.aliyuncs.com:3717/admin?replicaSet=mgset-77304659", + port=3717, + db="account", + collection="account" + ) + + def __del__(self): + self.mysql_db_vas.close() + self.col_ztp.close() + self.col_account.close() + + def uid2mid(self, uid): + q = { + "user_id": uid + } + doc = self.col_account.find_one(q) + return safe_get_int(doc, "_id") + + # 获取代运营管理的空间 + def get_tp_zids(self, tp_mids): + q = { + "third_partner_mid": {"$in": tp_mids} + } + docs = self.col_ztp.find(q) + zids = list() + for d in docs: + zid = safe_get_int(d, "zid") + zids.append(zid) + return zids + + # 获取zid解锁空间 + def get_zone_admission_cnt(self, zids, st, et): + sql = ''' + select zid, count(1) as admission_cnt from vas_zone_member where zid in (%s) and ct>=%d and ct<%d + ''' % (get_list_str(zids), st, et) + rows = self.mysql_db_vas.query(sql) + zid_cnt_map = dict() + for row in rows: + zid = safe_get_int(row, "zid") + ac = safe_get_int(row, "admission_cnt") + zid_cnt_map[zid] = ac + return zid_cnt_map + + def proc(self): + tp_uids = [983565, 231638] + tp_mids = list() + for uid in tp_uids: + tp_mids = tp_mids.append(self.uid2mid(uid)) + print(tp_mids) diff --git a/zone/get_streamer_moments.py b/zone/get_streamer_moments.py new file mode 100644 index 0000000..1cfb7e2 --- /dev/null +++ b/zone/get_streamer_moments.py @@ -0,0 +1,50 @@ +from lib.all import * + +import requests + +cookies = { + 'account': '%7B%22mid%22%3A161%2C%22name%22%3A%22user_246567%22%2C%22user_id%22%3A246567%2C%22avatar%22%3A%7B%22image_ids%22%3A%5B1%5D%2C%22video_ids%22%3Anull%2C%22images%22%3A%5B%7B%22id%22%3A1%2C%22w%22%3A100%2C%22h%22%3A100%2C%22fmt%22%3A%22image%2Fpng%22%2C%22urls%22%3A%5B%22https%3A%2F%2Ffilecdn01.tiefen.fun%2Fimgprod%2F8d%2Fd5%2F7ddc-1225-4f4c-ace0-c1c63f46b23d%22%5D%7D%5D%2C%22videos%22%3A%5B%5D%7D%2C%22level%22%3A1%2C%22role%22%3A1%2C%22current_exp%22%3A0%2C%22current_level_exp%22%3A100%2C%22is_dnd_mode_enabled%22%3A0%2C%22gold_num%22%3A0%2C%22diamond_num%22%3A0%2C%22is_a_member%22%3A1%2C%22ct%22%3A1703437387%2C%22ut%22%3A1714759503%7D', + 'token': 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhdXRob3JpemVkIjp0cnVlLCJtaWQiOjE2MSwidG9rZW5fdXVpZCI6MTgxOTY3MzM2Nzk0MzM0MDAzMn0.hc2N0QpwNVvoYO0kKl6j_LRp3Dg7KR4PmnfLD0Bq0jY', +} + +headers = { + 'accept': '*/*', + 'accept-language': 'zh-CN,zh;q=0.9,ko;q=0.8,ja;q=0.7', + 'content-type': 'application/json', + # 'cookie': 'account=%7B%22mid%22%3A161%2C%22name%22%3A%22user_246567%22%2C%22user_id%22%3A246567%2C%22avatar%22%3A%7B%22image_ids%22%3A%5B1%5D%2C%22video_ids%22%3Anull%2C%22images%22%3A%5B%7B%22id%22%3A1%2C%22w%22%3A100%2C%22h%22%3A100%2C%22fmt%22%3A%22image%2Fpng%22%2C%22urls%22%3A%5B%22https%3A%2F%2Ffilecdn01.tiefen.fun%2Fimgprod%2F8d%2Fd5%2F7ddc-1225-4f4c-ace0-c1c63f46b23d%22%5D%7D%5D%2C%22videos%22%3A%5B%5D%7D%2C%22level%22%3A1%2C%22role%22%3A1%2C%22current_exp%22%3A0%2C%22current_level_exp%22%3A100%2C%22is_dnd_mode_enabled%22%3A0%2C%22gold_num%22%3A0%2C%22diamond_num%22%3A0%2C%22is_a_member%22%3A1%2C%22ct%22%3A1703437387%2C%22ut%22%3A1714759503%7D; token=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhdXRob3JpemVkIjp0cnVlLCJtaWQiOjE2MSwidG9rZW5fdXVpZCI6MTgxOTY3MzM2Nzk0MzM0MDAzMn0.hc2N0QpwNVvoYO0kKl6j_LRp3Dg7KR4PmnfLD0Bq0jY', + 'origin': 'https://op.tiefen.fun', + 'priority': 'u=1, i', + 'referer': 'https://op.tiefen.fun/EditSpacePost', + 'sec-ch-ua': '"Not/A)Brand";v="8", "Chromium";v="126", "Google Chrome";v="126"', + 'sec-ch-ua-mobile': '?0', + 'sec-ch-ua-platform': '"macOS"', + 'sec-fetch-dest': 'empty', + 'sec-fetch-mode': 'cors', + 'sec-fetch-site': 'same-origin', + 'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/126.0.0.0 Safari/537.36', +} + +json_data = { + 'user_id': 616367, + 'offset': 0, + 'limit': 1000, + 'b_mid': 161, + 'b_ch': 'op', + 'b_ts': 1722679061111, + 'b_token': 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhdXRob3JpemVkIjp0cnVlLCJtaWQiOjE2MSwidG9rZW5fdXVpZCI6MTgxOTY3MzM2Nzk0MzM0MDAzMn0.hc2N0QpwNVvoYO0kKl6j_LRp3Dg7KR4PmnfLD0Bq0jY', +} + +response = requests.post('https://op.tiefen.fun/op/zone_moment/list_by_user_id', cookies=cookies, headers=headers, json=json_data) +res = json.loads(response.text) +data = safe_get_dict(res, "data") +lis = safe_get_list(data, "list") +for l in lis: + media_component = safe_get_dict(l, "media_component") + video_ids = safe_get_list(media_component, "video_ids") + for vid in video_ids: + print("{}:true,".format(vid)) + +# Note: json_data will not be serialized by requests +# exactly as it was in the original request. +# data = '{"user_id":616367,"offset":0,"limit":1000,"b_mid":161,"b_ch":"op","b_ts":1722679061111,"b_token":"eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhdXRob3JpemVkIjp0cnVlLCJtaWQiOjE2MSwidG9rZW5fdXVpZCI6MTgxOTY3MzM2Nzk0MzM0MDAzMn0.hc2N0QpwNVvoYO0kKl6j_LRp3Dg7KR4PmnfLD0Bq0jY"}' +# response = requests.post('https://op.tiefen.fun/op/zone_moment/list_by_user_id', cookies=cookies, headers=headers, data=data)