diff --git a/DB.py b/DB.py index f5018ac..963bb61 100644 --- a/DB.py +++ b/DB.py @@ -62,6 +62,10 @@ video = Table("sh_dm_video_v2", _meta, Column("status", Integer), Column("createtime", Integer), Column("updatetime", Integer), + + Column("watch_number", Integer), + Column("follow_number", Integer), + Column("video_number", Integer), ) @@ -520,7 +524,9 @@ class DBSA: "public_time": data["create_time"], "cover_pic": data["cover_pic"], "sort": data["sort"], "u_xid": data["u_xid"], "u_id": data["u_id"], "u_pic": data["u_pic"], "u_name": data["u_name"], - "status": 1, "createtime": now_ts, "updatetime": now_ts, + "status": 1, "createtime": now_ts, "updatetime": now_ts, "watch_number": data.get("view", 0), + "follow_number": data.get("fans", 0), + "video_number": data.get("videos", 0), } with cls._lock: @@ -601,12 +607,6 @@ class DBSA: @classmethod def update_video_stats(cls, locator: dict, stats: dict) -> int: - """ - 立即更新 sh_dm_video_v2 表中的统计字段。n - :param locator: 用于定位行的字典,必须包含: v_xid, rn - :param stats: 需要更新的统计字段,如 {"fans": 633, "videos": 10090, "view": 1678408} - :return: 受影响的行数 - """ v_xid = locator.get("v_xid") rn = locator.get("rn") if not v_xid or not rn: diff --git a/main.py b/main.py index e562572..9893f28 100644 --- a/main.py +++ b/main.py @@ -117,7 +117,6 @@ def get_searchInfo(keyword, level, headers, proxy_name, r=2): "u_name": iteminfo.get('owner.screenname'), "u_pic": iteminfo.get('owner.avatar_60_url') } - print(v_data) video_list.append(v_data) time.sleep(1) return video_list