import json
from common.mq import MQ
from tqdm import tqdm

if __name__ == "__main__":
    path = 'resend.txt'
    with open(path, "r", encoding="utf-8") as f:
        datas = f.readlines()
    for line in tqdm(datas):
        video_dict = {}
        msg = json.loads(line)['msg']
        # print(msg)
        platform = json.loads(line)['platform']
        strategy = json.loads(line)['strategy']
        strs = msg.replace("CrawlerEtlParam", "")[1: -1]
        str_list = strs.split(", ")
        str_list = [i.replace("'", "") for i in str_list]
        key_dict = {
            "crawlerRule": "crawler_rule",
            "userId": "user_id",
            "outUserId": "out_user_id",
            "userName": "user_name",
            "avatarUrl": "avatar_url",
            "outVideoId": "out_video_id",
            "videoTitle": "video_title",
            "coverUrl": "cover_url",
            "videoUrl": "video_url",
            "publishTime": "publish_time",
            "playCnt": "play_cnt",
            "likeCnt": "like_cnt",
            "shareCnt": "share_cnt",
            "collectionCnt": "collection_cnt",
            "commentCnt": "comment_cnt",
            "strategyType": "strategy"
        }
        for index, i in enumerate(str_list[:-4]):
            if "=" not in i:
                continue
            else:
                key = i.split("=")[0]
                value = i[len(key) + 1:]
                new_key = key_dict.get(key, key)
                video_dict[new_key] = value.replace("null", "")

        video_dict['strategy'] = strategy
        video_dict['platform'] = platform
        video_dict['crawler_rule'] = json.dumps({})
        print(json.dumps(video_dict, ensure_ascii=False, indent=4))
        mq = MQ(topic_name="topic_crawler_etl_" + "prod")
        mq.send_msg(video_dict)

    # video_dict = {'video_title': '吴尊友因病去世！吴老师，您真的不容易，千言万语，汇成一句话您走好❗️', 'video_id': '5262651713', 'duration': 49, 'play_cnt': 71, 'like_cnt': 0, 'comment_cnt': 0, 'share_cnt': 1, 'user_name': '夏日❤️莲莲', 'publish_time_stamp': 1698398572, 'publish_time_str': '2023-10-27 17:22:52', 'video_width': 537, 'video_height': 954, 'avatar_url': 'https://cdn-xphoto2.xiaoniangao.cn/4987933869?Expires=1704038400&OSSAccessKeyId=LTAI4G2W1FsgwzAWYpPoB3v6&Signature=wopOmtlcp9tGyWHYW9uy7DIXO%2Bg%3D&x-oss-process=image%2Fresize%2Cw_200%2Ch_200%2Climit_0%2Finterlace%2C1%2Fquality%2Cq_50%2Fcrop%2Cw_200%2Ch_200%2Cg_center%2Fformat%2Cjpg%2Fauto-orient%2C0', 'profile_id': 55888345, 'profile_mid': 185546, 'cover_url': 'https://cdn-xphoto2.xiaoniangao.cn/5262652619?Expires=1704038400&OSSAccessKeyId=LTAI4G2W1FsgwzAWYpPoB3v6&Signature=qIIRzRICgyv40n3uMFeMwHCY8JY%3D&x-oss-process=image%2Fresize%2Cw_690%2Ch_385%2Climit_0%2Finterlace%2C1%2Fformat%2Cjpg%2Fauto-orient%2C0', 'video_url': 'https://cdn-xalbum2.xiaoniangao.cn/5262651713?Expires=1704038400&OSSAccessKeyId=LTAI5tB7cRkYiqHcTdkVprwb&Signature=hFGFAB49mmgUYwYcF4679bE%2BgLg%3D', 'session': 'xiaoniangao-author-1698402882'}
    # video_dict['strategy'] = "author"
    # video_dict['platform'] = "xiaoniangao"
    # video_dict['user_id'] = 58528269
    # video_dict['out_video_id'] = video_dict['video_id']
    # print(json.dumps(video_dict, ensure_ascii=False, indent=4))
    # mq = MQ(topic_name="topic_crawler_etl_" + "prod")
    # mq.send_msg(video_dict)
    #