From 8ac52bd600bf9550a1a32d1652dcdfcf49a1fa4d Mon Sep 17 00:00:00 2001
From: MG <michael-guenther.goetz@stud.uni-bamberg.de>
Date: Fri, 8 Jun 2018 22:30:22 +0200
Subject: [PATCH] Init models

---
 roofis2/roomservice/admin.py  | 10 ++++
 roofis2/roomservice/models.py | 90 +++++++++++++++++++++++++++++++++++
 2 files changed, 100 insertions(+)

diff --git a/roofis2/roomservice/admin.py b/roofis2/roomservice/admin.py
index 8c38f3f..5ef9967 100644
--- a/roofis2/roomservice/admin.py
+++ b/roofis2/roomservice/admin.py
@@ -1,3 +1,13 @@
 from django.contrib import admin
+from .models import Staff, Room, RoomType, BookingGroup, Booking, Equipment, Location, Buildings, NumEquipment
 
 # Register your models here.
+admin.site.register(Staff)
+admin.site.register(Room)
+admin.site.register(RoomType)
+admin.site.register(BookingGroup)
+admin.site.register(Booking)
+admin.site.register(Equipment)
+admin.site.register(Location)
+admin.site.register(Buildings)
+admin.site.register(NumEquipment)
diff --git a/roofis2/roomservice/models.py b/roofis2/roomservice/models.py
index 71a8362..dd6c465 100644
--- a/roofis2/roomservice/models.py
+++ b/roofis2/roomservice/models.py
@@ -1,3 +1,93 @@
 from django.db import models
+from django.contrib.auth.models import User
+
 
 # Create your models here.
+class Location(models.Model):
+    name = models.CharField(max_length=16)
+
+    def __str__(self):
+        return '{}'.format(self.name)
+
+
+class Buildings(models.Model):
+    name = models.CharField(max_length=16)
+    location = models.ForeignKey(Location, on_delete=models.CASCADE)
+
+    def __str__(self):
+        return '{}'.format(self.name)
+
+
+class RoomType(models.Model):
+    type = models.CharField(max_length=32)
+
+    def __str__(self):
+        return '{}'.format(self.type)
+
+
+class BookingGroup(models.Model):
+    name = models.CharField(max_length=64)
+
+    def __str__(self):
+        return '{}'.format(self.name)
+
+
+class Staff(models.Model):
+    booking_group = models.ManyToManyField(BookingGroup)
+    # username = personal_id
+    user = models.OneToOneField(User, on_delete=models.CASCADE)
+
+    def __str__(self):
+        return '{}'.format(self.user.username)
+
+
+class Room(models.Model):
+    building = models.ForeignKey(Buildings, on_delete=models.CASCADE)
+    room_number = models.CharField(max_length=16)
+    capacity = models.IntegerField()
+    seating = models.BooleanField()
+    barrier_free = models.BooleanField()
+    cooling = models.BooleanField()
+    room_type = models.ForeignKey(RoomType, on_delete=models.PROTECT)
+    floor = models.SmallIntegerField()
+    admin = models.ForeignKey(BookingGroup, on_delete=models.SET_DEFAULT, default=1)
+    service_staff = models.ForeignKey(Staff, on_delete=models.PROTECT)
+
+    def __str__(self):
+        return '{} - {}'.format(self.building.name, self.room_number)
+
+
+class Equipment(models.Model):
+    name = models.CharField(max_length=32)
+
+    def __str__(self):
+        return '{}'.format(self.name)
+
+
+class NumEquipment(models.Model):
+    room = models.ForeignKey(Room, on_delete=models.CASCADE)
+    equipment = models.ForeignKey(Equipment, on_delete=models.CASCADE)
+    count = models.SmallIntegerField()
+
+    def __str__(self):
+        return '{} - {} - {}'.format(self.room.room_number, self.equipment.name, self.count)
+
+
+class Booking(models.Model):
+    DAY = 0
+    WEEK = 1
+    EVERY_TWO_WEEKS = 2
+    MONTH = 3
+
+    TYPE_CHOICES = ((DAY, 'DAY'), (WEEK, 'WEEK'), (EVERY_TWO_WEEKS, 'EVERY_TWO_WEEKS'), (MONTH, 'MONTH'))
+    room = models.ForeignKey(Room, on_delete=models.CASCADE)
+    staff = models.ForeignKey(Staff, on_delete=models.CASCADE)
+    start_date = models.DateField()
+    end_date = models.DateField()
+    start_time = models.TimeField()
+    end_time = models.TimeField()
+    intervall = models.IntegerField(choices=TYPE_CHOICES)
+
+    def __str__(self):
+        return '{} - {} - {}'.format(self.room.room_number, self.start_date.strftime('%Y - %m - %d'),
+                                     self.end_date.strftime('%Y - %m - %d'))