ai-yunying/douyin/user.py

from conf.config import logger,OUTPUT
from dataset import Table


class User:
    def __init__(self, db=ai_yunying_db) -> None:
        self.db = db
        self.table_user:Table = self.db.get_table("user")

    # 解析来自这个包请求返回的信息： https://www.douyin.com/aweme/v1/web/user/profile/other/
    # 返回的包内容如： .\testm\user_profile_response.py
    def filter_from_user_profile(self, body:dict):
        user = body['user']
        save_data = {
            "uid": user.get('uid'),
            "nickname": user.get('nickname'),
            "avatar_medium": user.get("avatar_medium"),
            "sec_uid": user.get('sec_uid'),
            "signature": user.get('signature'),
            "city": user.get('city'),
            "ip_location": user.get('ip_location'),
            "province": user.get('province'),
            "school_name" : user.get('school_name'),
            "follow_status": user.get('follow_status'),
            "follower_count" : user.get('follower_count'),
            "total_favorited" : user.get('total_favorited'),
            "aweme_count": user.get('aweme_count'),
        }
        return save_data
    def save_user_profile_to_db(self, body:dict):
        save_data = self.filter_from_user_profile(body)
        id = self.table_user.insert_ignore(save_data, keys=["uid"])
        logger.debug(f"插入/存在则忽略用户 id {id}")
        if id:
            return self.table_user.find_one(id=id)
        else:
            return self.table_user.find_one(uid=save_data['uid'])
        
user = User()