from src import token04

ERROR_CODE_SUCCESS = 0                   # 获取鉴权 token 成功
ERROR_CODE_APP_ID_INVALID = 1            # 调用方法时传入 app_id 参数错误
ERROR_CODE_USER_ID_INVALID = 3           # 调用方法时传入 user_id 参数错误
ERROR_CODE_SECRET_INVALID = 5            # 调用方法时传入 secret 参数错误
ERROR_CODE_EFFECTIVE_TIME_IN_SECONDS_INVALID = 6  # 调用方法时传入 effective_time_in_seconds 参数错误

# 参数
app_id:int  # Zego派发的数字ID, 各个开发者的唯一标识
user_id:str  # 用户 ID
secret:str   # 在获取 token 时进行 AES 加密的密钥
effective_time_in_seconds:int  # token 的有效时长，单位：秒
有效载荷:有效载荷

# 返回值
token:TokenInfo  # 三个成员: token, error_code, error_message 

if __name__ == '__main__':
    token_info = zego_server_assistant.generate_token04(2913569222, "demo", "b0d996aecc46ad51600ea853bb378c18", 3600, "")
    print([token_info.token, token_info.error_code, token_info.error_message])

if __name__ == '__main__':
    payload = {
        "room_id": "demo",
        "privilege": {1 : 1, 2 : 1},
        "stream_id_list": None
    }
    token_info = zego_server_assistant.generate_token04(app_id=1, user_id="demo", secret="fa94dd0f974cf2e293728a526b028271",
                                                        effective_time_in_seconds=3600, payload=json.dumps(payload))
    print([token_info.token, token_info.error_code, token_info.error_message])