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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
|
from django.db import models
from django.contrib.auth.models import AbstractBaseUser
from undelete.models import TrashableMixin
import random, datetime, os
from sbhs_server.helpers import mailer, simple_encrypt
from django.contrib.auth.models import UserManager
from sbhs_server import settings
from django.core.exceptions import ObjectDoesNotExist
#from yaksh.models import Profile
# Create your models here.
class Board(TrashableMixin):
""" Declaring varibales
mid = Integer and unique
online = Boolean, Default - True
temp_offline = Boolean, default - False
created_at = DateTime
updated_at = DateTime
"""
mid = models.IntegerField(unique=True)
online = models.BooleanField(default=True)
temp_offline = models.BooleanField(default=False)
created_at = models.DateTimeField(auto_now_add=True, editable=False)
updated_at = models.DateTimeField(auto_now=True, editable=False)
@staticmethod
def can_do_random_allotment():
return not os.path.exists(os.path.join(settings.BASE_DIR, "WORKSHOP_MODE"))
@staticmethod
def toggle_random_allotment():
""" Enables to toggle allotment of machines between Workshop mode and Random mode
"""
if Board.can_do_random_allotment():
f = open(os.path.join(settings.BASE_DIR, "WORKSHOP_MODE"), "w")
f.close()
else:
os.remove(os.path.join(settings.BASE_DIR, "WORKSHOP_MODE"))
@staticmethod
def allot_board():
"""
"""
if Board.can_do_random_allotment():
online_boards_count = len(settings.online_mids)
board_num = random.randrange(online_boards_count)
return settings.online_mids[board_num]
else:
online_boards = sorted(settings.online_mids)
# When the account table is empty, allocate first board
try:
last_allocated_MID = Account.objects.select_related().order_by("-id")[0].board.mid;
for o in online_boards:
if o > last_allocated_MID:
return Board.objects.get(mid=o).id
except Exception as e:
pass
# check if there is at least one online board
try:
return Board.objects.get(mid=online_boards[0]).id
except Exception as e:
return -1
def image_link(self):
""" Function to show the image obtained from webcam
"""
return settings.WEBCAM_STATIC_DIR + "image" + str(self.mid) + ".jpeg"
class Account(TrashableMixin, AbstractBaseUser):
name = models.CharField(max_length=255)
username = models.CharField(max_length=127, unique=True)
email = models.EmailField(max_length=255, unique=True)
# password = models.CharField(max_length=255) # Already covered in AbstractBaseUser
is_active = models.BooleanField(default=False)
is_admin = models.BooleanField(default=False)
board = models.ForeignKey("Board")
created_at = models.DateTimeField(auto_now_add=True, editable=False)
updated_at = models.DateTimeField(auto_now=True, editable=False)
objects = UserManager()
USERNAME_FIELD = "username"
def get_full_name(self):
return self.name
def get_short_name(self):
return self.name
def send_confirmation(self):
""" Function to show message to the user to confirm their account by visiting the link sent
to their e-mail.
"""
message = """Hi,\n\nPlease visit the link """ + settings.BASE_URL + """account/confirm/"""
message = message + self.confirmation_token()
message = message + """ to confirm your account.\n\n\nRegards,\nVlabs Team"""
mailer.email(self.email, "Please confirm your account", message)
def send_password_link(self, token):
""" Function to show message to user to visit the link in order to change the password.
"""
message = """Hi,\n\nPlease visit the link """ + settings.BASE_URL + """password/edit/"""
message = message + token
message = message + """ to change your password.\n\n\nRegards,\nVlabs Team"""
mailer.email(self.email, "SBHS vlabs password reset link", message)
def get_profile(self):
return self.profile
def confirmation_token(self):
return simple_encrypt.encrypt(self.email)
class Slot(TrashableMixin):
start_hour = models.IntegerField()
start_minute = models.IntegerField()
end_hour = models.IntegerField()
end_minute = models.IntegerField()
created_at = models.DateTimeField(auto_now_add=True, editable=False)
updated_at = models.DateTimeField(auto_now=True, editable=False)
def __str__(self):
sh = str(self.start_hour) if len(str(self.start_hour)) == 2 else "0" + str(self.start_hour)
sm = str(self.start_minute) if len(str(self.start_minute)) == 2 else "0" + str(self.start_minute)
eh = str(self.end_hour) if len(str(self.end_hour)) == 2 else "0" + str(self.end_hour)
em = str(self.end_minute) if len(str(self.end_minute)) == 2 else "0" + str(self.end_minute)
return sh + ":" + sm + " -- " + eh + ":" + em
@staticmethod
def indices(self, other):
""" now = datetime.datetime.now()
cur_hour = now.hour
s = abs(cur_hour - self.start_hour)
s = s + 100 if self.start_hour < cur_hour else s
o = abs(cur_hour - other.start_hour)
o = o + 100 if other.start_hour < cur_hour else o
"""
return self.start_hour, other.start_hour
def __lt__(self, other):
s, o = Slot.indices(self, other)
return s < o
def __gt__(self, other):
s, o = Slot.indices(self, other)
return s > o
def __ne__(self, other):
s, o = Slot.indices(self, other)
return s != o
def __eq__(self, other):
s, o = Slot.indices(self, other)
return s == o
def __le__(self, other):
s, o = Slot.indices(self, other)
return s <= o
def __ge__(self, other):
s, o = Slot.indices(self, other)
return s >= o
@staticmethod
def current_slots(mid):
""" Function to display the current slots to the user.
Input: mid
Output: slots
"""
now = datetime.datetime.now()
slots = list(Slot.objects.filter(start_hour=now.hour, end_minute__gt=now.minute))
bookings = Booking.objects.filter(booking_date__year=now.year,
booking_date__month=now.month,
booking_date__day=now.day, slot__in=slots,
account__board__mid=mid)
for b in bookings:
if b.slot in slots:
slots.remove(b.slot)
return slots
@staticmethod
def slots_now():
now = datetime.datetime.now()
slots = list(Slot.objects.filter(start_hour=now.hour))
return slots
@staticmethod
def get_free_slots(mid):
now = datetime.datetime.now()
slots = list(Slot.objects.filter(start_hour__gte=now.hour))
bookings = Booking.objects.filter(booking_date__year=now.year,
booking_date__month=now.month,
booking_date__day=now.day, slot__in=slots,
account__board__mid=mid)
for b in bookings:
if b.slot in slots:
slots.remove(b.slot)
return sorted(slots)
@staticmethod
def get_free_slots_on(date, mid):
now = datetime.datetime.now()
if date.strftime("%Y-%m-%d") == now.strftime("%Y-%m-%d"):
slots = list(Slot.get_free_slots(mid))
elif date > now:
slots = list(Slot.objects.all())
bookings = Booking.objects.filter(booking_date__year=date.year,
booking_date__month=date.month,
booking_date__day=date.day,
account__board__mid=mid)
for b in bookings:
if b.slot in slots:
slots.remove(b.slot)
else:
slots = []
return sorted(slots)
class Booking(TrashableMixin):
account = models.ForeignKey("Account")
slot = models.ForeignKey("Slot")
booking_date = models.DateTimeField()
created_at = models.DateTimeField(auto_now_add=True, editable=False)
updated_at = models.DateTimeField(auto_now=True, editable=False)
def start_time(self):
return self.booking_date.replace(hour=self.slot.start_hour, minute=self.slot.start_minute)
def end_time(self):
return self.booking_date.replace(hour=self.slot.end_hour, minute=self.slot.end_minute)
class Experiment(TrashableMixin):
booking = models.ForeignKey("Booking")
log = models.CharField(max_length=255)
checksum = models.CharField(max_length=255, default="NONE")
created_at = models.DateTimeField(auto_now_add=True, editable=False)
updated_at = models.DateTimeField(auto_now=True, editable=False)
|
