feat: v2.0 — maintenance module, AI analysis, station power fix
- Add full 检修维护中心 (6.4): 3-type work orders (消缺/巡检/抄表), asset management, warehouse, work plans, billing settlement - Add AI智能分析 tab with LLM-powered diagnostics (StepFun + ZhipuAI) - Add AI模型配置 settings page (provider, temperature, prompts) - Fix station power accuracy: use API station total (station_power) instead of inverter-level computation — eliminates timing gaps - Add 7 new DB models, 4 new API routers, 5 new frontend pages - Migrations: 009 (maintenance expansion) + 010 (AI analysis) - Version bump: 1.6.1 → 2.0.0 Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Du Wenbo
244
core/backend/app/api/v1/billing.py
Normal file
244
core/backend/app/api/v1/billing.py
Normal file
@@ -0,0 +1,244 @@
|
||||
from datetime import datetime, timezone
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from sqlalchemy import select, func, extract
|
||||
from pydantic import BaseModel
|
||||
from app.core.database import get_db
|
||||
from app.core.deps import get_current_user, require_roles
|
||||
from app.models.maintenance import BillingRecord
|
||||
from app.models.user import User
|
||||
|
||||
router = APIRouter(prefix="/billing", tags=["电费结算"])
|
||||
|
||||
|
||||
# ── Pydantic Schemas ────────────────────────────────────────────────
|
||||
|
||||
class BillingCreate(BaseModel):
|
||||
station_name: str
|
||||
billing_type: str # "generation", "consumption", "grid_feed"
|
||||
year: int
|
||||
month: int
|
||||
generation_kwh: float | None = None
|
||||
consumption_kwh: float | None = None
|
||||
grid_feed_kwh: float | None = None
|
||||
unit_price: float | None = None
|
||||
total_amount: float | None = None
|
||||
status: str = "draft"
|
||||
invoice_number: str | None = None
|
||||
invoice_date: str | None = None
|
||||
payment_date: str | None = None
|
||||
notes: str | None = None
|
||||
|
||||
|
||||
class BillingUpdate(BaseModel):
|
||||
station_name: str | None = None
|
||||
billing_type: str | None = None
|
||||
year: int | None = None
|
||||
month: int | None = None
|
||||
generation_kwh: float | None = None
|
||||
consumption_kwh: float | None = None
|
||||
grid_feed_kwh: float | None = None
|
||||
unit_price: float | None = None
|
||||
total_amount: float | None = None
|
||||
status: str | None = None
|
||||
invoice_number: str | None = None
|
||||
invoice_date: str | None = None
|
||||
payment_date: str | None = None
|
||||
notes: str | None = None
|
||||
|
||||
|
||||
# ── Helpers ─────────────────────────────────────────────────────────
|
||||
|
||||
def _billing_to_dict(b: BillingRecord) -> dict:
|
||||
return {
|
||||
"id": b.id, "station_name": b.station_name,
|
||||
"billing_type": b.billing_type,
|
||||
"year": b.year, "month": b.month,
|
||||
"generation_kwh": b.generation_kwh,
|
||||
"consumption_kwh": b.consumption_kwh,
|
||||
"grid_feed_kwh": b.grid_feed_kwh,
|
||||
"unit_price": b.unit_price,
|
||||
"total_amount": b.total_amount,
|
||||
"status": b.status,
|
||||
"invoice_number": b.invoice_number,
|
||||
"invoice_date": str(b.invoice_date) if b.invoice_date else None,
|
||||
"payment_date": str(b.payment_date) if b.payment_date else None,
|
||||
"notes": b.notes,
|
||||
"created_by": b.created_by,
|
||||
"created_at": str(b.created_at) if b.created_at else None,
|
||||
"updated_at": str(b.updated_at) if b.updated_at else None,
|
||||
}
|
||||
|
||||
|
||||
# ── Billing CRUD ───────────────────────────────────────────────────
|
||||
|
||||
@router.get("")
|
||||
async def list_billing(
|
||||
station_name: str | None = None,
|
||||
billing_type: str | None = None,
|
||||
status: str | None = None,
|
||||
year: int | None = None,
|
||||
month: int | None = None,
|
||||
page: int = Query(1, ge=1),
|
||||
page_size: int = Query(20, ge=1, le=100),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
user: User = Depends(get_current_user),
|
||||
):
|
||||
query = select(BillingRecord)
|
||||
if station_name:
|
||||
query = query.where(BillingRecord.station_name == station_name)
|
||||
if billing_type:
|
||||
query = query.where(BillingRecord.billing_type == billing_type)
|
||||
if status:
|
||||
query = query.where(BillingRecord.status == status)
|
||||
if year:
|
||||
query = query.where(BillingRecord.year == year)
|
||||
if month:
|
||||
query = query.where(BillingRecord.month == month)
|
||||
|
||||
count_q = select(func.count()).select_from(query.subquery())
|
||||
total = (await db.execute(count_q)).scalar()
|
||||
|
||||
query = query.order_by(BillingRecord.year.desc(), BillingRecord.month.desc(), BillingRecord.id.desc())
|
||||
query = query.offset((page - 1) * page_size).limit(page_size)
|
||||
result = await db.execute(query)
|
||||
return {
|
||||
"total": total,
|
||||
"items": [_billing_to_dict(b) for b in result.scalars().all()],
|
||||
}
|
||||
|
||||
|
||||
@router.get("/stats")
|
||||
async def billing_stats(
|
||||
year: int | None = None,
|
||||
db: AsyncSession = Depends(get_db),
|
||||
user: User = Depends(get_current_user),
|
||||
):
|
||||
query = select(BillingRecord)
|
||||
if year:
|
||||
query = query.where(BillingRecord.year == year)
|
||||
|
||||
# Total generation
|
||||
gen_q = select(func.sum(BillingRecord.generation_kwh))
|
||||
if year:
|
||||
gen_q = gen_q.where(BillingRecord.year == year)
|
||||
total_generation = (await db.execute(gen_q)).scalar() or 0
|
||||
|
||||
# Total amount
|
||||
amt_q = select(func.sum(BillingRecord.total_amount))
|
||||
if year:
|
||||
amt_q = amt_q.where(BillingRecord.year == year)
|
||||
total_amount = (await db.execute(amt_q)).scalar() or 0
|
||||
|
||||
# By month
|
||||
month_q = select(
|
||||
BillingRecord.month,
|
||||
func.sum(BillingRecord.generation_kwh).label("generation"),
|
||||
func.sum(BillingRecord.total_amount).label("amount"),
|
||||
).group_by(BillingRecord.month).order_by(BillingRecord.month)
|
||||
if year:
|
||||
month_q = month_q.where(BillingRecord.year == year)
|
||||
month_result = await db.execute(month_q)
|
||||
by_month = [
|
||||
{"month": row[0], "generation_kwh": float(row[1] or 0), "total_amount": float(row[2] or 0)}
|
||||
for row in month_result.all()
|
||||
]
|
||||
|
||||
# By type
|
||||
type_q = select(
|
||||
BillingRecord.billing_type,
|
||||
func.sum(BillingRecord.total_amount).label("amount"),
|
||||
).group_by(BillingRecord.billing_type)
|
||||
if year:
|
||||
type_q = type_q.where(BillingRecord.year == year)
|
||||
type_result = await db.execute(type_q)
|
||||
by_type = {row[0]: float(row[1] or 0) for row in type_result.all()}
|
||||
|
||||
return {
|
||||
"total_generation_kwh": float(total_generation),
|
||||
"total_amount": float(total_amount),
|
||||
"by_month": by_month,
|
||||
"by_type": by_type,
|
||||
}
|
||||
|
||||
|
||||
@router.get("/export")
|
||||
async def export_billing(
|
||||
station_name: str | None = None,
|
||||
year: int | None = None,
|
||||
month: int | None = None,
|
||||
db: AsyncSession = Depends(get_db),
|
||||
user: User = Depends(get_current_user),
|
||||
):
|
||||
query = select(BillingRecord)
|
||||
if station_name:
|
||||
query = query.where(BillingRecord.station_name == station_name)
|
||||
if year:
|
||||
query = query.where(BillingRecord.year == year)
|
||||
if month:
|
||||
query = query.where(BillingRecord.month == month)
|
||||
query = query.order_by(BillingRecord.year.desc(), BillingRecord.month.desc())
|
||||
result = await db.execute(query)
|
||||
records = [_billing_to_dict(b) for b in result.scalars().all()]
|
||||
return {
|
||||
"columns": [
|
||||
"station_name", "billing_type", "year", "month",
|
||||
"generation_kwh", "consumption_kwh", "grid_feed_kwh",
|
||||
"unit_price", "total_amount", "status",
|
||||
"invoice_number", "invoice_date", "payment_date",
|
||||
],
|
||||
"data": records,
|
||||
}
|
||||
|
||||
|
||||
@router.get("/{billing_id}")
|
||||
async def get_billing(
|
||||
billing_id: int,
|
||||
db: AsyncSession = Depends(get_db),
|
||||
user: User = Depends(get_current_user),
|
||||
):
|
||||
result = await db.execute(select(BillingRecord).where(BillingRecord.id == billing_id))
|
||||
record = result.scalar_one_or_none()
|
||||
if not record:
|
||||
raise HTTPException(status_code=404, detail="结算记录不存在")
|
||||
return _billing_to_dict(record)
|
||||
|
||||
|
||||
@router.post("")
|
||||
async def create_billing(
|
||||
data: BillingCreate,
|
||||
db: AsyncSession = Depends(get_db),
|
||||
user: User = Depends(require_roles("admin", "energy_manager")),
|
||||
):
|
||||
record = BillingRecord(
|
||||
**data.model_dump(exclude={"invoice_date", "payment_date"}),
|
||||
created_by=user.id,
|
||||
)
|
||||
if data.invoice_date:
|
||||
record.invoice_date = datetime.fromisoformat(data.invoice_date)
|
||||
if data.payment_date:
|
||||
record.payment_date = datetime.fromisoformat(data.payment_date)
|
||||
db.add(record)
|
||||
await db.flush()
|
||||
return _billing_to_dict(record)
|
||||
|
||||
|
||||
@router.put("/{billing_id}")
|
||||
async def update_billing(
|
||||
billing_id: int,
|
||||
data: BillingUpdate,
|
||||
db: AsyncSession = Depends(get_db),
|
||||
user: User = Depends(require_roles("admin", "energy_manager")),
|
||||
):
|
||||
result = await db.execute(select(BillingRecord).where(BillingRecord.id == billing_id))
|
||||
record = result.scalar_one_or_none()
|
||||
if not record:
|
||||
raise HTTPException(status_code=404, detail="结算记录不存在")
|
||||
for k, v in data.model_dump(exclude_unset=True, exclude={"invoice_date", "payment_date"}).items():
|
||||
setattr(record, k, v)
|
||||
if data.invoice_date:
|
||||
record.invoice_date = datetime.fromisoformat(data.invoice_date)
|
||||
if data.payment_date:
|
||||
record.payment_date = datetime.fromisoformat(data.payment_date)
|
||||
return _billing_to_dict(record)
|
||||
Reference in New Issue
Block a user