madhav-datt/kgp-hms

#
# IIT Kharagpur - Hall Management System
# System to manage Halls of residences, Warden grant requests, student complaints
# hall worker attendances and salary payments
#
# MIT License
#

"""
@ authors: Madhav Datt, Avikalp Srivastava
"""

from ..database import db_func as db
from ..database import password_validation as pv
from ..workers import attendant


class Warden(object):
    """Contains details of Warden

    Attributes:
        warden_ID: Integer to uniquely identify warden
        password: Hashed string after adding salt
        name: String
        email: String
        hall_ID: Integer to identify hall of residence
    """

    def __init__(self, password, name, email, hall_ID, controlling_warden=False,
                 rebuild=False, warden_ID=None):
        """
        Init Warden with details for object creation
        """

        # The rebuild flag, if true, denotes that the object is being made from
        # data already present in the database
        # If False, a new data row is added to the specific table
        if not rebuild:
            self.warden_ID = db.add("warden")
            self.password = password
            self.name = name
            self.email = email
            self.hall_ID = hall_ID
            self.controlling_warden = controlling_warden
        else:
            self.warden_ID = warden_ID
            self._password = password
            self._name = name
            self._email = email
            self._hall_ID = hall_ID
            self._controlling_warden = controlling_warden


    # warden_ID getter and setter functions
    @property
    def warden_ID(self):
        return self._warden_ID

    @warden_ID.setter
    def warden_ID(self, warden_ID):
        self._warden_ID = warden_ID

    # password getter and setter functions
    @property
    def password(self):
        return self._password

    @password.setter
    def password(self, password):
        self._password = pv.hash_password(password)
        db.update("warden", self.warden_ID, "password", self.password)

    # name getter and setter functions
    @property
    def name(self):
        return self._name

    @name.setter
    def name(self, name):
        self._name = name
        db.update("warden", self.warden_ID, "name", self.name)

    # controlling_warden getter and setter functions
    @property
    def controlling_warden(self):
        return self._controlling_warden

    @controlling_warden.setter
    def controlling_warden(self, controlling_warden):
        self._controlling_warden = controlling_warden
        db.update("warden", self.warden_ID, "controlling_warden", self.controlling_warden)

    # email getter and setter functions
    @property
    def email(self):
        return self._email

    @email.setter
    def email(self, email):
        self._email = email
        db.update("warden", self.warden_ID, "email", self.email)

    # hall_ID getter and setter functions
    @property
    def hall_ID(self):
        return self._hall_ID

    @hall_ID.setter
    def hall_ID(self, hall_ID):
        self._hall_ID = hall_ID
        db.update("warden", self.warden_ID, "hall_ID", self.hall_ID)

    def total_occupancy(self, hall_table):
        """
        Return dictionary with occupancy of all Halls
        Works only if called on object with controlling_warden = True
        hall_table = dbr.rebuild("hall") is passed as parameter
        Return None if not called by controlling_warden object
        """

        if not self.controlling_warden:
            return None

        # Dictionary occupancy_table as elements in the following form
        # {hall_ID : (single_room_occupancy, double_room_occupancy)}
        occupancy_table = {'single_total': 0, 'single_occupy': 0, 'double_total': 0, 'double_occupy': 0}
        for key in hall_table:
            occupancy_table['single_total'] += hall_table[key].single_room_count
            occupancy_table['single_occupy'] += hall_table[key].single_room_occupancy
            occupancy_table['double_total'] += hall_table[key].double_room_count
            occupancy_table['double_occupy'] += hall_table[key].double_room_occupancy

        return occupancy_table

    def salary_charge(self, worker_table):
        """
        salary_charge getter function
        Calculate and return total salary
        worker_table = dbr.rebuild("worker") passed as parameter
        grant_request_object.salary_charge(dbr.rebuild("worker"))
        """

        total_salary = 0.

        for key in worker_table:
            if worker_table[key].hall_ID == self.hall_ID and isinstance(worker_table[key], attendant.Attendant):
                if worker_table[key].daily_wage == 0:
                    total_salary = total_salary + worker_table[key].monthly_salary
                else:
                    total_salary = total_salary + \
                                   worker_table[key].daily_wage * \
                                   worker_table[key].monthly_attendance

            elif worker_table[key].hall_ID == self.hall_ID:
                total_salary += worker_table[key].monthly_salary

        return total_salary