Upload server.py

server.py

@@ -540,80 +540,102 @@ def build_invoice(payload: Dict[str, Any], business: Dict[str, Any], client: Dic
     }
 
 
-@app.route("/api/invoices", methods=["GET", "POST"])
-def api_invoices() -> Any:
-    try:
-        login_key = require_auth()
-    except PermissionError:
-        return jsonify({"error": "Brak autoryzacji."}), 401
-
-    if request.method == "GET":
-        if DATABASE_AVAILABLE:
-            try:
-                account_row = get_account_row(login_key)
-            except KeyError:
-                return jsonify({"error": "Nie znaleziono konta."}), 404
-            rows = fetch_all(
-                """
-                SELECT invoice_number AS invoice_id,
-                       to_char(issued_at, 'YYYY-MM-DD HH24:MI') AS issued_at,
-                       sale_date,
-                       total_gross
-                FROM invoices
-                WHERE account_id = %s
-                ORDER BY issued_at DESC
-                LIMIT %s
-                """,
-                (account_row["id"], INVOICE_HISTORY_LIMIT),
-            )
-            return jsonify({"invoices": rows})
-
-        data = load_store()
-        try:
-            account = get_account(data, login_key)
-        except KeyError:
-            return jsonify({"error": "Nie znaleziono konta."}), 404
-        invoices = account.get("invoices", [])[:INVOICE_HISTORY_LIMIT]
-        return jsonify({"invoices": invoices})
-
-    payload = request.get_json(force=True)
-    data = load_store()
-    try:
-        account = get_account(data, login_key)
-    except KeyError:
-        return jsonify({"error": "Nie znaleziono konta."}), 404
-
-    business = account.get("business")
-    if not business:
-        return jsonify({"error": "Ustaw dane sprzedawcy przed dodaniem faktury."}), 400
-
-    client = validate_client(payload)
-    try:
-        invoice = build_invoice(payload, business, client)
-    except ValueError as error:
-        return jsonify({"error": str(error)}), 400
-
-    if DATABASE_AVAILABLE:
-        account_row = get_account_row(login_key)
-        client_id = upsert_client(
-            account_row["id"],
-            {
-                "name": client["name"],
-                "address_line": client["address_line"],
-                "postal_code": client["postal_code"],
-                "city": client["city"],
-                "tax_id": client["tax_id"],
-                "phone": client.get("phone"),
-            },
-        )
-        insert_invoice(account_row["id"], client_id, invoice)
-        return jsonify({"message": "Faktura zostala zapisana."})
-
-    invoices = account.setdefault("invoices", [])
-    invoices.insert(0, invoice)
-    account["invoices"] = invoices[:INVOICE_HISTORY_LIMIT]
-    save_store(data)
-    return jsonify({"message": "Faktura zostala zapisana."})
+@app.route("/api/invoices", methods=["GET", "POST"])
+def api_invoices() -> Any:
+    try:
+        login_key = require_auth()
+    except PermissionError:
+        return jsonify({"error": "Brak autoryzacji."}), 401
+
+    if request.method == "GET":
+        if DATABASE_AVAILABLE:
+            try:
+                account_row = get_account_row(login_key)
+            except KeyError:
+                return jsonify({"error": "Nie znaleziono konta."}), 404
+            rows = fetch_all(
+                """
+                SELECT invoice_number AS invoice_id,
+                       to_char(issued_at, 'YYYY-MM-DD HH24:MI') AS issued_at,
+                       sale_date,
+                       total_gross
+                FROM invoices
+                WHERE account_id = %s
+                ORDER BY issued_at DESC
+                LIMIT %s
+                """,
+                (account_row["id"], INVOICE_HISTORY_LIMIT),
+            )
+            return jsonify({"invoices": rows})
+
+        data = load_store()
+        try:
+            account = get_account(data, login_key)
+        except KeyError:
+            return jsonify({"error": "Nie znaleziono konta."}), 404
+        invoices = account.get("invoices", [])[:INVOICE_HISTORY_LIMIT]
+        return jsonify({"invoices": invoices})
+
+    payload = request.get_json(force=True)
+
+    if DATABASE_AVAILABLE:
+        try:
+            account_row = get_account_row(login_key)
+        except KeyError:
+            return jsonify({"error": "Nie znaleziono konta."}), 404
+        business = fetch_one(
+            """
+            SELECT company_name, owner_name, address_line, postal_code,
+                   city, tax_id, bank_account
+            FROM business_profiles
+            WHERE account_id = %s
+            """,
+            (account_row["id"],),
+        )
+        if not business:
+            return jsonify({"error": "Ustaw dane sprzedawcy przed dodaniem faktury."}), 400
+
+        client = validate_client(payload)
+        try:
+            invoice = build_invoice(payload, business, client)
+        except ValueError as error:
+            return jsonify({"error": str(error)}), 400
+
+        client_id = upsert_client(
+            account_row["id"],
+            {
+                "name": client["name"],
+                "address_line": client["address_line"],
+                "postal_code": client["postal_code"],
+                "city": client["city"],
+                "tax_id": client["tax_id"],
+                "phone": client.get("phone"),
+            },
+        )
+        insert_invoice(account_row["id"], client_id, invoice)
+        return jsonify({"message": "Faktura zostala zapisana.", "invoice": invoice})
+
+    data = load_store()
+    try:
+        account = get_account(data, login_key)
+    except KeyError:
+        return jsonify({"error": "Nie znaleziono konta."}), 404
+
+    business = account.get("business")
+    if not business:
+        return jsonify({"error": "Ustaw dane sprzedawcy przed dodaniem faktury."}), 400
+
+    client = validate_client(payload)
+    try:
+        invoice = build_invoice(payload, business, client)
+    except ValueError as error:
+        return jsonify({"error": str(error)}), 400
+
+    invoices = account.setdefault("invoices", [])
+    invoices.insert(0, invoice)
+    account["invoices"] = invoices[:INVOICE_HISTORY_LIMIT]
+    save_store(data)
+    return jsonify({"message": "Faktura zostala zapisana.", "invoice": invoice})
 
 
 @app.route("/api/invoices/summary", methods=["GET"])