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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
import random, datetime, os
from datetime import datetime, timedelta
from django.db import models
from django.utils import timezone
from django.contrib.auth.models import ContentType
from django.contrib.auth.models import User, Group, Permission
from sbhs_server import settings
# from django.conf import settings
MOD_GROUP_NAME = 'moderator'
def create_group(group_name, app_label):
try:
group = Group.objects.get(name=group_name)
except Group.DoesNotExist:
group = Group(name=group_name)
group.save()
# Get the models for the given app
content_types = ContentType.objects.filter(app_label=app_label)
permission_list = Permission.objects.filter(
content_type__in=content_types
)
group.permissions.add(*permission_list)
group.save()
return group
class Board(models.Model):
""" SBHS Board attributes"""
mid = models.IntegerField(unique=True)
online = models.BooleanField(default=False)
usb_id = models.IntegerField(default="0")
raspi_path = models.URLField(max_length=200, default="localhost:8000")
def save_board_details(self, raspi_ip, usb_mac_map):
for devices in usb_mac_map:
board = Board.objects.filter(mid=devices["sbhs_mac_id"])
if board.exists():
board = board.first()
else:
board = self
board.mid = devices["sbhs_mac_id"]
board.raspi_path = raspi_ip
board.usb_path = int(devices["usb_id"])
board.online = True
board.save()
def switch_off_inactive_boards(self, online_mids):
all_boards = Board.objects.all()
for board in all_boards:
if board.mid not in online_mids:
board.online = False
elif board.mid == online_mids:
board.online = True
board.save()
class Meta:
ordering = ['mid',]
def __str__(self):
return '{}: {}'.format(self.mid, self.online)
class Profile(models.Model):
"""
Profile model to store user details.
"""
user = models.OneToOneField(User)
roll_number = models.CharField(max_length=20)
institute = models.CharField(max_length=128)
department = models.CharField(max_length=64)
position = models.CharField(max_length=64)
is_moderator = models.BooleanField(default=False)
is_email_verified = models.BooleanField(default=False)
activation_key = models.CharField(max_length=255,blank=True,null=True)
key_expiry_time = models.DateTimeField(blank=True,null=True)
def _toggle_moderator_group(self, group_name):
group = Group.objects.get(name=group_name)
if self.is_moderator:
self.user.groups.add(group)
else:
self.user.groups.remove(group)
def save(self, *args, **kwargs):
if self.pk is not None:
old_profile = Profile.objects.get(pk=self.pk)
if old_profile.is_moderator != self.is_moderator:
self._toggle_moderator_group(group_name=MOD_GROUP_NAME)
super(Profile, self).save(*args, **kwargs)
def __str__(self):
return '%s' % (self.user)
class SlotManager(models.Manager):
def get_current_slots(self, user):
now = timezone.now()
slots = self.filter(user=user, start_time__lte=now, end_time__gt=now)
return slots
class Slot(models.Model):
user = models.ForeignKey(User)
start_time = models.DateTimeField("Start time of a slot",
default=timezone.now())
end_time = models.DateTimeField("End time of a slot",
default=timezone.now()+timedelta(
minutes=settings.SLOT_DURATION))
objects = SlotManager()
def __str__(self):
return '{} {}'.format(self.start_time, self.end_time)
class Experiment(models.Model):
slot = models.ForeignKey("Slot")
log = models.CharField(max_length=255)
checksum = models.CharField(max_length=255, null=True, blank=True)
class UserBoard(models.Model):
user = models.ForeignKey(User)
board = models.ForeignKey(Board)
|
