Tooltron

from django.db import models

from django.contrib.auth.models import User

from django.db.models.signals import post_save

# Machine Model

class Machine(models.Model):

  type = models.CharField(max_length=20)

  id = models.CharField(max_length=10, primary_key=True)

  maint = models.BooleanField(default=False)

  dstart = models.DateTimeField(blank=True, null=True)

  dend = models.DateTimeField(blank=True, null=True)

  def __unicode__(self):

    return u'%s %s' % (self.type, self.id)

# User Model

class RoboUser(models.Model):

  # Field is required when using profiles 

  user = models.OneToOneField(User)

  # Roboclub RFID Card number

  rfid = models.CharField(max_length=10, blank=True, default=0)

  # Roboclub Shop Access Permissions

  machines = models.ManyToManyField(Machine, blank=True, null=True)

  # Cell Phone

  cell = models.DecimalField(max_digits=10, decimal_places=0, blank=True, null=True)

  # Class Level

  UNDERGRAD = 'UG'

  GRADUATE = 'GR'

  AFFILIATE = 'AF'

  OTHER = 'OH'

  CLASS_LEVEL_CHOICES = (

        (UNDERGRAD, 'Undergraduate'),

        (GRADUATE, 'Graduate Student'),

        (AFFILIATE, 'Non-Student CMU Affiliate'),

        (OTHER, 'Other User'),

  )

  class_level = models.CharField(max_length=2, 

                                 choices=CLASS_LEVEL_CHOICES,

                                 default=UNDERGRAD)

  # Graduation Year

  #grad_year = models.DecimalField(max_digits=4, decimal_places=0)

  grad_year = models.IntegerField(blank=True, null=True)

  # Primary and Secondary Major/Minors

  major = models.CharField(max_length=20)

  sec_major_one = models.CharField(max_length=20, blank=True, null=True)

  sec_major_two = models.CharField(max_length=20, blank=True, null=True)

  #Club Rank

  JUNIOR_MEM = 'JM'

  SENIOR_MEM = 'SM'

  OFFICER = 'OM'

  CLUB_RANK_CHOICES = (

      (JUNIOR_MEM, 'Junior Member'),

      (SENIOR_MEM, 'Senior Member'),

      (OFFICER, 'Officer'),

  )

  club_rank = models.CharField(max_length=2, 

                               choices=CLUB_RANK_CHOICES,

                               default=JUNIOR_MEM)

  # Roboclub Transaction Info

  dues_paid = models.DateField(blank=True, null=True)

  tshirt_rec = models.BooleanField(default=False)

  # Shop and E-Bench Status

  GOOD = 'GD'

  FIRST_WARN = 'FS'

  SECOND_WARN = 'SD'

  SEM_BAN = 'SB'

  CLUB_BAN = 'CB'

  STATUS_CHOICES = (

      (GOOD, 'Good Standing'),

      (FIRST_WARN, 'First Warning Recieved'),

      (SECOND_WARN, 'Second Warning Recieved'),

      (SEM_BAN, 'Semester Ban'),

      (CLUB_BAN, 'Club Ban')

  )

  bench_status = models.CharField(max_length=2,

                                  choices=STATUS_CHOICES,

                                  default=GOOD)

  shop_status = models.CharField(max_length=2,

                                  choices=STATUS_CHOICES,

                                  default=GOOD)

  def __unicode__(self):

    return self.user.username;

# needed for Django Auth model

def create_roboclub_user(sender, instance, created, **kwargs):

  if created:

      RoboUser.objects.create(user=instance)

post_save.connect(create_roboclub_user, sender=User)

# Event Model

class Event(models.Model):

  type = models.CharField(max_length=30)

  tstart = models.DateTimeField()

  tend = models.DateTimeField()

  user = models.ForeignKey('RoboUser')

  succ = models.BooleanField(default=False)

  imgurl = models.URLField()

  machine = models.ForeignKey('Machine')

  project = models.ForeignKey('Project')

  matuse = models.TextField()

  def __unicode__(self):

    return u'%s %s %s' (self.type, self.user.username, self.succ)

# Project Model

class Project(models.Model):

  name = models.CharField(max_length=30)

  primuser = models.ForeignKey('RoboUser', related_name='pri+')

  users = models.ManyToManyField('RoboUser', related_name='u+')

  charge = models.BooleanField(default=False)

  def __unicode__(self):

    return self.name

# Roboclub Resources Model

class RoboResource(models.Model):

  type = models.CharField(max_length=30)

  id = models.CharField(max_length=20, primary_key=True)

  checked_out = models.BooleanField(default=False)

  user = models.ForeignKey('RoboUser', related_name='u+', blank=True)

  time_out = models.DateTimeField(blank=True,null=True)

  time_due = models.DateTimeField(blank=True,null=True)

  officer = models.ForeignKey('RoboUser', related_name='o+', blank=True)

  def __unicode__(self):

    return u'%s %s %s' (self.type, self.id, self.checked_out)