diff options
| -rw-r--r-- | sbhs/models.py | 86 |
1 files changed, 28 insertions, 58 deletions
diff --git a/sbhs/models.py b/sbhs/models.py index 4c1205a..34fdb7e 100644 --- a/sbhs/models.py +++ b/sbhs/models.py @@ -1,17 +1,20 @@ import random, datetime, os +from datetime import datetime, timedelta from django.db import models -from django.contrib.auth.models import User, Group, Permission +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) + group = Group.objects.get(name=group_name) except Group.DoesNotExist: group = Group(name=group_name) group.save() @@ -30,48 +33,25 @@ class Board(models.Model): mid = models.IntegerField(unique=True) online = models.BooleanField(default=False) - # 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 = [int(x) for x in settings.online_mids] - # online_boards = sorted(online_boards) - - # # 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 int(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" + '0'+str(self.mid) + ".jpeg" + 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) + 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) + 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) @@ -92,32 +72,22 @@ class Profile(models.Model): class Slot(models.Model): user = models.ForeignKey(User) - start_time = models.DateTimeField("Start time of a slot") - duration = models.IntegerField(default=settings.SLOT_DURATION) + 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)) def __str__(self): - return '{} {}'.format(self.start_time, self.duration) + return '{} {}'.format(self.start_time, self.end_time) + + def slots_now(): + now = datetime.datetime.now() + slots = Slot.objects.filter(start_time=now) + return slots 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 Webcam(): -# """ -# Utility function to capture webcam feeds using streamer -# """ -# def __init__(self): -# pass - -# @classmethod -# def load_image(className,mid): - -# if int(mid) : -# command = "timeout 2s streamer -q -f jpeg -c /dev/video" + '0'+str(mid) -# print 'command1', command -# command += " -o " + settings.WEBCAM_DIR + "image" + '0'+str(mid) + ".jpeg" -# print 'command2', command -# os.system(command) -#
\ No newline at end of file |