You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
2.0 KiB
62 lines
2.0 KiB
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")
|