Upload main.py

main.py

@@ -447,3 +447,90 @@ def root_redirect():
 from openai_integration import router as ai_router
 app.include_router(ai_router, prefix="/ai", tags=["ai"])
 
+# ===== ここから追記（main.py の import 群より下ならどこでもOK） =====
+from pydantic import Field
+
+class WizardItemIn(BaseModel):
+    description: str
+    quantity: float = Field(gt=0, default=1)
+    unit_price: float = Field(ge=0)
+    tax_rate: float = Field(ge=0, default=0.0)
+
+class WizardCustomerIn(BaseModel):
+    # 既存顧客を使うなら id を、無ければ name 等で新規作成
+    id: int | None = None
+    name: str | None = None
+    email: str | None = None
+    phone: str | None = None
+    address: str | None = None
+    city: str | None = None
+    country: str | None = None
+
+class WizardInvoiceIn(BaseModel):
+    customer: WizardCustomerIn
+    due_date: date | None = None
+    notes: str | None = None
+    items: list[WizardItemIn]
+
+class WizardInvoiceOut(BaseModel):
+    customer_id: int
+    invoice_id: int
+    totals: MoneyBreakdown
+
+@app.post("/wizard/invoice", dependencies=[Depends(require_api_key)], response_model=WizardInvoiceOut)
+def wizard_create_invoice(payload: WizardInvoiceIn, session: Session = Depends(get_session)):
+    """
+    顧客IDがあればそれを使用。無ければ name などから顧客を新規作成。
+    その後、請求書を作成し、明細を一括登録して totals を返す。
+    """
+    # 1) 顧客を決定
+    cust_id: int | None = payload.customer.id
+    if cust_id:
+        cust = session.get(Customer, cust_id)
+        if not cust:
+            raise HTTPException(404, "Customer id not found")
+    else:
+        if not payload.customer.name:
+            raise HTTPException(422, "Either customer.id or customer.name is required")
+        # 既存に同名があればそれを使い、無ければ作成（簡易重複回避）
+        existing = session.exec(
+            select(Customer).where(Customer.name == payload.customer.name)
+        ).first()
+        if existing:
+            cust = existing
+        else:
+            cust = Customer(
+                name=payload.customer.name.strip(),
+                email=(payload.customer.email or None),
+                phone=(payload.customer.phone or None),
+                address=(payload.customer.address or None),
+                city=(payload.customer.city or None),
+                country=(payload.customer.country or None),
+            )
+            session.add(cust)
+            session.commit()
+            session.refresh(cust)
+        cust_id = cust.id
+
+    # 2) 請求書を作成
+    inv = Invoice(customer_id=cust_id, due_date=payload.due_date, notes=payload.notes)
+    session.add(inv)
+    session.commit()
+    session.refresh(inv)
+
+    # 3) 明細を一括追加
+    for it in payload.items:
+        session.add(InvoiceItem(
+            invoice_id=inv.id,
+            description=it.description,
+            quantity=it.quantity,
+            unit_price=it.unit_price,
+            tax_rate=it.tax_rate,
+        ))
+    session.commit()
+
+    items = session.exec(select(InvoiceItem).where(InvoiceItem.invoice_id == inv.id)).all()
+    totals = compute_totals(items)
+
+    return WizardInvoiceOut(customer_id=cust_id, invoice_id=inv.id, totals=totals)
+# ===== 追記ここまで =====