| from __future__ import annotations |
| import csv |
| import io |
| from datetime import datetime, timezone |
|
|
| from aiogram.types import BufferedInputFile |
|
|
| from bot.database.models import UserModel |
|
|
|
|
| async def convert_users_to_csv(users: list[UserModel]) -> BufferedInputFile: |
| """Export all users in csv file.""" |
| columns = UserModel.__table__.columns |
| data = [[getattr(user, column.name) for column in columns] for user in users] |
|
|
| s = io.StringIO() |
| csv.writer(s).writerow(columns) |
| csv.writer(s).writerows(data) |
| s.seek(0) |
|
|
| return BufferedInputFile( |
| file=s.getvalue().encode("utf-8"), |
| filename=f"users_{datetime.now(timezone.utc).strftime('%Y.%m.%d_%H.%M')}.csv", |
| ) |
|
|
