decorators.py
2.41 KB
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
from django.conf import settings
from functools import wraps
from .models import Action, Resource, Action_Resource, Log, Notification
def log_decorator(log_action = '', log_resource = ''):
def _log_decorator(view_function):
def _decorator(request, *args, **kwargs):
response = view_function(request, *args, **kwargs)
if request.user.is_authenticated and request.POST:
action = Action.objects.filter(name = log_action)
resource = Resource.objects.filter(name = log_resource)
if not action:
action = Action(name = log_action)
action.save()
else:
action = action[0]
if not resource:
resource = Resource(name = log_resource)
resource.save()
else:
resource = resource[0]
action_resource = Action_Resource.objects.filter(action = action, resource = resource)
if not action_resource:
action_resource = Action_Resource(action = action, resource = resource)
action_resource.save()
else:
action_resource = action_resource[0]
log = Log()
log.user = request.user
log.action_resource = action_resource
log.save()
return response
return wraps(view_function)(_decorator)
return _log_decorator
def notification_decorator(read = False, message = '', actor = None, users = [], not_action='', not_resource=''):
def _notification_decorator(view_function):
def _decorator(request, *args, **kwargs):
#Do something before the call
response = view_function(request, *args, **kwargs)
action = Action.objects.filter(name = not_action)
resource = Resource.objects.filter(name = not_resource)
if not action:
action = Action(name = not_action)
action.save()
else:
action = action[0]
if not resource:
resource = Resource(name = not_resource)
resource.save()
else:
resource = resource[0]
action_resource = Action_Resource.objects.filter(action = action, resource = resource)
if not action_resource:
action_resource = Action_Resource(action = action, resource = resource)
action_resource.save()
else:
action_resource = action_resource[0]
if request.user.is_authenticated: #the user was authenticated by the view
notification = Notification(actor = request.user, message= message,
action_resource = action_resource, user = request.user)
#Do something after the call
return response
return wraps(view_function)(_decorator)
return _notification_decorator