feat: add pcf analysis web app and api

tests/test_web.py

@@ -0,0 +1,104 @@
+from pathlib import Path
+import os
+import threading
+from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
+
+from fastapi.testclient import TestClient
+
+from app.main import app
+
+
+def test_web_upload_sample_files():
+    client = TestClient(app)
+    opened = []
+    try:
+        files = []
+        for path in sorted(Path("PCF文件样本(1)").glob("*.pcf")):
+            handle = path.open("rb")
+            opened.append(handle)
+            files.append(("files", (path.name, handle, "text/plain")))
+
+        response = client.post("/api/jobs", files=files)
+        assert response.status_code == 200
+        payload = response.json()
+        assert payload["input_files"] == 20
+        assert payload["weld_count"] == 169
+        assert len(payload["tables"]["import_rows"]) == 169
+        assert {"key": "unit_code", "label": "单位代码"} in payload["tables"]["import_columns"]
+        assert "unit_code" in payload["tables"]["import_rows"][0]
+        assert "line_no" in payload["tables"]["import_rows"][0]
+        assert "单位代码" not in payload["tables"]["import_rows"][0]
+        assert payload["tables"]["issue_rows"]
+        assert "单位代码" in payload["tables"]["import_headers"]
+
+        xls = client.get(payload["downloads"]["xls"])
+        assert xls.status_code == 200
+        assert xls.content[:8] == b"\xd0\xcf\x11\xe0\xa1\xb1\x1a\xe1"
+    finally:
+        for handle in opened:
+            handle.close()
+
+
+def test_parse_url_api_returns_json_payload(tmp_path):
+    sample = sorted(Path("PCF文件样本(1)").glob("*.pcf"))[0]
+    hosted = tmp_path / sample.name
+    hosted.write_bytes(sample.read_bytes())
+
+    class QuietHandler(SimpleHTTPRequestHandler):
+        def log_message(self, format, *args):
+            return
+
+    old_cwd = os.getcwd()
+    os.chdir(tmp_path)
+    server = ThreadingHTTPServer(("127.0.0.1", 0), QuietHandler)
+    thread = threading.Thread(target=server.serve_forever, daemon=True)
+    thread.start()
+    try:
+        url = f"http://127.0.0.1:{server.server_port}/{hosted.name}"
+        client = TestClient(app)
+        response = client.post(
+            "/api/AnalysisPcf",
+            json={"attachment_url": url},
+        )
+        assert response.status_code == 200
+        payload = response.json()
+        assert payload["input_files"] == 1
+        assert payload["weld_count"] > 0
+        assert payload["tables"]["import_rows"]
+        assert "line_no" in payload["tables"]["import_rows"][0]
+        assert "管线代号" not in payload["tables"]["import_rows"][0]
+        assert "files" not in payload
+        assert "downloads" not in payload
+        assert payload["report"]["summary"]["weld_count"] == payload["weld_count"]
+    finally:
+        server.shutdown()
+        server.server_close()
+        os.chdir(old_cwd)
+
+
+def test_parse_url_api_accepts_local_file_path():
+    sample = sorted(Path("PCF文件样本(1)").glob("*.pcf"))[0].resolve()
+    client = TestClient(app)
+    response = client.post(
+        "/api/AnalysisPcf",
+        json={"attachment_url": str(sample)},
+    )
+    assert response.status_code == 200
+    payload = response.json()
+    assert payload["input_files"] == 1
+    assert payload["weld_count"] > 0
+    assert payload["tables"]["import_rows"]
+
+
+def test_parse_url_api_accepts_file_uri():
+    sample = sorted(Path("PCF文件样本(1)").glob("*.pcf"))[0].resolve()
+    client = TestClient(app)
+    response = client.post(
+        "/api/AnalysisPcf",
+        json={"attachment_url": sample.as_uri()},
+    )
+    assert response.status_code == 200
+    payload = response.json()
+    assert payload["input_files"] == 1
+    assert payload["weld_count"] > 0
+    assert "downloads" not in payload