Django之缓存、信号和图片验证码、ORM性能
一、 缓存
1、 介绍
缓存通俗来说:就是把数据先保存在某个地方,下次再读取的时候不用再去原位置读取,让访问速度更快。
缓存机制图解
2、Django中提供了6种缓存方式
1. 开发调试
2. 内存
3. 文件
4. 数据库
5. Memcache缓存(python-memcached模块)
6. Memcache缓存(pylibmc模块)
注意:下面对缓存进行配置,其实就是设置了缓存方式,是为了设置缓存的存放位置,如果没有设置缓存方式,那么就默认使用的是本地内存缓存的方式。
3、 配置缓存(在setting中配置)
配置缓存的方式 1. 开发调试 # 此为开始调试用,实际内部不做任何操作 CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.dummy.DummyCache', # 引擎 'TIMEOUT': 300, # 缓存超时时间(默认300,None表示永不过期,0表示立即过期) 'OPTIONS':{ 'MAX_ENTRIES': 300, # 最大缓存个数(默认300) 'CULL_FREQUENCY': 3, # 缓存到达最大个数之后,剔除缓存个数的比例,即:1/CULL_FREQUENCY(默认3) }, 'KEY_PREFIX': '', # 缓存key的前缀(默认空) 'VERSION': 1, # 缓存key的版本(默认1) 'KEY_FUNCTION' 函数名 # 生成key的函数(默认函数会生成为:【前缀:版本:key】) } } # 自定义key def default_key_func(key, key_prefix, version): """ Default function to generate keys. Constructs the key used by all other methods. By default it prepends the `key_prefix'. KEY_FUNCTION can be used to specify an alternate function with custom key making behavior. """ return '%s:%s:%s' % (key_prefix, version, key) def get_key_func(key_func): """ Function to decide which key function to use. Defaults to ``default_key_func``. """ if key_func is not None: if callable(key_func): return key_func else: return import_string(key_func) return default_key_func 2. 内存 # 此缓存将内容保存至内存的变量中 CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache', 'LOCATION': 'unique-snowflake', 'TIMEOUT': 300, # 缓存超时时间(默认300,None表示永不过期,0表示立即过期) 'OPTIONS': { 'MAX_ENTRIES': 300, # 最大缓存个数(默认300) 'CULL_FREQUENCY': 3, # 缓存到达最大个数之后,剔除缓存个数的比例,即:1/CULL_FREQUENCY(默认3) }, } } 3. 文件 # 此缓存将内容保存至文件 # 配置: CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.filebased.FileBasedCache', 'LOCATION': '/var/tmp/django_cache', # 文件路径 } } # 注:其他配置同开发调试版本 4. 数据库 # 此缓存将内容保存至数据库 # 配置: CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.db.DatabaseCache', 'LOCATION': 'my_cache_table', # 数据库表 } } # 注:执行创建表命令 python manage.py createcachetable 5. Memcache缓存(python-memcached模块) # 此缓存使用python-memcached模块连接memcache CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.memcached.MemcachedCache', 'LOCATION': '127.0.0.1:11211', } } CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.memcached.MemcachedCache', 'LOCATION': 'unix:/tmp/memcached.sock', } } CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.memcached.MemcachedCache', 'LOCATION': [ '172.19.26.240:11211', '172.19.26.242:11211', ] } } 6. Memcache缓存(pylibmc模块) # 此缓存使用pylibmc模块连接memcache CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.memcached.PyLibMCCache', 'LOCATION': '127.0.0.1:11211', } } CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.memcached.PyLibMCCache', 'LOCATION': '/tmp/memcached.sock', } } CACHES = { 'default': { 'BACKEND': 'django.core.cache.backends.memcached.PyLibMCCache', 'LOCATION': [ '172.19.26.240:11211', '172.19.26.242:11211', ] } }
4、 简单的应用
配置好了缓存的方式之后,我们使用缓存机制也是非常简单的
1. 给单独的视图应用缓存: 粒度适中 方式一:直接给视图函数添加装饰器 # views.py import datetime from django.views.decorators.cache import cache_page @cache_page(15) # 缓存15秒后失效 def test_cache(request): now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') return HttpResponse(now) 方式二:给路由添加 # urls.py from myapp.views import test_cache from django.views.decorators.cache import cache_page urlpatterns = [ url(r'^testcache/$', cache_page(15)(test_cache)), ] 方式一和二选择一种即可 2. 全站应用: 粒度最大(settings.py) 使用中间件,经过一系列的认证等操作,如果内容在缓存中存在,则使用FetchFromCacheMiddleware获取内容并返回给用户, 当返回给用户之前,判断缓存中是否已经存在,如果不存在则UpdateCacheMiddleware会将缓存保存至缓存,从而实现全站缓存(所有视图都进行缓存) MIDDLEWARE = [ # 站点缓存 , 注意必须在第一个位置 'django.middleware.cache.UpdateCacheMiddleware', # 其他中间件... # 站点缓存 , 注意必须在最后一个位置 'django.middleware.cache.FetchFromCacheMiddleware', ] CACHE_MIDDLEWARE_ALIAS = "" CACHE_MIDDLEWARE_SECONDS = 300 # 缓存有效时间 CACHE_MIDDLEWARE_KEY_PREFIX = "" 3. 局部视图(在HTML页面设置哪些需要缓存):粒度最细 a. 引入TemplateTag {% load cache %} b. 使用缓存 {% cache 300 '缓存key' %} # 缓存key的名字可以是随意的 缓存内容 {% endcache %}
二、 序列化
1、介绍
关于Django中的序列化主要应用在将数据库中检索的数据返回给客户端用户,特别的Ajax请求一般返回的为Json格式。
2、serializers
from django.core import serializers def get_value(request): users = models.User.objects.all() ret = serializers.serialize('json', users) return HttpResponse(ret)
3、自定义序列化
由于json模块并不能转换时间类型的数据,因此需要我们自定义一个类来处理时间类型的数据 import json from datetime import datetime, date data = [ # data数据中有datetime类型的值, json不能直接序列化 {"pk": 1, "name": "\u83b9\u83b9", "age": 18, 'birth': datetime.now()}, {"pk": 2, "name": "\u5c0f\u5fae", "age": 16, 'birth': datetime.now()}, {"pk": 3, "name": "\u5c0f\u9a6c\u54e5", "age": 8, 'birth': datetime.now()}, {"pk": 4, "name": "qqq", "age": 5, 'birth': datetime.now()}, {"pk": 5, "name": "www", "age": 5, 'birth': datetime.now()} ] # json序列化的时候是调用JSONEncoder这个类的default方法进行序列化的 class JsonCustomEncoder(json.JSONEncoder): # 自定义一个类,重新json.dumps的default方法 def default(self, field): # 循环每个字段的值 if isinstance(field, datetime): # 如果这个值是datetime类型,我们自己把它转成字符串类型的时间 return field.strftime('%Y-%m-%d %H:%M:%S') elif isinstance(field, date): # 如果这个值是date类型,我们自己把它转成字符串类型的时间 return field.strftime('%Y-%m-%d') else: return json.JSONEncoder.default(self, field) # 如果这个值不是时间类型,调用其父类原本的default方法进行序列化 print(json.dumps(data,cls=JsonCustomEncoder)) # cls指定序列化的时候去执行这个类
三、 信号
1、介绍
Django中提供了“信号调度”,用于在框架执行操作时解耦。通俗来讲,就是一些动作发生的时候,信号允许特定的发送者去提醒一些接受者。
2、内置信号
Model signals pre_init # django的model执行其构造方法前,自动触发 post_init # django的model执行其构造方法后,自动触发 pre_save # django的model对象保存前,自动触发 post_save # django的model对象保存后,自动触发 pre_delete # django的model对象删除前,自动触发 post_delete # django的model对象删除后,自动触发 m2m_changed # django的model中使用m2m字段操作第三张表(add,remove,clear)前后,自动触发 class_prepared # 程序启动时,检测已注册的app中modal类,对于每一个类,自动触发 Management signals pre_migrate # 执行migrate命令前,自动触发 post_migrate # 执行migrate命令后,自动触发 Request/response signals request_started # 请求到来前,自动触发 request_finished # 请求结束后,自动触发 got_request_exception # 请求异常后,自动触发 Test signals setting_changed # 使用test测试修改配置文件时,自动触发 template_rendered # 使用test测试渲染模板时,自动触发 Database Wrappers connection_created # 创建数据库连接时,自动触发
3、使用
1. 场景:数据库增加一条数据时,就记录一条日志,若不使用信号,则需要在每个创建语句下面写记录日志的语句。 2. 介绍 对于Django内置的信号,仅需注册指定信号,当程序执行相应操作时,自动触发注册函数 注册信号,写入与project同名的文件夹下的_init_.py文件中,也是换数据库引擎的地方。 3. 注册信号步骤 1. 导入需要的信号模块(这里列出全部模块,实际开发的时候需要哪个就导入哪个) from django.core.signals import request_finished from django.core.signals import request_started from django.core.signals import got_request_exception from django.db.models.signals import class_prepared from django.db.models.signals import pre_init, post_init from django.db.models.signals import pre_save, post_save from django.db.models.signals import pre_delete, post_delete from django.db.models.signals import m2m_changed from django.db.models.signals import pre_migrate, post_migrate from django.test.signals import setting_changed from django.test.signals import template_rendered from django.db.backends.signals import connection_created 2. 定义函数来处理信号 # 方法一 from django.db.models.signals import post_save # 函数名可随意,但是参数(sender, **kwargs)是固定的,就这两个参数 def callback(sender, **kwargs): print("xxoo_callback") print(sender, kwargs) post_save.connect(callback) # 注册post_save信号:django的model对象保存后,自动触发callback函数 # post_save信号中,render就是触发信号的那一个ORM类(表) # kwargs就是这个类的一些参数:instance是这个类的实例,created:是否是创建操作 # 方法二 from django.db.models.signals import post_save from django.dispatch import receiver @receiver(post_save) def my_callback(sender, **kwargs): print("xxoo_callback") print(sender, kwargs) # 方法三:指定触发者 from django.db.models.signals import post_save from django.dispatch import receiver from myapp.models import MyModel # 指定只有MyModel这个类才能触发这个函数 @receiver(post_save, sender=MyModel) def my_callback(sender, **kwargs): print("xxoo_callback") print(sender, kwargs) # 或者 post_save.connect(callback, sender=MyModel)
4、自定义信号
a. 定义信号 在某py文件中定义信号。 import django.dispatch # pizza_done是信号名 # providing_args是传给信号绑定的函数的kwargs pizza_done = django.dispatch.Signal(providing_args=["toppings", "size"]) b. 注册信号 在_init_.py 中注册信号 from 路径 import pizza_done def callback(sender, **kwargs): print("callback") print(sender,kwargs) pizza_done.connect(callback) c. 触发信号 from 路径 import pizza_done pizza_done.send(sender='seven',toppings=123, size=456) 由于内置信号的触发者已经集成到Django中,所以其会自动调用,而对于自定义信号则需要开发者在任意位置触发。
四、 ORM性能相关
在ORM中,Queryset使用的是懒查询,就是当你没有使用这个结果的时候,它是不会去数据库帮你查询的,
只有在访问Queryset的内容的时候,Django才会真正进行数据库的访问,因此如果按照默认的查询方式去遍历取值,
那么会造成多次的数据库查询,效率非常低。
1.普通性能相关
表结构
class Role(models.Model): name = models.CharField(max_length=32) class User(models.Model): name = models.CharField(max_length=32) age = models.IntegerField() role = models.ForeignKey('Role', null=True, blank=True)
示例
1、 直接查询--> [ 对象 ] 用的时候注意,只拿自己表中的字段,别跨表,比如all_users有3条数据,user表通过外键关联role表, 如果要跨表拿到role表的name字段: all_users = models.User.objects.all() for user in all_users: print(user.name, user.age, user.role.name) 其实一个进行了四次查询,第一次查询出all_users,然后每次的user.role.name都去role表查 2、 要用到跨表字段的时候,使用values或values_list查询速度更快,只需一次查询即可--> [{}] all_users = models.User.objects.all().values('name','age','role__name') for user in all_users: print(user['name'], user['age'], user['role__name']) 3、 only:将指定的字段查询加载出来,后续再访问指定的字段就不需要再查询数据库 # 用的时候注意,只拿自己指定的字段 all_users = models.User.objects.all().only('name') for user in all_users: print(user.name) # 访问指定的字段name不需要再去查询数据库 for user in all_users: print(user.age) # 访问不是指定的字段,每一次都去查一次数据库 4、 defer:将除了指定的字段查询加载出来,后续再访问指定的字段就不需要再查询数据库(only的反义词) all_users = models.User.objects.all().defer('name') for user in all_users: print(user.name) # 访问指定的字段name,每一次都需要去查询数据库 for user in all_users: print(user.age) # 访问不是指定的字段,不需要再查询数据库
2.连表性能相关
表结构
from django.db import models CHOICES = ((1, "python"), (2, "linux"), (3, "go")) class Publisher(models.Model): id = models.AutoField(primary_key=True) name = models.CharField(max_length=20) def __str__(self): return self.name class Book(models.Model): title = models.CharField(max_length=20) publisher = models.ForeignKey(to='Publisher', on_delete=models.CASCADE, null=True) category = models.IntegerField(choices=CHOICES, default=1) authors = models.ManyToManyField('Author', related_name='books') def __str__(self): return self.title class Author(models.Model): name = models.CharField(max_length=12) # 姓名 gender = models.SmallIntegerField(choices=((1, '男'), (2, '女'), (3, '保密')), default=3) # 性别 info = models.OneToOneField(to='AuthorInfo') # 详细信息 def __str__(self): return self.name # 作者详细信息表 class AuthorInfo(models.Model): city = models.CharField(max_length=12) # 住址
示例
1、select_related 优点:使用连表查询方式(join),从而减少查询次数,最终达到优化和提高性能效果。 缺点:如果关联的表太多,join得到的表太长,会严重影响数据库性能。 应用场景:连表较少的时候使用,如:外键(ForeignKey)、一对一字段(OneToOneField)的正向查询时使用, 反向查询时,如果查询的是多条数据(ForeignKey、ManyToManyField),不能使用select_related, OneToOneField查询的还是单条数据,因此还是可以使用 2、prefetch_related 优点:首先分别查询每个表,获取各个表的数据,存放在内存中,然后通过Python处理他们之间的关联。 应用场景:连表较多的时候使用,如:一对多字段(其实也是ForeignKey),多对多字段(ManyToManyField), 无论是正向查询还是反向查询,都可以使用。 """一对一字段的正向查询""" ret = Author.objects.filter(id=1).select_related('info') for i in ret: print(i.info.city) # select_related:只进行了一次查询 (0.001) SELECT "myapp_author"."id", "myapp_author"."name", "myapp_author"."gender", "myapp_author"."info_id", "myapp_authorinfo"."id", "myapp_authorinfo"."city" FROM "myapp_author" INNER JOIN "myapp_authorinfo" ON ("myapp_author"."info_id" = "myapp_authorinfo"."id") WHERE "myapp_author"."id" >= 1; args=(1,) # prefetch_related:两个表分别查询,进行了两次查询 ret = Author.objects.filter(id=1).prefetch_related('info') for i in ret: print(i.info.city) (0.001) SELECT "myapp_author"."id", "myapp_author"."name", "myapp_author"."gender", "myapp_author"."info_id" FROM "myapp_author" WHERE "myapp_author"."id" >= 1; args=(1,) (0.000) SELECT "myapp_authorinfo"."id", "myapp_authorinfo"."city" FROM "myapp_authorinfo" WHERE "myapp_authorinfo"."id" IN (1, 2); args=(1, 2) """一对一字段的反向查询""" ret = AuthorInfo.objects.filter(id=1).select_related('author') for i in ret: print(i.author.name) (0.000) SELECT "myapp_authorinfo"."id", "myapp_authorinfo"."city", "myapp_author"."id", "myapp_author"."name", "myapp_author"."gender", "myapp_author"."info_id" FROM "myapp_authorinfo" LEFT OUTER JOIN "myapp_author" ON ("myapp_authorinfo"."id" = "myapp_author"."info_id") WHERE "myapp_authorinfo"."id" >= 1; args=(1,) ret = AuthorInfo.objects.filter(id=1).prefetch_related('author') for i in ret: print(i.author.name) (0.000) SELECT "myapp_authorinfo"."id", "myapp_authorinfo"."city" FROM "myapp_authorinfo" WHERE "myapp_authorinfo"."id" >= 1; args=(1,) (0.000) SELECT "myapp_author"."id", "myapp_author"."name", "myapp_author"."gender", "myapp_author"."info_id" FROM "myapp_author" WHERE "myapp_author"."info_id" IN (1, 2); args=(1, 2) """一对多字段的正向查询""" # select_related:只进行了一次查询 ret = Book.objects.filter(id=1).select_related('publisher') for i in ret: print(i.publisher.name) (0.000) SELECT "myapp_book"."id", "myapp_book"."title", "myapp_book"."publisher_id", "myapp_book"."category", "myapp_publisher"."id", "myapp_publisher"."name" FROM "myapp_book" LEFT OUTER JOIN "myapp_publisher" ON ("myapp_book"."publisher_id" = "myapp_publisher"."id") WHERE "myapp_book"."id" >= 1; args=(1,) # prefetch_related:两个表分别查询,进行了两次查询 ret = Book.objects.filter(id=1).prefetch_related('publisher') for i in ret: print(i.publisher.name) (0.001) SELECT "myapp_book"."id", "myapp_book"."title", "myapp_book"."publisher_id", "myapp_book"."category" FROM "myapp_book" WHERE "myapp_book"."id" = 1; args=(1,) (0.000) SELECT "myapp_publisher"."id", "myapp_publisher"."name" FROM "myapp_publisher" WHERE "myapp_publisher"."id" IN (1); args=(1,) """一对多字段的反向查询""" # 查询的是多条数据时,不能使用select_related # prefetch_related:两个表分别查询,进行了两次查询 ret = Publisher.objects.filter(id=1).prefetch_related('book_set') for i in ret: print(i.book_set.all()) (0.000) SELECT "myapp_publisher"."id", "myapp_publisher"."name" FROM "myapp_publisher" WHERE "myapp_publisher"."id" = 1; args=(1,) (0.000) SELECT "myapp_book"."id", "myapp_book"."title", "myapp_book"."publisher_id", "myapp_book"."category" FROM "myapp_book" WHERE "myapp_book"."publisher_id" IN (1); args=(1,) """多对多字段的正向查询""" # 反向查询的是多条数据时,不能使用select_related # prefetch_related:两个表分别查询,进行了两次查询 ret = Book.objects.filter(id=1).prefetch_related('authors') for i in ret: print(i.authors.all()) (0.001) SELECT "myapp_book"."id", "myapp_book"."title", "myapp_book"."publisher_id", "myapp_book"."category" FROM "myapp_book" WHERE "myapp_book"."id" = 1; args=(1,) (0.002) SELECT ("myapp_book_authors"."book_id") AS "_prefetch_related_val_book_id", "myapp_author"."id", "myapp_author"."name", "myapp_author"."gender", "myapp_author"."info_id" FROM "myapp_author" INNER JOIN "myapp_book_authors" ON ("myapp_author"."id" = "myapp_book_authors"."author_id") WHERE "myapp_book_authors"."book_id" IN (1); args=(1,) """多对多字段的反向查询""" ret = Author.objects.filter(id=1).prefetch_related('books') # 设置了related_name for i in ret: print(i.books.all()) (0.000) SELECT "myapp_author"."id", "myapp_author"."name", "myapp_author"."gender", "myapp_author"."info_id" FROM "myapp_author" WHERE "myapp_author"."id" = 1; args=(1,) (0.000) SELECT ("myapp_book_authors"."author_id") AS "_prefetch_related_val_author_id", "myapp_book"."id", "myapp_book"."title", "myapp_book"."publisher_id", "myapp_book"."category" FROM "myapp_book" INNER JOIN "myapp_book_authors" ON ("myapp_book"."id" = "myapp_book_authors"."book_id") WHERE "myapp_book_authors"."author_id" IN (1); args=(1,) 总结: select_related使用的是join查询,但是当连接的表过多时,会导致性能下降, 适用于 "一" 这张表的查询 prefetch_related 是分别对每个表进行查询,然后通过Python处理去处理查询的数据, 无论是 "一" 还是 "多" 这张表,都能使用
五、 验证码
1、随机验证码python代码
import random def get_code(): code = '' for i in range(6): num = str(random.randint(0, 9)) # 数字 lower = chr(random.randint(97, 122)) # 小写字母 upper = chr(random.randint(65, 90)) # 大写字母 c = random.choice([num, lower, upper]) # 随机选取一个 code += str(c) return code
2、如何生成图片
1. 验证码的形式 回想一下,平时我们输入验证码的时候,是不是都是看着一张图片,图片上显示验证码,我们看着图片输入验证码。 当然现在还有滑动的,点击等等,这里我们先学习图片的形式。 2. 实现步骤 1, 准备一张没有任何内容的图片 2, 安装python专门处理图片的第三方包 pip install Pillow 3, 包的导入 from PIL import Image, ImageDraw, ImageFont 4, Image:生成一张图片 ImageDraw:生成一个画笔,用于在图片上画验证码 ImageFont:字体的格式和大小 5,示例 from PIL import Image, ImageDraw, ImageFont # 返回随机的RGB数字 def random_color(): return random.randint(0, 255), random.randint(0, 255), random.randint(0, 255) def get_code(): with open('1.png', 'wb') as f: # 第一步:生成一张图片(画布) # 创建一个随机颜色的图片对象 # 参数:颜色模式,图片大小,图片颜色 img_obj = Image.new('RGB', (250, 35), random_color()) # 第二步:在该图片对象上生成一个画笔对象 draw_obj = ImageDraw.Draw(img_obj) # 使用什么字体,字体大小,kumo.ttf是我本地下载好的字体文件(sc.chinaz.com)可下载 font_obj = ImageFont.truetype('static/font/kumo.ttf', 28) # 生成验证码 code = '' for i in range(6): num = str(random.randint(0, 9)) # 数字 lower = chr(random.randint(97, 122)) # 小写字母 upper = chr(random.randint(65, 90)) # 大写字母 c = random.choice([num, lower, upper]) # 随机选取一个 code += str(c) # 用画笔把验证码画到图片上 # 参数:xy:坐标,画在哪个位置;text:画的内容;fill:画什么颜色;font:字体格式 draw_obj.text((35 + i*30, 0), c, fill=random_color(), font=font_obj) # 保存图片 img_obj.save(f) get_code() 6,缺点 上面的代码是在你的硬盘上存了一张图片,如果要在页面上展示,你还得进行文件的读, 这样的话不仅浪费硬盘空间,效率还不够高,因此我们应该把图片写到内存,从内存中取,效率就快很多了, 然后把图片的验证码数据存到session,这样登录的时候就可以校验了。
3、在视图中使用验证码
1. urls urlpatterns = [ # 获取图片的路由 url(r'^login/', views.login), url(r'^v_code/', views.v_code), ] 2. 在页面中点击验证码图片,刷新验证码 <!DOCTYPE html> <html lang="zh-CN"> <head> <meta http-equiv="content-Type" charset="UTF-8"> <meta http-equiv="x-ua-compatible" content="IE=edge"> <meta name="viewport" content="width=device-width, initial-scale=1"> <title>Title</title> </head> <body> <form action='' method='POST'> {% csrf_token %} <input type='text' name='username'>用户名 <input type='password' name='password'>密码 <img src="/v_code/" alt="图片加载失败" id="v_code"> <button type="submit">登录</button> </form> <script> img = document.getElementById('v_code'); img.onclick = function () { img.src += '?' } </script> </body> </html> 3. 验证码视图函数 from PIL import Image, ImageDraw, ImageFont import random def random_color(): return random.randint(0, 255), random.randint(0, 255), random.randint(0, 255) def v_code(request): # 第一步:生成一张图片(画布) # 创建一个随机颜色的图片对象 # 参数:颜色模式,图片大小,图片颜色 img_obj = Image.new('RGB', (250, 35), random_color()) # 第二步:在该图片对象上生成一个画笔对象 draw_obj = ImageDraw.Draw(img_obj) # 使用什么字体,字体大小 font_obj = ImageFont.truetype('static/font/kumo.ttf', 28) # 生成验证码 code = '' for i in range(6): num = str(random.randint(0, 9)) # 数字 lower = chr(random.randint(97, 122)) # 小写字母 upper = chr(random.randint(65, 90)) # 大写字母 c = random.choice([num, lower, upper]) # 随机选取一个 code += str(c) # 用画笔把验证码画到图片上 # 参数:xy:坐标,画在哪个位置;text:画的内容;fill:画什么颜色;font:字体格式 draw_obj.text((35 + i*30, 0), c, fill=random_color(), font=font_obj) # 把图片里面的验证码的内容写到session,且忽略大小写 request.session['v_code'] = code.upper() # 把图片写到内存 from io import BytesIO f1 = BytesIO() # 类似于文件的文件句柄:f1 = open() # 把图片保存到内存 img_obj.save(f1, format="PNG") # 从内存中取数据 img_data = f1.getvalue() return HttpResponse(img_data, content_type='image/png') 4. 登录视图函数 def login(request): err_msg = '' if request.method == 'POST': username = request.POST.get('username') password = request.POST.get('password') v_code = request.POST.get('v_code', '').upper() if v_code == request.session.get('v_code'): obj = auth.authenticate(request, username=username, password=password) if obj: auth.login(request, obj) # 认证成功 初始化权限信息 ret = init_permission(request, obj) if ret: return ret return redirect(reverse('my_customer')) err_msg = '用户名或密码错误' else: err_msg = '验证码错误' return render(request, 'login.html', {'err_msg': err_msg})
4、验证码的额外小知识
画完验证码后,可以添加一些干扰 就是在 draw_obj.text((35 + i*30, 0), c, fill=random_color(), font=font_obj)之后加 1. 加干扰线 width = 250 # 图片宽度(防止越界) height = 35 for i in range(5): x1 = random.randint(0, width) x2 = random.randint(0, width) y1 = random.randint(0, height) y2 = random.randint(0, height) draw_obj.line((x1, y1, x2, y2), fill=random_color()) 2. 加干扰点 for i in range(40): draw_obj.point([random.randint(0, width), random.randint(0, height)], fill=random_color()) x = random.randint(0, width) y = random.randint(0, height) draw_obj.arc((x, y, x+4, y+4), 0, 90, fill=random_color())