import MySQLdb
import member
import group
from exceptions import MemberNotFoundException


class Admidio:
    """Creates the Admidio Class"""

    def __init__(self, mysql_host, mysql_user, mysql_pass, mysql_db):
        self.db = MySQLdb.connect(
            host=mysql_host, user=mysql_user, passwd=mysql_pass, db=mysql_db)
        self.cursor = self.db.cursor()
        self.members = dict()
        self.groups = dict()
        self.events = dict()
        sql = "SELECT usf_id,usf_name_intern FROM `adm_user_fields`"
        self.cursor.execute(sql)
        self.userfields = dict(self.cursor.fetchall())

        sql = """SELECT usr_id, usr_login_name, usr_password
                 FROM adm_users WHERE usr_valid = 1"""
        self.cursor.execute(sql)
        self.initMembers()
        self.initGroups()

    def __del__(self):
        self.db.close()

    def initMembers(self):
        """Initialises all members in the Admidio installation"""

        sql = """SELECT usr_id, usr_login_name, usr_password
                 FROM adm_users WHERE usr_valid = 1"""
        self.cursor.execute(sql)

        for row in self.cursor.fetchall():
            self.members[row[0]] = member.Member(
                self.db, row[0], row[1],
                row[2], self.userfields, self)

    def initGroups(self):
        """Initialises all groups in the Admidio installation"""

        sql = """SELECT rol_id, rol_name, rol_description
                 FROM adm_roles
                 WHERE rol_visible=1 AND rol_valid=1"""
        self.cursor.execute(sql)

        for row in self.cursor.fetchall():
            self.groups[row[0]] = group.Group(self.db, row[0], row[1], self)

    def getMemberFromID(self, user_id):
        """
        Returns Member object if user_id exists
        otherwise throws MemberNotFoundException
        """
        if user_id in self.members:
            return self.members[user_id]
        else:
            raise MemberNotFoundException(
                f"Member with id {user_id} not found")