用python写了个桌面版OTP小工具

Z先森 2022-05-18 PM 197℃ 1条

迫于不想每次都拿手机出来解锁并打开OTP软件,用python3写了个桌面版OTP程序,代码如下:

GetOTP.py
# -*- coding: utf-8 -*-
import base64
import configparser
import hashlib
import hmac
import struct
import time
import tkinter as tk

file = 'config.ini'
con = configparser.ConfigParser()
con.read(file, encoding='utf-8')
jump_str = con.get('secret','jump_str')
vpn_str = con.get('secret','vpn_str')

# 根据secret,生成固定key
def hotp(counter, secret):
    basedSecret = base64.b32decode(secret)
    structSecret = struct.pack(">Q", counter)
    hmacSecret = hmac.new(basedSecret, structSecret, hashlib.sha1).digest()
    ordSecret = hmacSecret[19] & 15
    tokenSecret = (struct.unpack(">I", hmacSecret[ordSecret:ordSecret+4])[0] & 0x7fffffff) % 1000000
    return tokenSecret

# 根据secret,生成时间步长为timeset的key
def totp(secret, timeset):
    counter = int(time.time()) // timeset
    return str(hotp(counter, secret))

# 每隔500毫秒,刷新key
def refreshOTP():
    textotp.configure(text=r'堡垒机    ' + totp(jump_str, 30).zfill(6) + '\nVPN       ' + totp(vpn_str, 120).zfill(6), font=("Microsoft YaHei", 16))
    windows.after(500, refreshOTP)

# 设置tk窗口参数
windows = tk.Tk()
windows.title('OTP')
windows.geometry('200x50')
textotp = tk.Label(windows, text=r'堡垒机    ' + totp(jump_str, 30).zfill(6) + '\nVPN       ' + totp(vpn_str, 120).zfill(6), font=("Microsoft YaHei", 16))
textotp.pack()
windows.after(500, refreshOTP)
windows.mainloop()
配置文件config.ini

XXX部分为BASE32后的字符串

[secret]
jump_str = XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
vpn_str = XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
效果如下:

git地址:

https://github.com/jacksnowfuck/py-otp.git

标签: none

如无特殊说明,本博所有文章均为博主原创。

评论啦~



唉呀 ~ 仅有一条评论


  1. XX
    XX

    0.0

    回复 2022-05-18 16:55