-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
common.py
43 lines (36 loc) · 1.34 KB
/
common.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#! /usr/bin/env python
# -*- coding: utf-8 -*-
from django.http import HttpResponse
from config.views import get_dir
import json
import redis
def token_verify():
def decorator(view_func):
def _wrapped_view(request, *args, **kwargs):
set_token = get_dir('token')
error_info = "Post forbidden, your token error!!"
if request.method == 'POST':
post_token = json.loads(request.body)
if set_token == post_token["token"]:
return view_func(request, *args, **kwargs)
else:
return HttpResponse(error_info, status=403)
if request.GET:
post_token = request.GET['token']
if set_token == post_token:
return view_func(request, *args, **kwargs)
else:
return HttpResponse(error_info, status=403)
return HttpResponse(error_info, status=403)
return _wrapped_view
return decorator
class GetRedis(object):
host = get_dir("redis_host")
port = get_dir("redis_port")
db = get_dir("redis_db")
password = get_dir("redis_password")
@classmethod
def connect(cls):
conn = redis.StrictRedis(host=cls.host, port=cls.port,
password=cls.password, db=cls.db)
return conn