liz/PluralKit
liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add message count to member card. Closes #12

Browse Source
This commit is contained in:
Ske 2018-09-07 17:39:01 +02:00
parent f067485e88
commit 16173e428b
3 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/pluralkit/bot/commands/__init__.py
View File

@ -60,8 +60,7 @@ class CommandContext:
system = await self.get_system() system = await self.get_system()
if not system: if not system:
raise CommandError( raise CommandError("No system registered to this account. Use `pk;system new` to register one.")
embeds.error("No system registered to this account. Use `pk;system new` to register one."))
return system return system

7
src/pluralkit/bot/utils.py
View File

@ -152,6 +152,10 @@ async def generate_member_info_card(conn, member: Member) -> discord.Embed:
if member.pronouns: if member.pronouns:
card.add_field(name="Pronouns", value=member.pronouns) card.add_field(name="Pronouns", value=member.pronouns)
message_count = await db.get_member_message_count(conn, member.id)
if message_count > 0:
card.add_field(name="Message Count", value=str(message_count), inline=True)
if member.prefix or member.suffix: if member.prefix or member.suffix:
prefix = member.prefix or "" prefix = member.prefix or ""
suffix = member.suffix or "" suffix = member.suffix or ""
@ -162,6 +166,5 @@ async def generate_member_info_card(conn, member: Member) -> discord.Embed:
card.add_field(name="Description", card.add_field(name="Description",
value=member.description, inline=False) value=member.description, inline=False)
card.set_footer(text="System ID: {} | Member ID: {}".format( card.set_footer(text="System ID: {} | Member ID: {}".format(system.hid, member.hid))
system.hid, member.hid))
return card return card

4
src/pluralkit/db.py
View File

@ -259,6 +259,10 @@ async def delete_message(conn, message_id: str):
logger.debug("Deleting message (id={})".format(message_id)) logger.debug("Deleting message (id={})".format(message_id))
await conn.execute("delete from messages where mid = $1", int(message_id)) await conn.execute("delete from messages where mid = $1", int(message_id))
@db_wrap
async def get_member_message_count(conn, member_id: int) -> int:
return await conn.fetchval("select count(*) from messages where member = $1", member_id)
@db_wrap @db_wrap
async def front_history(conn, system_id: int, count: int): async def front_history(conn, system_id: int, count: int):
return await conn.fetch("""select return await conn.fetch("""select