# 获取用户简略预览数据的 API 实现，此 API 无需任何登录即可调取

import json
from API_TitleServer import apiSDGB
from Config import *
import time
import random

def apiGetUserPreview(userId) -> str:
    data = json.dumps({
        "userId": int(userId)
    })
    preview_result = apiSDGB(data, "GetUserPreviewApi", userId)
    return preview_result

# CLI 示例
if __name__ == "__main__":
    #userId = input("请输入用户 ID：")
    userId = testUid
    print(apiGetUserPreview(userId))


def crawlAllUserPreview():
    # 这里设置开始和结束的 UserId
    BeginUserId = 11000000
    EndUserId = 12599999

    # 打开文件，准备写入
    with open('Remi_UserID_DB_Output.txt', 'w', encoding="utf-8") as f:
        # 遍历 UserId
        for userId in range(BeginUserId, EndUserId + 1):
            # 调用 API
            try:
                userPreview = apiGetUserPreview(userId)
                currentUser = json.loads(userPreview)
                if currentUser["userId"] is not None:
                    # 每爬到一个就把它存到一个文件里面，每个一行
                    f.write(userPreview + "\n")
                else:
                    f.write("\n")
            except:
                f.write("ERROR\n")
                time.sleep(4)
            f.flush()
            # 随机等待0.2-0.5秒
            time.sleep(random.uniform(0.2, 0.5))

    print('Finished!')

if __name__ == "__main__":
    crawlAllUserPreview()