hteppl
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
from io import StringIO
|
|
from typing import Any
|
|
|
|
from aiogram.types import BufferedInputFile
|
|
|
|
|
|
async def export_users_csv(session: Any) -> BufferedInputFile:
|
|
# Получение данных о платежах пользователя
|
|
query = """
|
|
SELECT
|
|
u.tg_id,
|
|
u.username,
|
|
u.first_name,
|
|
u.last_name,
|
|
u.language_code,
|
|
u.is_bot,
|
|
c.balance,
|
|
c.trial
|
|
FROM users u
|
|
LEFT JOIN connections c ON u.tg_id = c.tg_id
|
|
"""
|
|
|
|
users = await session.fetch(query)
|
|
|
|
buffer = StringIO()
|
|
buffer.write("tg_id,username,first_name,last_name,language_code,is_bot,balance,trial\n")
|
|
|
|
# Запись данных
|
|
for user in users:
|
|
buffer.write(
|
|
f"{user['tg_id']},{user['username']},{user['first_name']},{user['last_name']},"
|
|
f"{user['language_code']},{user['is_bot']},{user['balance']},{user['trial']}\n"
|
|
)
|
|
|
|
# Перемещение указателя в начало для чтения
|
|
buffer.seek(0)
|
|
|
|
return BufferedInputFile(
|
|
file=buffer.getvalue().encode("utf-8-sig"),
|
|
filename="users_export.csv"
|
|
)
|
|
|
|
|
|
async def export_payments_csv(session: Any) -> BufferedInputFile:
|
|
# Получение данных о всех платежах
|
|
query = """
|
|
SELECT
|
|
u.tg_id,
|
|
u.username,
|
|
u.first_name,
|
|
u.last_name,
|
|
p.amount,
|
|
p.payment_system,
|
|
p.status,
|
|
p.created_at
|
|
FROM users u
|
|
JOIN payments p ON u.tg_id = p.tg_id
|
|
"""
|
|
payments = await session.fetch(query)
|
|
return _export_payments_csv(payments, "payments_export.csv")
|
|
|
|
|
|
async def export_user_payments_csv(tg_id: int, session: Any) -> BufferedInputFile:
|
|
# Получение данных о платежах пользователя
|
|
query = """
|
|
SELECT
|
|
u.tg_id,
|
|
u.username,
|
|
u.first_name,
|
|
u.last_name,
|
|
p.amount,
|
|
p.payment_system,
|
|
p.status,
|
|
p.created_at
|
|
FROM users u
|
|
JOIN payments p ON u.tg_id = p.tg_id
|
|
WHERE u.tg_id = $1
|
|
"""
|
|
payments = await session.fetch(query, tg_id)
|
|
return _export_payments_csv(payments, f"payments_export_{tg_id}.csv")
|
|
|
|
|
|
def _export_payments_csv(payments: list, filename: str) -> BufferedInputFile:
|
|
# Формирование CSV данных через StringIO
|
|
buffer = StringIO()
|
|
buffer.write("tg_id,username,first_name,last_name,amount,payment_system,status,created_at\n")
|
|
|
|
# Запись данных
|
|
for payment in payments:
|
|
buffer.write(
|
|
f"{payment['tg_id']},{payment['username']},{payment['first_name']},{payment['last_name']},"
|
|
f"{payment['amount']},{payment['payment_system']},{payment['status']},{payment['created_at']}\n"
|
|
)
|
|
|
|
# Перемещение указателя в начало для чтения
|
|
buffer.seek(0)
|
|
|
|
return BufferedInputFile(
|
|
file=buffer.getvalue().encode("utf-8-sig"),
|
|
filename=filename
|
|
)
|