2af95ebcf1
CMS editor: formatting toolbar with H1-H3, bold, italic, strikethrough, links, images, lists, blockquotes, code blocks, horizontal rules. Keyboard shortcuts (Ctrl+B/I/D/K). Improved markdown preview rendering. Import page: shows current data summary with year badge, stats grid, last import date. Year input for new imports. Preview with sample table. Backend: added data_year and data_imported_at fields to TariffParams, returned in stats endpoint. Import sets data_imported_at automatically. Seed sets data_year=2018. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
107 lines
3.2 KiB
Python
107 lines
3.2 KiB
Python
"""Seed the database with Saoû data from Eau2018.xls."""
|
|
|
|
import asyncio
|
|
import sys
|
|
import os
|
|
from datetime import datetime
|
|
|
|
sys.path.insert(0, os.path.dirname(__file__))
|
|
|
|
import xlrd
|
|
from sqlalchemy import select
|
|
|
|
from app.database import engine, async_session, init_db
|
|
from app.models import Commune, TariffParams, Household, AdminUser
|
|
from app.services.auth_service import hash_password
|
|
from app.services.import_service import generate_auth_code
|
|
|
|
XLS_PATH = os.path.join(os.path.dirname(__file__), "..", "Eau2018.xls")
|
|
|
|
|
|
async def seed():
|
|
await init_db()
|
|
|
|
async with async_session() as db:
|
|
# Check if already seeded
|
|
result = await db.execute(select(Commune).where(Commune.slug == "saou"))
|
|
if result.scalar_one_or_none():
|
|
print("Saoû already seeded.")
|
|
return
|
|
|
|
# Create commune
|
|
commune = Commune(
|
|
name="Saoû",
|
|
slug="saou",
|
|
description="Commune de Saoû - Tarification progressive de l'eau",
|
|
)
|
|
db.add(commune)
|
|
await db.flush()
|
|
|
|
# Create tariff params
|
|
params = TariffParams(
|
|
commune_id=commune.id,
|
|
abop=100,
|
|
abos=100,
|
|
recettes=75000,
|
|
pmax=20,
|
|
vmax=2100,
|
|
data_year=2018,
|
|
data_imported_at=datetime.utcnow(),
|
|
)
|
|
db.add(params)
|
|
|
|
# Create super admin (manages all communes)
|
|
super_admin = AdminUser(
|
|
email="superadmin@sejeteralo.fr",
|
|
hashed_password=hash_password("superadmin"),
|
|
full_name="Super Admin",
|
|
role="super_admin",
|
|
)
|
|
db.add(super_admin)
|
|
|
|
# Create commune admin for Saoû (manages only this commune)
|
|
commune_admin = AdminUser(
|
|
email="saou@sejeteralo.fr",
|
|
hashed_password=hash_password("saou2024"),
|
|
full_name="Admin Saoû",
|
|
role="commune_admin",
|
|
)
|
|
commune_admin.communes.append(commune)
|
|
db.add(commune_admin)
|
|
|
|
# Import households from Eau2018.xls
|
|
book = xlrd.open_workbook(XLS_PATH)
|
|
sheet = book.sheet_by_name("CALCULS")
|
|
nb_hab = 363
|
|
|
|
existing_codes = set()
|
|
for r in range(1, nb_hab + 1):
|
|
name = sheet.cell_value(r, 0)
|
|
status = sheet.cell_value(r, 3)
|
|
volume = sheet.cell_value(r, 4)
|
|
price = sheet.cell_value(r, 33)
|
|
|
|
code = generate_auth_code()
|
|
while code in existing_codes:
|
|
code = generate_auth_code()
|
|
existing_codes.add(code)
|
|
|
|
household = Household(
|
|
commune_id=commune.id,
|
|
identifier=str(name).strip(),
|
|
status=str(status).strip().upper(),
|
|
volume_m3=float(volume),
|
|
price_paid_eur=float(price) if price else 0.0,
|
|
auth_code=code,
|
|
)
|
|
db.add(household)
|
|
|
|
await db.commit()
|
|
print(f"Seeded: commune 'saou', {nb_hab} households")
|
|
print(f" Super admin: superadmin@sejeteralo.fr / superadmin")
|
|
print(f" Commune admin Saoû: saou@sejeteralo.fr / saou2024")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(seed())
|