Python中的google authenticator认证过程

天利家私

Python的google authenticator认证


环境描述

• python 3.7
  
• 所需安装包 ：
  
• pyotp qrcode Image
  

实现原理

• 使用pyotp 的python模块生成google auth 需要的密钥
  
• 根据密钥生成条形码图片
  
• 使用google authenticator 客户端扫描条形码，客户端根据时间及密钥经过算法 生成6位数的验证码
  
• 平台二次认证通过对输入的验证码进行校验，校验也是基于时间和密钥
  

代码实现

1. #!/usr/bin/env python
   
2. # -*- coding: utf-8 -*-
   
3. # @Author : SLP
   
4. # @Time : 2021/10/3 10:25
   
5. 
   
6. import os
   
7. import traceback
   
8. import pyotp
   
9. from qrcode import QRCode, constants
   
10. 
    
11. 
    
12. class GoogleAuthenticatorClient:
    
13.     def __init__(self, secret_key=None):
    
14.         self.secret_key = secret_key
    
15. 
    
16.     def create_secret(self):
    
17.         """
    
18.         生成google auth 需要的密钥
    
19.         :return:
    
20.         """
    
21.         self.secret_key = pyotp.random_base32(64)
    
22.         return self.secret_key
    
23. 
    
24.     def create_secret_qrcode(self, name=None, issuer_name=None, save_to_file=True):
    
25.         """
    
26.         根据用户名及密钥生成二维码图片
    
27.         :param name:用户名
    
28.         :param issuer_name:发行人
    
29.         :param save_to_file: 保存至文件
    
30.         :return:
    
31.         """
    
32.         data = pyotp.totp.TOTP(self.secret_key).provisioning_uri(name=name, issuer_name=issuer_name)
    
33.         qr = QRCode(
    
34.             version=1,
    
35.             error_correction=constants.ERROR_CORRECT_L,
    
36.             box_size=6,
    
37.             border=4, )
    
38.         try:
    
39.             qr.add_data(data)
    
40.             qr.make(fit=True)
    
41.             img = qr.make_image()
    
42.             if save_to_file:
    
43.                 base_dir = os.path.dirname(os.path.abspath(__file__))
    
44.                 dir_path = os.path.join(base_dir, 'static', 'image')
    
45.                 if not os.path.exists(dir_path):
    
46.                     os.makedirs(dir_path)
    
47.                 filepath = dir_path + os.sep + self.secret_key + '.png'
    
48.                 img.save(filepath)  # 保存条形码图片
    
49.                 return True, filepath
    
50.             else:
    
51.                 return img.get_image()
    
52.         except Exception as e:
    
53.             traceback.print_exc()
    
54.             return False, None
    
55. 
    
56.     def verify_code_func(self, verify_code):
    
57.         t = pyotp.TOTP(self.secret_key)
    
58.         result = t.verify(verify_code)
    
59.         return result
    
60. 
    
61. 
    
62. if __name__ == '__main__':
    
63.     secret_key = 'PU6PY6FWPVQ4BXE7ZP6X7YMVM3BH3ODS7SW53GL3LJPED7AAQUVF2EKP6AGNFFOX'
    
64.     google_auth_ = GoogleAuthenticatorClient(secret_key=secret_key)
    
65.     # secret = google_auth_.create_secret()
    
66.     # print('秘钥', secret)
    
67.     # # 生成图片二维码
    
68.     image = google_auth_.create_secret_qrcode(name='slp', issuer_name='GoldBull', save_to_file=False)
    
69.     print(image.show())
    
70. 
    
71.     # 验证
    
72.     # res = google_auth_.verify_code_func(verify_code='635543')
    
73.     # print(res)

复制代码

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

来源:https://www.jb51.net/python/331227sfi.htm
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！