add a user export method

Author: zefanjajobse

cogs/admin.py

@@ -16,6 +16,7 @@
 from utils.register import register
 from utils.role_management import RoleManagement
 from utils.server_settings import get_guild, update_guild
+from utils.users import get_users_csv
 from utils.voice_channel import create_voice_channel
 from utils.voice_channels import add_voice_channel
 
@@ -84,6 +85,24 @@ async def username_autocomplete(
                 for player in users.scalars()
             ]
 
+
+    @group.command(name="export", description="Export all registered users")
+    @app_commands.guild_only()
+    @app_commands.default_permissions(administrator=True)
+    @app_commands.checks.has_permissions(administrator=True)
+    async def export_registered_users(self, interaction: discord.Interaction) -> None:
+        """Export all registered users"""
+        await interaction.response.defer()
+        if interaction.guild is None:
+            return  # is already set to guild_only
+        async with self.bot.db.create_session() as session:
+            total, registered_users = await get_users_csv(session, interaction.guild_id)
+            if total <= 0:
+                await interaction.followup.send("There are currently no registered users", ephemeral=True)
+                return
+            await interaction.followup.send("Registered users:", ephemeral=True, file=registered_users)
+
+
     @group.command(name="unregister", description="Unregister a user")
     @app_commands.describe(username="EA username")
     @app_commands.guild_only()

utils/users.py

@@ -0,0 +1,23 @@
+import csv
+import io
+
+import discord
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from database.dto.users import User
+
+async def get_users_csv(session: AsyncSession, server_id: int) -> tuple[int, discord.File]:
+    stmt = (
+        select(User)
+        .filter(User.server_id == server_id)
+    )
+    res = (await session.execute(stmt)).all()
+    total = len(res)
+    with io.StringIO() as data_stream:
+        outcsv = csv.writer(data_stream)
+        outcsv.writerow(User.__table__.columns.keys())     
+        for row in res:
+            outcsv.writerow([row[0].server_id, row[0].discord_id, row[0].username, row[0].player_id, row[0].kdr_role_id, row[0].user_id, row[0].created_at, row[0].updated_at])
+        data_stream.seek(0)
+        return (total, discord.File(data_stream, filename="channel_names.csv"))