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

Add message count to export file. Closes #43.

Browse Source
This commit is contained in:
Ske 2019-02-16 14:38:44 +01:00
parent 3e7800971b
commit a5265b318f
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/pluralkit/bot/commands/misc_commands.py
View File

@ -52,6 +52,8 @@ async def invite_link(ctx: CommandContext):
async def export(ctx: CommandContext): async def export(ctx: CommandContext):
working_msg = await ctx.message.channel.send("Working...")
system = await ctx.ensure_system() system = await ctx.ensure_system()
members = await system.get_members(ctx.conn) members = await system.get_members(ctx.conn)
@ -76,7 +78,8 @@ async def export(ctx: CommandContext):
"description": member.description, "description": member.description,
"prefix": member.prefix, "prefix": member.prefix,
"suffix": member.suffix, "suffix": member.suffix,
"created": member.created.isoformat() "created": member.created.isoformat(),
"message_count": await member.message_count(ctx.conn)
} for member in members } for member in members
], ],
"accounts": [str(uid) for uid in accounts], "accounts": [str(uid) for uid in accounts],
@ -88,8 +91,10 @@ async def export(ctx: CommandContext):
] # TODO: messages ] # TODO: messages
} }
await working_msg.delete()
f = io.BytesIO(json.dumps(data).encode("utf-8")) f = io.BytesIO(json.dumps(data).encode("utf-8"))
await ctx.message.channel.send(content="Here you go!", file=discord.File(fp=f, filename="system.json")) await ctx.message.channel.send(content="Here you go!", file=discord.File(fp=f, filename="pluralkit_system.json"))
async def tell(ctx: CommandContext): async def tell(ctx: CommandContext):