feat(bot): Add bans migration script.

This commit is contained in:
Nyeki.py
2024-10-17 01:44:28 -03:00
parent c745df0a9d
commit 0335c77e84
6 changed files with 857 additions and 10 deletions
+24
View File
@@ -0,0 +1,24 @@
from datetime import datetime
import csv
import asyncio
from payphone.db import setup, models
async def main():
await setup.init()
with open("bans.csv", "r", encoding="latin-1") as f:
reader = csv.reader(f)
for row in reader:
user_id = int(row[1])
reason = row[2].strip()[:500]
expires = datetime.fromisoformat(row[9].strip()) if row[9].strip() else None
await models.Ban.create(user_id=user_id, reason=reason, expires_at=expires)
asyncio.get_event_loop().run_until_complete(main())