【分享】利用decorator实现Django表单防重复提交

背景：

 

 　　我的用例中不可出现重复的记录，如：下订单，用户填好表单然后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

 

 

 　　使用示例：

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

 

 

本博客所有原创内容均可随意转载，但请注明来源！