Transition over to use db instead of fetching discord REST

This commit is contained in:
Jeremy Zhang
2017-05-06 18:19:20 -07:00
parent 9c24093901
commit fc169c2063
4 changed files with 47 additions and 24 deletions

View File

@ -6,6 +6,6 @@ from guilds import Guilds
from unauthenticated_users import UnauthenticatedUsers
from unauthenticated_bans import UnauthenticatedBans
from authenticated_users import AuthenticatedUsers
from guild_members import GuildMembers
from guild_members import GuildMembers, list_all_guild_members
from keyvalue_properties import KeyValueProperties, set_keyvalproperty, get_keyvalproperty, getexpir_keyvalproperty, setexpir_keyvalproperty, ifexists_keyvalproperty, delete_keyvalproperty
from messages import Messages, get_channel_messages

View File

@ -1,4 +1,5 @@
from titanembeds.database import db
import json
class GuildMembers(db.Model):
__tablename__ = "guild_members"
@ -26,3 +27,19 @@ class GuildMembers(db.Model):
def __repr__(self):
return '<GuildMembers {0} {1} {2} {3} {4}>'.format(self.id, self.guild_id, self.user_id, self.username, self.discriminator)
def list_all_guild_members(guild_id):
memlist = []
members = db.session.query(GuildMembers).filter(GuildMembers.guild_id == guild_id).all()
for member in members:
memlist.append({
"user": {
"username": member.username,
"discriminator": member.discriminator,
"id": member.user_id,
"avatar": member.avatar
},
"roles": json.loads(member.roles),
"nickname": member.nickname,
})
return memlist