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"交出你的验证码！
"\
f"{ecode}
" \
f"请复制到注册窗口中完成注册，感谢您的支持。
"
# 实例化邮件对象，并指定邮件的关键信息
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(')
# 文章编辑完成后当然就得发布文章，某种意义上来说就是一个请求而已。但是要优化好整个发布功能，其实要考虑的问题是很多的。
# ![image.png]( "image.png")
# 首先要解决的问题是图片压缩的问题，作者发布文章时，并不会去关注图片有多大，只是简单的上传并确保前端能正常显示。
# 
# 图片压缩分两种压缩方式，一种是压缩图片的尺寸，另外一种是压缩图片的大小。
# 
#
# '''
#
# list = parse_image_url(content)
#
# thumb = generate_thumb(list)
#
# print(thumb)