python工具方法

isExists = os.path.exists(down_path)
if not isExists:
os.makedirs(down_path)
print(down_path + ' 创建成功')
else:
# 如果目录存在则不创建，并提示目录已存在
print(down_path + ' 目录已存在')

查看文件是否存在

import requests
import json
import urllib
import os.path

headers = {'Accept':'application/json'}
response = requests.get('https://manhua.163.com/book/catalog/5015165829890111936.json',headers=headers)
response = json.loads(response.text)
print(type(response))
index=1
print(response['catalog']['sections'])
for i in response['catalog']['sections']:
for j in i['sections']:
# print(j['title'])
titleName=str(index)+' '+j['title']
print(titleName)
index+=1

isExists = os.path.exists(titleName)

# 判断结果
if not isExists:

os.makedirs(titleName)

print(titleName + ' 创建成功')
else:
# 如果目录存在则不创建，并提示目录已存在
print(titleName + ' 目录已存在')

爬虫设置

print(''.join(random.sample(['z','y','x','w','v','u','t','s','r','q','p','o','n','m','l','k','j','i','h','g','f','e','d','c','b','a'], 5)))

产生随机的五个字符

Token产生与验证

import time
import base64
import hmac

def generate_token(key, expire=7260):
ts_str = str(time.time() + expire)
ts_byte = ts_str.encode("utf-8")
sha1_tshex_str = hmac.new(key.encode("utf-8"), ts_byte, 'sha1').hexdigest()
token = ts_str+':'+sha1_tshex_str
b64_token = base64.urlsafe_b64encode(token.encode("utf-8"))
return b64_token.decode("utf-8")

def certify_token(key, token):
token_str = base64.urlsafe_b64decode(token).decode('utf-8')
token_list = token_str.split(':')
if len(token_list) != 2:
return False
ts_str = token_list[0]
if float(ts_str) < time.time():
return False
known_sha1_tsstr = token_list[1]
sha1 = hmac.new(key.encode("utf-8"), ts_str.encode('utf-8'), 'sha1')
calc_sha1_tsstr = sha1.hexdigest()
if calc_sha1_tsstr != known_sha1_tsstr:
return False
return True


mykey = "fdjaljfal"
mykey="mytoken64"
mytoken = generate_token(mykey)
print(mytoken)
mytoken="MTU1ODM3NzAzOS41NjQwNTgzOmFkZmIxYzYzY2EyMGFmMzRiNDgwMmY1ZTQ1NWFkMmQyOTdhNzcwN2E="
print(certify_token(mykey,mytoken))

图形验证码

import base64
import random
from io import BytesIO

values = ''.join(random.sample(['z','y','x','w','v','u','t','s','r','q','p','o','n','m','l','k','j','i','h','g','f','e','d','c','b','a'], 5))
print(values)
from captcha.image import ImageCaptcha

img = ImageCaptcha(width=160, height=60, fonts=['./wqy.ttf'], font_sizes=None)
image = img.generate_image(values)
image.show()


output_buffer = BytesIO()
image.save(output_buffer, format='JPEG')
byte_data = output_buffer.getvalue()
base64_str = base64.b64encode(byte_data).decode("utf-8")
pic="data:image/jpeg;base64,"+str(base64_str)
print(pic)

print(base64_str)

文件上传

def post(self):
file = self.request.files.get('file')
r = []
for f in file:
hash = hashlib.sha256()
hash.update(f['body'])
filename = hash.hexdigest()
_, ext = os.path.splitext(f['filename'])
if ext:
filename += ext
with open(f'{STATIC_PATH}/{filename}', 'wb') as i:
i.write(f['body'])
r.append(f'/static/{filename}')
self.json(r)

【文章原创作者：防ddos攻击 http://www.558idc.com/shsgf.html 复制请保留原URL】