python模拟生成steam手机二次验证令牌

最近在尝试着写个steam自动上架交易的程序，结果发现涉及的东西好多…
steam登录的时候需要输入手机令牌，手机令牌是动态生成的，每隔30s更新一次，断网的情况下也不会受影响
这种令牌大多是基于TOTP（TOTP是基于散列的消息认证码（HMAC）的示例。 它使用加密哈希函数将密钥与当前时间戳组合在一起以生成一次性密码。 ）实现的，，，
根据秘钥和时间戳生成验证码的算法参考了sda的代码，改了N次之后终于正常了。。。
秘钥就是sda（steam desktop authenticator）生成的mafile文件中的shared_secrets，有了它就可以生成令牌验证码了

import sys
sys.path.append(r'E:\anaconda\Lib\site-packages')
from time import time
from Cryptodome.Hash import  HMAC,SHA1
import requests
from struct import pack,unpack
import json
from base64 import b64decode
from apscheduler.schedulers.background import BackgroundScheduler  

two_factor_time_query_url = "https://api.steampowered.com/ITwoFactorService/QueryTime/v0001"

#从文件中读入shared_secrets
with open(r'C:\Users\Administrator\Desktop\test_authenticator\data.txt') as fn:
    js = fn.read()
    dic = json.loads(js)  
    shared_secret = dic.get('shared_secret')
#得到时间补偿量，否则不能产生正确的code
def get_time_offset():
    twofactor_querytime_response = requests.post(url=two_factor_time_query_url, params={'http_timeout': 10}).json()
    ts = int(time())
    return int(twofactor_querytime_response.get('response', {}).get('server_time', ts)) - ts

def hmac_sha1(secret, data):
    return HMAC.new(secret, data, SHA1).digest()

#根据校准后的时间生成令牌
def generate_twofactor_code_for_time(shared_secret,aligned_time):
    hmac = hmac_sha1(b64decode(shared_secret),
                     pack('>Q', int(aligned_time)//30))  # this will NOT stop working in 2038

    start = ord(hmac[19:20]) & 0xF
    codeint = unpack('>I', hmac[start:start+4])[0] & 0x7fffffff

    charset = '23456789BCDFGHJKMNPQRTVWXY'
    code = ''

    for _ in range(5):
        codeint, i = divmod(codeint, len(charset))
        code += charset[i]
    return code  


#two_factor_code = generate_twofactor_code_for_time(shared_secret,aligned_time)  


def timedTask():
    aligned_time = int(time() + get_time_offset())     #补偿后的时间
    print('\rremain time:{0} {1}'.format(29 - aligned_time%30,generate_twofactor_code_for_time(shared_secret,aligned_time)),end='',flush=True)
  
    
if __name__ == '__main__':
    # 创建后台执行的 schedulers
    scheduler = BackgroundScheduler()  
    # 添加调度任务
    # 调度方法为 timedTask，触发器选择 interval(间隔性)，间隔时长为 2 秒
    scheduler.add_job(timedTask, 'interval', seconds=1)
    # 启动调度任务
    scheduler.start() 

下一步研究一下