Upload server.py

server.py

@@ -198,40 +198,58 @@ def static_files(filename: str) -> Any:
 
 
 @app.route("/api/register", methods=["POST"])
-def api_register() -> Any:
-    payload = request.get_json(force=True)
-    email = payload.get("email")
-    password = payload.get("password")
-    confirm = payload.get("confirm_password")
-
-    if password != confirm:
-        return jsonify({"error": "Hasla musza byc identyczne."}), 400
-    if len(password or "") < PASSWORD_MIN_LENGTH:
-        return jsonify({"error": "Haslo jest za krotkie."}), 400
+def api_register() -> Any:
+    payload = request.get_json(force=True)
+    email = payload.get("email")
+    password = payload.get("password")
+    confirm = payload.get("confirm_password")
+    business_fields = [
+        "company_name",
+        "owner_name",
+        "address_line",
+        "postal_code",
+        "city",
+        "tax_id",
+        "bank_account",
+    ]
+    business_data: Dict[str, str] = {}
+
+    for field in business_fields:
+        value = (payload.get(field) or "").strip()
+        if not value:
+            return jsonify({"error": f"Pole {field} jest wymagane."}), 400
+        business_data[field] = value
+
+    if password != confirm:
+        return jsonify({"error": "Hasla musza byc identyczne."}), 400
+    if len(password or "") < PASSWORD_MIN_LENGTH:
+        return jsonify({"error": "Haslo jest za krotkie."}), 400
 
     login_key, display_email = normalize_email(email)
     password_hash = hash_password(password)
 
-    if DATABASE_AVAILABLE:
-        if fetch_one("SELECT 1 FROM accounts WHERE login = %s", (login_key,)):
-            return jsonify({"error": "Konto o podanym emailu juz istnieje."}), 400
-        create_account(login_key, display_email, password_hash)
-        return jsonify({"message": "Konto zostalo utworzone."})
-
-    data = load_store()
-    if login_key in data["accounts"]:
-        return jsonify({"error": "Konto o podanym emailu juz istnieje."}), 400
-
-    data["accounts"][login_key] = {
-        "login": login_key,
-        "email": display_email,
-        "password_hash": password_hash,
-        "business": None,
-        "invoices": [],
-        "logo": None,
-        "created_at": datetime.utcnow().isoformat(timespec="seconds"),
-    }
-    save_store(data)
+    if DATABASE_AVAILABLE:
+        if fetch_one("SELECT 1 FROM accounts WHERE login = %s", (login_key,)):
+            return jsonify({"error": "Konto o podanym emailu juz istnieje."}), 400
+        account_id = create_account(login_key, display_email, password_hash)
+        update_business(account_id, business_data)
+        return jsonify({"message": "Konto zostalo utworzone."})
+
+    data = load_store()
+    accounts = data.setdefault("accounts", {})
+    if login_key in accounts:
+        return jsonify({"error": "Konto o podanym emailu juz istnieje."}), 400
+
+    accounts[login_key] = {
+        "login": login_key,
+        "email": display_email,
+        "password_hash": password_hash,
+        "business": business_data,
+        "invoices": [],
+        "logo": None,
+        "created_at": datetime.utcnow().isoformat(timespec="seconds"),
+    }
+    save_store(data)
     return jsonify({"message": "Konto zostalo utworzone."})