python totp代码
import time
import datetime
import math
import hmac
import base64
import qrcode
from PIL import Image
from pyzbar import pyzbar
from hashlib import sha1 '''
depend: qrcode,pillow,PIL,pyzbar ''' class GoogleAuthenticator(object): def __init__(self,secretKey=None,digits=6,interval=30):
self.secretKey = secretKey
self.digits = digits
self.interval = interval def __str_extend(self,old_str,length,extend_str):
new_strs = None
if len(old_str) < length:
clen = length - len(old_str)
s0 = ''.join([extend_str for i in range(clen)])
new_strs = s0 + old_str
elif len(old_str) == length:
new_strs = old_str
return new_strs def __str_split(self,old_str,split_len,prefix='',suffix=''):
array = []
str_len = len(old_str)
for i in range(0,str_len,4):
if i + 4 < str_len:
array.append(prefix+old_str[i:i+4]+suffix)
else:
array.append(prefix+old_str[i:str_len]+suffix)
return array def __base32tohex(self,base32str):
base32chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567"
bits = ""
for i in range(len(base32str)):
char = str(base32str[i]).upper()
idx = int(base32chars.index(char))
bit = self.__str_extend(format(idx,'b'),5,'0')
if bit is None:
raise Exception('bit={},len={}'.format(bit,len(bit)))
bits += bit
bitArray = self.__str_split(bits,4,'0b')
hexstr = ''.join([format(int(i,2),'x') for i in bitArray])
return hexstr def __get_HexSecret(self):
hexstr = self.__base32tohex(self.secretKey)
return hexstr def __byte_secret(self):
missing_padding = len(self.secretKey) % 8
if missing_padding != 0:
self.secret += '=' * (8 - missing_padding)
return base64.b32decode(self.secretKey, casefold=True) def __int_to_bytestring(self,i, padding=8):
result = bytearray()
while i != 0:
result.append(i & 0xFF)
i >>= 8
return bytes(bytearray(reversed(result)).rjust(padding, b'\0')) def __timecode(self, for_time):
i = time.mktime(for_time.timetuple())
return int(i / self.interval) def get_QR_url(self):
base_url = "https://chart.googleapis.com/chart?chs=200x200&cht=qr&chl=200x200&chld=M|0&cht=qr&chl="
QR_url = base_url + "otpauth://totp/user@host.com\%3Fsecret%3D" + self.secretKey
return QR_url def get_QR_code(self):
qr = qrcode.QRCode(version=1,error_correction=qrcode.constants.ERROR_CORRECT_L,box_size=10,border=4)
data = self.get_QR_url()
qr.add_data(data=data)
qr.make(fit=True)
img = qr.make_image(fill_color="green", back_color="white")
img.show() def QR2url(self,QR_path):
base_url = 'https://chart.googleapis.com/chart?chs=200x200&cht=qr&chl=200x200&chld=M|0&cht=qr&chl='
ourl = pyzbar.decode(Image.open(QR_path), symbols=[pyzbar.ZBarSymbol.QRCODE])[0].data.decode("utf-8")
QR_url = base_url + ourl
return QR_url def QR2code(self,QR_path):
QR_url = pyzbar.decode(Image.open(QR_path), symbols=[pyzbar.ZBarSymbol.QRCODE])[0].data.decode("utf-8")
i = QR_url.find('\%3Fsecret%3D')
l = len('\%3Fsecret%3D')
if i == -1:
i = QR_url.find('?secret=')
l = len('?secret=')
if i == -1:
raise Exception('QR is error')
j = QR_url.find('?',i+1)
if j == -1:
j = QR_url.find('\%3F',i+1)
if j == -1:
sk = QR_url[i+l:]
else:
sk = QR_url[i+l:j]
self.secretKey = sk
timestamp = self.__timecode(datetime.datetime.now())
str_code = self.generate_otp(timestamp)
return str_code def generate_otp(self,timestamp):
key = self.secretKey
hasher = hmac.new(self.__byte_secret(), self.__int_to_bytestring(timestamp), sha1)
# print(hasher.hexdigest())
hmac_hash = bytearray(hasher.digest())
offset = hmac_hash[-1] & 0xf
code = ((hmac_hash[offset] & 0x7f) << 24 |
(hmac_hash[offset + 1] & 0xff) << 16 |
(hmac_hash[offset + 2] & 0xff) << 8 |
(hmac_hash[offset + 3] & 0xff))
str_code = str(code % 10 ** self.digits)
while len(str_code) < self.digits:
str_code = '0' + str_code
return str_code def getTotp(self):
timestamp = self.__timecode(datetime.datetime.now())
str_code = self.generate_otp(timestamp)
return str_code def at(self, for_time, counter_offset=0):
if not isinstance(for_time, datetime.datetime):
for_time = datetime.datetime.fromtimestamp(int(for_time))
return self.generate_otp(self.timecode(for_time) + counter_offset) def verifyTotp(self,str_code):
gTotp = self.getTotp()
if gTotp == str_code:
return True
else:
return False
python totp代码的更多相关文章
- Python一行代码
1:Python一行代码画出爱心 print]+(y*-)**-(x**(y*<= ,)]),-,-)]) 2:终端路径切换到某文件夹下,键入: python -m SimpleHTTPServ ...
- python爬虫代码
原创python爬虫代码 主要用到urllib2.BeautifulSoup模块 #encoding=utf-8 import re import requests import urllib2 im ...
- Python小代码_2_格式化输出
Python小代码_2_格式化输出 name = input("name:") age = input("age:") job = input("jo ...
- Python小代码_1_九九乘法表
Python小代码_1_九九乘法表 max_num = 9 row = 1 while row <= max_num: col = 1 while col <= row: print(st ...
- Python IDLE 代码高亮主题
Python IDLE 代码高亮主题 使用方法: 打开C盘我的 C:\Documents and Settings\你的用户名.idlerc文件夹 里面会有一个 config-highlight.cf ...
- 代码块: 以冒号作为开始,用缩进来划分作用域,这个整体叫做代码块,python的代码块可以提升整体的整齐度,提高开发效率
# ### 代码块: 以冒号作为开始,用缩进来划分作用域,这个整体叫做代码块 if 5 == 5: print(1) print(2) if True: print(3) print(4) if Fa ...
- uiautomatorviewer 优化定位符生成,支持生成Java,Python自动化代码
项目介绍 二次开发 uiautomatorviewer 优化定位符生成,支持生成Java,Python自动化代码,修复自带工具画面有动态加载时截图失败问题,优化自带工具截图速度 ,实现类似录制脚本功能 ...
- Python实现代码统计工具——终极加速篇
Python实现代码统计工具--终极加速篇 声明 本文对于先前系列文章中实现的C/Python代码统计工具(CPLineCounter),通过C扩展接口重写核心算法加以优化,并与网上常见的统计工具做对 ...
- Python静态代码检查工具Flake8
简介 Flake8 是由Python官方发布的一款辅助检测Python代码是否规范的工具,相对于目前热度比较高的Pylint来说,Flake8检查规则灵活,支持集成额外插件,扩展性强.Flake8是对 ...
随机推荐
- 阿里巴巴java规则p3c结合sonar使用
sonar插件位置: jar包放至$SONAR_HOME/extensions/plugins下,重启sonar即可 如何在sonarqube的pmd插件中整合阿里开发规范 alibaba p3c s ...
- Ubuntu 14.04安装vim8
本文介绍两种方法安装 VIM8.1 方法一:GitHub下载源码手动安装 1.查看系统是否有安装vim:如果有,先删除 dpkg -l | grep vim 2.从git上下载 git clone h ...
- VScode中文乱码的问题
在vscode中打开文件,文件中的中文乱码了.解决方案: 1.vscode右下角,点击utf-8 (选择编码) 2.出现下面选择框,点击第一个 3.输入gb2312,如图选择 4.此刻解决乱码问题了. ...
- Django:(08)序列化器
1.序列化和反序列化变量从内存中变成可存储或传输的过程称之为序列化,序列化之后,就可以把序列化后的内容写入磁盘,或者通过网络传输到别的机器上.反过来,把变量内容从序列化的对象重新读到内存里称之为反序列 ...
- 通过bat批处理程序如何实现在多个txt文件后面加上相同的一行文字
通过bat批处理程序如何实现在多个txt文件后面加上相同的一行文字 set/p a=输入要增加的文字 for /f "delims=" %%i in ('dir /b *.txt' ...
- webdriver(处理select下拉框元素)
"""处理下拉框""" from selenium import webdriver from selenium.webdriver.com ...
- TF-IDF算法之关键词提取
(注:本文转载自阮一峰老师的博文,原文地址:http://www.ruanyifeng.com/blog/2013/03/tf-idf.html) 这个标题看上去好像很复杂,其实我要谈的是一个很简单的 ...
- 删除Vue中无权限的【node_modules】文件
npm install rimraf -g rimraf node_modules
- 数字麦克风PDM转PCM与STM32 I2S接口应用----重要文档列表
数字麦克风PDM脉冲到PCM信号需要一个二次采样,ST 提过了PDM2PCM的软件包,可以完成上面的工作.软件包源码没有开源,使用手册也简洁的让人抓狂,我觉得可能是因为ST更高级的MCU直接带了硬解码 ...
- POJ 1734:Sightseeing trip
Sightseeing trip Time Limit: 1000MS Memory Limit: 65536K Total Submissions: Accepted: Special Judge ...