|
|
import random, string, time
|
|
|
from datetime import datetime
|
|
|
from io import BytesIO
|
|
|
from PIL import Image, ImageFont, ImageDraw
|
|
|
|
|
|
class ImageCode:
|
|
|
# 生成用于绘制字符串的随机颜色
|
|
|
def rand_color(self):
|
|
|
red = random.randint(32, 200)
|
|
|
green = random.randint(22, 255)
|
|
|
blue = random.randint(0, 200)
|
|
|
return red, green, blue
|
|
|
|
|
|
# 生成4位随机字符串
|
|
|
def gen_text(self):
|
|
|
# sample用于从一个大的列表或字符串中，随机取得N个字符，来构建出一个子列表
|
|
|
list = random.sample(string.ascii_letters+string.digits, 4)
|
|
|
return ''.join(list)
|
|
|
|
|
|
# 画一些干扰线，其中draw为PIL中的ImageDraw对象
|
|
|
def draw_lines(self, draw, num, width, height):
|
|
|
for num in range(num):
|
|
|
x1 = random.randint(0, int(width / 2))
|
|
|
y1 = random.randint(0, int(height / 2))
|
|
|
x2 = random.randint(0, int(width))
|
|
|
y2 = random.randint(int(height / 2), height)
|
|
|
draw.line(((x1, y1), (x2, y2)), fill='black', width=2)
|
|
|
|
|
|
# 绘制验证码图片
|
|
|
def draw_verify_code(self):
|
|
|
code = self.gen_text()
|
|
|
width, height = 120, 50 # 设定图片大小，可根据实际需求调整
|
|
|
# 创建图片对象，并设定背景色为白色
|
|
|
im = Image.new('RGB', (width, height), 'white')
|
|
|
# 选择使用何种字体及字体大小
|
|
|
font = ImageFont.truetype(font='arial.ttf', size=40)
|
|
|
draw = ImageDraw.Draw(im) # 新建ImageDraw对象
|
|
|
# 绘制字符串
|
|
|
for i in range(4):
|
|
|
draw.text((5 + random.randint(-3, 3) + 23 * i, 5 + random.randint(-3, 3)),
|
|
|
text=code[i], fill=self.rand_color(), font=font)
|
|
|
# 绘制干扰线
|
|
|
self.draw_lines(draw, 4, width, height)
|
|
|
# im.show() # 如需临时调试，可以直接将生成的图片显示出来
|
|
|
return im, code
|
|
|
|
|
|
# 定义一个方法，用于生成图片验证码并返回验证码字符串和图片数据
|
|
|
def get_code(self):
|
|
|
# 调用内部的draw_verify_code方法生成图片和验证码字符串
|
|
|
# 这里假设draw_verify_code返回一个元组，其中包含PIL Image对象和验证码字符串
|
|
|
image, code = self.draw_verify_code()
|
|
|
|
|
|
# 创建一个BytesIO对象，它是一个在内存中读写bytes的类文件对象
|
|
|
# 这里用它来保存生成的图片数据，而不需要实际写入文件
|
|
|
buf = BytesIO()
|
|
|
|
|
|
# 使用PIL Image对象的save方法将图片保存到BytesIO对象中
|
|
|
# 'jpeg'指定了保存图片的格式为JPEG
|
|
|
image.save(buf, 'jpeg')
|
|
|
|
|
|
# 从BytesIO对象中获取存储的字节数据
|
|
|
# 这些数据代表了图片的内容，可以用于发送到前端或存储在数据库中
|
|
|
bstring = buf.getvalue()
|
|
|
|
|
|
# 返回一个元组，包含验证码字符串和图片数据的字节串
|
|
|
# 这样调用者就可以根据返回的验证码字符串进行验证，并显示图片数据给用户
|
|
|
return code, bstring
|
|
|
|
|
|
# 发送邮箱验证码
|
|
|
from smtplib import SMTP_SSL
|
|
|
from email.mime.text import MIMEText
|
|
|
from email.header import Header
|
|
|
|
|
|
import smtplib
|
|
|
from email.mime.text import MIMEText
|
|
|
from email.header import Header
|
|
|
import random
|
|
|
import string
|
|
|
|
|
|
|
|
|
# 发送QQ邮箱验证码, 参数为收件箱地址和随机生成的验证码
|
|
|
def send_email(receiver, ecode):
|
|
|
sender = 'muyu <2733037909@qq.com>' # 你的邮箱账号和发件者签名
|
|
|
# 定义发送邮件的内容，支持HTML标签和CSS样式
|
|
|
content = f"<html><body><p>交出你的验证码！</p>"\
|
|
|
f"<p style='color: red; font-size: 20px;'>{ecode}</p>" \
|
|
|
f"<p>请复制到注册窗口中完成注册，感谢您的支持。</p></body></html>"
|
|
|
# 实例化邮件对象，并指定邮件的关键信息
|
|
|
message = MIMEText(content, 'html', 'utf-8')
|
|
|
# 指定邮件的标题，同样使用utf-8编码
|
|
|
message['Subject'] = Header('小小张同学的小张', 'utf-8')
|
|
|
message['From'] = sender # 指定发件人信息
|
|
|
message['To'] = receiver # 指定收件人邮箱地址
|
|
|
|
|
|
try:
|
|
|
smtpObj = smtplib.SMTP_SSL('smtp.qq.com', 465) # 连接到QQ邮件服务器的SSL端口
|
|
|
# 通过你的邮箱账号和授权码登录QQ邮箱（注意这里使用授权码而不是密码）
|
|
|
smtpObj.login('2733037909@qq.com', 'qghjougarncnddeh') # 请将'你的授权码'替换为实际的授权码
|
|
|
# 发送邮件，注意这里直接传递message对象，无需转换为字符串
|
|
|
smtpObj.sendmail(sender, receiver, message.as_string())
|
|
|
smtpObj.quit()
|
|
|
print("邮件发送成功！")
|
|
|
except Exception as e:
|
|
|
print("邮件发送失败：", e)
|
|
|
|
|
|
# 生成6位随机字符串作为邮箱验证码
|
|
|
|
|
|
|
|
|
def gen_email_code():
|
|
|
return ''.join(random.choices(string.ascii_letters + string.digits, k=6))
|
|
|
|
|
|
#
|
|
|
# # 测试邮箱能否发送成功
|
|
|
# code = gen_email_code()
|
|
|
# print(code)
|
|
|
# send_email('2082813839@qq.com', code)
|
|
|
|
|
|
# 单个模型类转换为标准的Python List数据
|
|
|
def model_list(result):
|
|
|
list = []
|
|
|
for row in result:
|
|
|
dict = {}
|
|
|
for k, v in row.__dict__.items():
|
|
|
if not k.startswith('_sa_instance_state'):
|
|
|
# 如果某个字段的值是datetime类型，则将其格式为字符串
|
|
|
if isinstance(v, datetime):
|
|
|
v = v.strftime('%Y-%m-%d %H:%M:%S')
|
|
|
dict[k] = v
|
|
|
list.append(dict)
|
|
|
|
|
|
return list
|
|
|
|
|
|
# SQLAlchemy连接查询两张表的结果集转换为[{},{}]
|
|
|
# Comment，Users， [(Comment, Users),(Comment, Users),(Comment, Users)]
|
|
|
def model_join_list(result):
|
|
|
list = [] # 定义列表用于存放所有行
|
|
|
for obj1, obj2 in result:
|
|
|
dict = {}
|
|
|
for k1, v1 in obj1.__dict__.items():
|
|
|
if not k1.startswith('_sa_instance_state'):
|
|
|
if not k1 in dict: # 如果字典中已经存在相同的Key则跳过
|
|
|
dict[k1] = v1
|
|
|
for k2, v2 in obj2.__dict__.items():
|
|
|
if not k2.startswith('_sa_instance_state'):
|
|
|
if not k2 in dict: # 如果字典中已经存在相同的Key则跳过
|
|
|
dict[k2] = v2
|
|
|
list.append(dict)
|
|
|
return list
|
|
|
|
|
|
# 压缩图片，通过参数width指定压缩后的图片大小
|
|
|
def compress_image(source, dest, width):
|
|
|
from PIL import Image
|
|
|
# 如果图片宽度大于1200，则调整为1200的宽度
|
|
|
im = Image.open(source)
|
|
|
x, y = im.size # 获取源图片的宽和高
|
|
|
if x > width:
|
|
|
# 等比例缩放
|
|
|
ys = int(y * width / x)
|
|
|
xs = width
|
|
|
# 调整当前图片的尺寸（同时也会压缩大小）
|
|
|
temp = im.resize((xs, ys), Image.ANTIALIAS)
|
|
|
# 将图片保存并使用80%的质量进行压缩
|
|
|
temp.save(dest, quality=80)
|
|
|
# 如果尺寸小于指定宽度则不缩减尺寸，只压缩保存
|
|
|
else:
|
|
|
im.save(dest, quality=80)
|
|
|
|
|
|
# 解析文章内容中的图片地址
|
|
|
def parse_image_url(content):
|
|
|
import re
|
|
|
temp_list = re.findall('<img src="(.+?)"', content)
|
|
|
url_list = []
|
|
|
for url in temp_list:
|
|
|
# 如果图片类型为gif，则直接跳过，不对其作任何处理
|
|
|
if url.lower().endswith('.gif'):
|
|
|
continue
|
|
|
url_list.append(url)
|
|
|
return url_list
|
|
|
|
|
|
# 远程下载指定URL地址的图片，并保存到临时目录中
|
|
|
def download_image(url, dest):
|
|
|
import requests
|
|
|
response = requests.get(url) # 获取图片的响应
|
|
|
# 将图片以二进制方式保存到指定文件中
|
|
|
with open(file=dest, mode='wb') as file:
|
|
|
file.write(response.content)
|
|
|
|
|
|
# 解析列表中的图片URL地址并生成缩略图，返回缩略图名称
|
|
|
def generate_thumb(url_list):
|
|
|
# 根据URL地址解析出其文件名和域名
|
|
|
# 通常建议使用文章内容中的第一张图片来生成缩略图
|
|
|
# 先遍历url_list，查找里面是否存在本地上传图片，找到即处理，代码运行结束
|
|
|
for url in url_list:
|
|
|
if url.startswith('/upload/'):
|
|
|
filename = url.split('/')[-1]
|
|
|
# 找到本地图片后对其进行压缩处理，设置缩略图宽度为400像素即可
|
|
|
compress_image('./resource/upload/' + filename,
|
|
|
'./resource/thumb/' + filename, 400)
|
|
|
return filename
|
|
|
|
|
|
# 如果在内容中没有找到本地图片，则需要先将网络图片下载到本地再处理
|
|
|
# 直接将第一张图片作为缩略图，并生成基于时间戳的标准文件名
|
|
|
url = url_list[0]
|
|
|
filename = url.split('/')[-1]
|
|
|
suffix = filename.split('.')[-1] # 取得文件的后缀名
|
|
|
thumbname = time.strftime('%Y%m%d_%H%M%S.' + suffix)
|
|
|
download_image(url, './resource/download/' + thumbname)
|
|
|
compress_image('./resource/download/' + thumbname, '../resource/thumb/' + thumbname, 400)
|
|
|
|
|
|
return thumbname # 返回当前缩略图的文件名
|
|
|
|
|
|
|
|
|
# if __name__ == '__main__':
|
|
|
#
|
|
|
# content = '''
|
|
|
# <p style="text-align:left;text-indent:28px">
|
|
|
# <span style="font-size:14px;font-family:宋体">文章编辑完成后当然就得发布文章，某种意义上来说就是一个请求而已。但是要优化好整个发布功能，其实要考虑的问题是很多的。</span></p>
|
|
|
# <p><img srcx="/upload/image.png" title="image.png" alt="image.png"/></p>
|
|
|
# <p><span style="font-size:14px;font-family:宋体">首先要解决的问题是图片压缩的问题，作者发布文章时，并不会去关注图片有多大，只是简单的上传并确保前端能正常显示。</span></p>
|
|
|
# <p><img src="http://www.woniuxy.com/page/img/banner/newBank.jpg"/></p>
|
|
|
# <p><span style="font-size:14px;font-family:宋体">图片压缩分两种压缩方式，一种是压缩图片的尺寸，另外一种是压缩图片的大小。
|
|
|
# </span><img src="http://ww1.sinaimg.cn/large/68b02e3bgy1g2rzifbr5fj215n0kg1c3.jpg"/>
|
|
|
# </p>
|
|
|
# '''
|
|
|
#
|
|
|
# list = parse_image_url(content)
|
|
|
#
|
|
|
# thumb = generate_thumb(list)
|
|
|
#
|
|
|
# print(thumb) |
