# -*- coding: utf-8 -*-
import pandas as pd
from collections import defaultdict
import json
from pathlib import Path
DATA = "6.28"
dd = defaultdict(list)
def save_file(l, filename="sss"):
_temp = Path().cwd()
file_path = _temp / f"{filename}_list.txt"
with file_path.open("w", encoding="utf-8") as fs:
fs.write("\n".join(l))
def excel_to_list():
sExcelFile = "./update_video_task.xlsx"
df = pd.read_excel(sExcelFile, sheet_name=‘sheet1‘)
df = df.fillna(method=‘ffill‘)
df = df.loc[:, ~df.columns.str.contains(‘^Unnamed‘)]
d = df.to_dict("records")
return d
def gen_new_list(l, data=DATA):
for d in l:
if str(d["data"]) == data:
name = d["name"]
source_token = int(d["token_id"])
mid = d["mid"]
yield {"name": name, "source_token": int(source_token), "mid": mid}
def create_task_list():
d = excel_to_list()
weibo_list = []
bilibili_list = []
for item in gen_new_list(d):
name = item["name"]
mid = item["mid"]
source_token = item["source_token"]
dd[name].append((source_token, mid))
for k, v in dd.items():
mid_list = [_id[1] for _id in v]
source_token = v[0][0]
item = {"source_token": source_token, "mid_list": mid_list}
url = f‘http://{{"data":[{json.dumps(item)}]}}‘
task = f"{k} {url}"
if k.startswith("zzz"):
zzz_list.append(task)
elif k.startswith("sss"):
sss_list.append(task)
save_file(ssss_list, filename="sss")
save_file(zzz_list, filename="zzz")
if __name__ == ‘__main__‘:
create_task_list()
原文地址:https://www.cnblogs.com/c-x-a/p/13204744.html