mixins.py
2.65 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
93
94
95
from django.conf import settings
import json
from .models import Action, Resource, Action_Resource, Log, Notification
from users.models import User
class LogMixin(object):
log_component = ""
log_context = {}
log_action = ""
log_resource = ""
def createLog(self, actor = None, component = '', log_action = '', log_resource = '', context = {}):
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 = actor
#log.context = json.dumps(context)
log.context = context
log.component = component
log.action_resource = action_resource
log.save()
def dispatch(self, request, *args, **kwargs):
return super(LogMixin, self).dispatch(request, *args, **kwargs)
class NotificationMixin(object):
message = ""
read = False
action_slug = ''
resource_name = ''
def createNotification(self, message='', actor=None, users = User.objects.all(), resource_slug='' ,action_slug = '',
resource_name='', resource_link=''): #the default will be a broadcast
action = Action.objects.filter(slug = action_slug)
resource = Resource.objects.filter(slug = resource_slug)
if action.exists():
action = action[0]
else:
action = Action(name = action_slug)
action.save()
if resource.exists():
resource = resource[0]
resource.url = resource_link
resource.save()
else:
resource = Resource(name = resource_name, url= resource_link)
resource.save()
action_resource = Action_Resource.objects.filter(action = action, resource = resource)
if action_resource.exists():
action_resource = action_resource[0]
else:
action_resource = Action_Resource(action = action, resource = resource)
action_resource.save()
for user in users:
notification = Notification(user=user, actor= actor, message=message, action_resource= action_resource)
notification.save()
def dispatch(self, request, *args, **kwargs):
"""
Not quite sure how to do about it"""
return super(NotificationMixin, self).dispatch(request, *args, **kwargs)
def createorRetrieveAction(self, action_name):
action = Action.objects.filter(name=action_name)
if action is None:
action = Action(name=action_name)
return action