def getCurrentRoles(guild_id: int, category: Util.RoleCategory = None):
if category:
query = KfpRole.select().where(KfpRole.guild_id == guild_id,
KfpRole.category == category)
else:
query = KfpRole.select().where(KfpRole.guild_id == guild_id)
result = []
if query.exists():
for role in query.iterator():
result.append(role)
return result
def getCategoryRole(guild_id: int, category: Util.RoleCategory):
query = KfpRole.select().where(KfpRole.guild_id == guild_id,
KfpRole.category == category)
if query.exists():
role: KfpRole = query.get()
return role.role_id
return ""
def getKfpRoleFromLevel(guild_id: int, level: int):
query = KfpRole.select().where(KfpRole.guild_id == guild_id,
level >= KfpRole.level,
KfpRole.category > 0)
if query.exists():
return query.order_by(KfpRole.level.desc()).get()
return None
def getKfpRolesBeforeLevel(guild_id: int, level: int):
query = KfpRole.select().where(KfpRole.guild_id == guild_id,
level > KfpRole.level)
result = []
if query.exists():
for role in query.order_by(KfpRole.level.desc()).iterator():
result.append(role)
return result
def getKfpRolesFromLevel(guild_id: int, level: int):
query = KfpRole.select().where(KfpRole.guild_id == guild_id,
level >= KfpRole.level,
KfpRole.category > 0)
roles = {}
if query.exists():
for role in query.order_by(KfpRole.level.desc()).iterator():
if role.category not in roles:
roles[role.category] = role
return list(roles.values())
def updateRole(guild_id: int,
role_id: int,
role_name: str,
color: str,
category: int = 0):
query = KfpRole.select().where(KfpRole.guild_id == guild_id,
KfpRole.role_id == role_id)
role: KfpRole
if query.exists():
role = query.get()
else:
role = KfpRole.create(guild_id=guild_id,
role_id=role_id,
role_name=role_name,
color=color,
category=category)
role.role_name = role_name
role.color = color
role.save()
return role
def wipeDataAndKeepTable():
KfpRole.drop_table()
KfpRole.create_table()
def deleteAllData():
KfpRole.drop_table()
def updateKfpRoleLevel(role: KfpRole, level: int):
role.level = level
role.save()
return role
def getRole(guild_id: int, role_id: int):
query = KfpRole.select().where(KfpRole.guild_id == guild_id,
KfpRole.role_id == role_id)
if query.exists():
return query.get()
return None