# -*- coding: utf-8 -*-
"""
Models for lecture recorder
"""
import json
import logging

from sqlalchemy import MetaData, CheckConstraint
from sqlalchemy.exc import IntegrityError

from datetime import datetime

from backend import db
from backend.models.recorder_model import Recorder
from backend.tools.scrape_rooms import scrape_rooms

logger = logging.getLogger("lrc." + __name__)

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=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=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)

    __table_args__ = (CheckConstraint("length(name) > 2", name="name_min_length"),)

    def __init__(self, **kwargs):
        db.Model.__init__(self, **kwargs)

    @staticmethod
    def get_by_name(name):
        """
        Find group by name
        :param name:
        :return:
        """
        return Room.query.filter(Room.name == name).first()

    @staticmethod
    def get_by_building_number(building_number):
        """
        Find group by name
        :param name:
        :return:
        """
        return Room.query.filter(Room.building_number == building_number)

    @staticmethod
    def get_all():
        """
        Return all rooms
        :return:
        """
        return Room.query.all()

    def __str__(self):
        return self.name

    def to_dict(self):
        return dict(id=self.id, name=self.name)

    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()
    logger.debug("tada")
    logger.debug("got {} rooms".format(len(rooms)))
    db_rooms = [
        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(db_rooms)
        db.session.commit()
        logger.debug("rooms commited to DB!")
    except IntegrityError as e:
        logger.error("Could not add rooms! ({})".format(e))
        db.session.rollback()