利用decorator实现Django表单防重复提交

最新推荐文章于 2021-01-30 05:52:33 发布

lbxhappy

最新推荐文章于 2021-01-30 05:52:33 发布

阅读量226

点赞数

CC 4.0 BY-SA版权

分类专栏： python django 文章标签： Django Python Cache

本文链接：https://blog.youkuaiyun.com/lbxhappy/article/details/83814812

python django 专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种防止Web表单重复提交的方法，通过生成随机校验字符串并利用session进行比对，确保用户不会因误操作而产生多余的数据记录。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

背景：

　　我的用例中不可出现重复的记录，如：下订单，用户填好表单然后Submit，当用户网速较慢时，很可能会习惯性的刷新当前页，而刷新操作会导致再次POST，此时若不加判断直接入库必然导致用户后台增加N个订单。

解决思路：

　　　Step 1：首先我们在进入表单填写页面时，对该页面（或view吧）随机生成一个校验字符串并存放于Session中，在页面form中新增一个hidden input来存放，以便提交时一起POST到服务器端；

　　　Step 2：当POST时，服务器端将收到的校验串与Session中对应值比对，若相同则正常提交，否则跳转至指定的错误提示页面；需要注意的是当比对成功后，必须立即将该Session的值设置为空，这样才可保证页面再次POST时表单中的校验串就无法于Session中的空值比对成功。

实现：

　　为了方便使用，我们将上述思路写成一个decorator(装饰器). 当然也是为了符合DRY嘛；代码很简单就看中间那几行，需注意唯一的一个参数page_key,为了不跟多个表单页面发生Session key冲突。（补充一点：必须将表单填写页面的view同时使用@never_cache装饰，因为django默认将所有view都做缓存，当再次进入表单页时，就不会重新生成随机串，导致校验无故失败。。。）

 
     # coding:utf-8 

     try : 

          from functools  import wraps 

     except ImportError: 

          from django.utils.functional  import wraps   # Python 2.4 fallback. 

     import random 

     from django.conf  import settings 

     from django.utils.decorators  import available_attrs 

     from django.utils.hashcompat  import md5_constructor 

     if hasattr (random,  'SystemRandom' ): 

          randrange  = random.SystemRandom().randrange 

     else : 

          randrange  = random.randrange 

     _MAX_CSRF_KEY  = 18446744073709551616L      # 2 << 63 

     def _get_new_submit_key(): 

          return md5_constructor( "%s%s" % (randrange( 0 , _MAX_CSRF_KEY), settings.SECRET_KEY)).hexdigest() 

     def anti_resubmit(page_key = ''): 

          def decorator(view_func): 

              @wraps (view_func, assigned = available_attrs(view_func)) 

              def _wrapped_view(request,  * args,  * * kwargs): 

                  if request.method  = = 'GET' : 

                      request.session[ '%s_submit' % page_key]  = _get_new_submit_key() 

                      print 'session:' + request.session.get( '%s_submit' % page_key) 

                  elif request.method  = = 'POST' : 

                      old_key  = request.session.get( '%s_submit' % page_key, '') 

                      if old_key  = = '': 

                          from django.http  import HttpResponseRedirect 

                          return HttpResponseRedirect( '/page_expir' ) 

                      request.session[ '%s_submit' % page_key]  = '' 

                  return view_func(request,  * args,  * * kwargs) 

              return _wrapped_view 

          return decorator

　　使用示例：

1 @anti_resubmit(page_key = ' your_view ' )
2 def your_view(request):
3 ''' 若是表单填写页和POST的view不是同一个，則需在两个view上都使用anti_resubmit装饰器 '''
4 pass # 您可别跟着pass噢