now scraping rooms from capmus mgmt

models/room_model.py

@@ -5,13 +5,12 @@ Models for lecture recorder
 import json
 
 from sqlalchemy import MetaData, CheckConstraint
+from sqlalchemy.exc import IntegrityError
+
+from datetime import datetime, timedelta
 
 from backend import db, app, login_manager
-import re
-from sqlalchemy import or_
-from datetime import datetime, timedelta
-from backend.models.recorder_model import Recorder
-
+from backend.tools.scrape_rooms import scrape_rooms
 
 metadata = MetaData()
 
@@ -19,10 +18,12 @@ metadata = MetaData()
 class Room(db.Model):
     id = db.Column(db.Integer, autoincrement=True, primary_key=True)
     created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow())
-    name = db.Column(db.Unicode(127), unique=True, nullable=False)
-    alternate_name = db.Column(db.Unicode(127), unique=True, nullable=True, default=None)
+    name = db.Column(db.Unicode(127), unique=False, nullable=False)
+    alternate_name = db.Column(db.Unicode(127), unique=False, nullable=True, default=None)
     comment = db.Column(db.Unicode(2047), unique=False, nullable=True, default="")
-    number = db.Column(db.Unicode(63), unique=True, nullable=True)
+    number = db.Column(db.Unicode(63), unique=False, nullable=True)
+    building_name = db.Column(db.Unicode(63), unique=False, nullable=True)
+    building_number = db.Column(db.Unicode(63), unique=False, nullable=True)
 
     recorder = db.relationship('Recorder', uselist=False, back_populates='room')  # one-to-one relation (uselist=False)
 
@@ -34,7 +35,6 @@ class Room(db.Model):
     def __init__(self, **kwargs):
         super(Room, self).__init__(**kwargs)
 
-
     @staticmethod
     def get_by_name(name):
         """
@@ -47,7 +47,7 @@ class Room(db.Model):
     @staticmethod
     def get_all():
         """
-        Return all groups
+        Return all rooms
         :return:
         """
         return Room.query.all()
@@ -61,3 +61,16 @@ class Room(db.Model):
     def toJSON(self):
         return json.dumps(self.to_dict(), default=lambda o: o.__dict__,
                           sort_keys=True, indent=4)
+
+
+def pre_fill_table():
+    rooms = scrape_rooms()
+    i_tunes_u_mappings = [Room(name=room['name'], number=room['room_number'],
+                               building_name=room['building_name'], building_number=room['building_number']) for room in
+                          rooms]
+
+    try:
+        db.session.bulk_save_objects(i_tunes_u_mappings)
+        db.session.commit()
+    except IntegrityError as e:
+        db.session.rollback()