Python中的google authenticator認證過程

更新時間：2024年11月22日 15:14:24 作者：slp_44777680

文章介紹了使用Python 3.7生成Google Authenticator所需密鑰的步驟,包括使用pyotp模塊生成密鑰、生成二維碼圖片以及通過客戶端掃描二維碼進行二次認證的實現(xiàn)原理

Python的google authenticator認證

環(huán)境描述

python 3.7
所需安裝包：
pyotp qrcode Image

實現(xiàn)原理

使用pyotp 的python模塊生成google auth 需要的密鑰
根據(jù)密鑰生成條形碼圖片
使用google authenticator 客戶端掃描條形碼，客戶端根據(jù)時間及密鑰經(jīng)過算法生成6位數(shù)的驗證碼
平臺二次認證通過對輸入的驗證碼進行校驗，校驗也是基于時間和密鑰

代碼實現(xiàn)

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : SLP
# @Time : 2021/10/3 10:25

import os
import traceback
import pyotp
from qrcode import QRCode, constants


class GoogleAuthenticatorClient:
    def __init__(self, secret_key=None):
        self.secret_key = secret_key

    def create_secret(self):
        """
        生成google auth 需要的密鑰
        :return:
        """
        self.secret_key = pyotp.random_base32(64)
        return self.secret_key

    def create_secret_qrcode(self, name=None, issuer_name=None, save_to_file=True):
        """
        根據(jù)用戶名及密鑰生成二維碼圖片
        :param name:用戶名
        :param issuer_name:發(fā)行人
        :param save_to_file: 保存至文件
        :return:
        """
        data = pyotp.totp.TOTP(self.secret_key).provisioning_uri(name=name, issuer_name=issuer_name)
        qr = QRCode(
            version=1,
            error_correction=constants.ERROR_CORRECT_L,
            box_size=6,
            border=4, )
        try:
            qr.add_data(data)
            qr.make(fit=True)
            img = qr.make_image()
            if save_to_file:
                base_dir = os.path.dirname(os.path.abspath(__file__))
                dir_path = os.path.join(base_dir, 'static', 'image')
                if not os.path.exists(dir_path):
                    os.makedirs(dir_path)
                filepath = dir_path + os.sep + self.secret_key + '.png'
                img.save(filepath)  # 保存條形碼圖片
                return True, filepath
            else:
                return img.get_image()
        except Exception as e:
            traceback.print_exc()
            return False, None

    def verify_code_func(self, verify_code):
        t = pyotp.TOTP(self.secret_key)
        result = t.verify(verify_code)
        return result


if __name__ == '__main__':
    secret_key = 'PU6PY6FWPVQ4BXE7ZP6X7YMVM3BH3ODS7SW53GL3LJPED7AAQUVF2EKP6AGNFFOX'
    google_auth_ = GoogleAuthenticatorClient(secret_key=secret_key)
    # secret = google_auth_.create_secret()
    # print('秘鑰', secret)
    # # 生成圖片二維碼
    image = google_auth_.create_secret_qrcode(name='slp', issuer_name='GoldBull', save_to_file=False)
    print(image.show())

    # 驗證
    # res = google_auth_.verify_code_func(verify_code='635543')
    # print(res)