Pythonでワンタイムパスワード：TOTP

ワンタイムパスワードのPythonでの実装をまとめます。

インストール

pip install pyotp qrcode[pil]

QRコード生成

ここで作成するQRコードをGoogle Authenticateなどのワンタイムパスワードアプリで読み込みます。

import pyotp
import qrcode

# 1. シークレットを作成
secret = pyotp.random_base32()

# 2. TOTPオブジェクト作成
totp = pyotp.TOTP(secret)

# 3. Google Authenticator用の登録URLを作成
#    name: ユーザー名やメールアドレス
#    issuer_name: サービス名
otp_uri = totp.provisioning_uri(
    name="[e-mail]",
    issuer_name="SankoSC"
)

print("secret =", secret)
print("otp_uri =", otp_uri)

# 4. QRコード画像を作成
img = qrcode.make(otp_uri)

# 5. 保存
img.save("totp_qr.png")

認証

import pyotp

secret = "シークレット"
code = input("Enter the OTP code: ")
if pyotp.TOTP(secret).verify(code):
	print("OK")
else:
	print("NG")