import hmac
import struct
import time
from hashlib import sha1
from urllib.parse import urlencode, quote
if __name__ == '__main__':
    # account會(huì)作為標(biāo)識(shí)顯示在身份驗(yàn)證器上
    account = input("please enter your account: ")
    # secret用于生成秘鑰
    secret = input("please enter your secret: ")
    # label會(huì)作為標(biāo)識(shí)顯示在身份驗(yàn)證器上
    label = input("please enter your label: ")
    # 將secret轉(zhuǎn)換成bytes
    s = secret.encode()
    # 獲取時(shí)間片（1990年1月1日0時(shí)開(kāi)始計(jì)時(shí)，30秒為一個(gè)單位）
    c = struct.pack(">Q", int(time.time()) // 30)
    # 根據(jù)secret和時(shí)間片指定sha1算法計(jì)算hash值,返回bytes類型hash值
    hmac_hash = hmac.new(s, c, sha1).digest()
    print("hmac_hash:", len(hmac_hash))
    # 取出hmac_hash的第19位和0xf做”與“運(yùn)算
    offset = hmac_hash[19] & 0xf
    print("offset:", offset)
    # 從hmac_hash中取出4個(gè)16進(jìn)制字節(jié)轉(zhuǎn)換為正整數(shù)（I）并取索引為[0]，再與16進(jìn)制0x7fffffff做與運(yùn)算,最后除以10的六次方
    google_code = (struct.unpack(">I", hmac_hash[offset: offset + 4])[0] & 0x7fffffff) % 10 ** 6
    print(google_code)
    # 若計(jì)算后結(jié)果不足6位, 則在左側(cè)補(bǔ)0
    google_code = f'{google_code:>06}'
    print(google_code)
    prefix = label
    prefix += f':{account}'
    ends = {
        'secret': secret,
        'label': label
    }
    base_uri = 'otpauth://totp/{prefix}?{ends}'
    # 調(diào)用草料二維碼生成api
    caoliao_qrcode_url = 'https://api.pwmqr.com/qrcode/create/?url={qr_content}'
    qr_uri = base_uri.format(prefix=prefix, ends=urlencode(ends))
    print(caoliao_qrcode_url.format(qr_content=quote(qr_uri)))