|
|
@@ -4,10 +4,11 @@ from fastapi import FastAPI
|
|
|
import os
|
|
|
import uuid
|
|
|
import re
|
|
|
+import zipfile
|
|
|
import json
|
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
from fastapi.middleware.gzip import GZipMiddleware
|
|
|
-
|
|
|
+import base64
|
|
|
from pydantic import BaseModel
|
|
|
from subdir import service
|
|
|
from subdir import db
|
|
|
@@ -630,11 +631,44 @@ async def updatezjcs(r: UpdateZjcs):
|
|
|
|
|
|
@app.post("/upload/")
|
|
|
async def upload(file: UploadFile):
|
|
|
-
|
|
|
- content = await file.read()
|
|
|
- string = content.decode('utf-8')
|
|
|
- await resolve(string)
|
|
|
- return [file.filename]
|
|
|
+ if file.filename.endswith('13jz') or file.filename.endswith("13jt"):
|
|
|
+ content = await file.read()
|
|
|
+ string = content.decode('utf-8')
|
|
|
+ await resolve(string)
|
|
|
+ return [file.filename]
|
|
|
+ elif file.filename.endswith("jszf"):
|
|
|
+ content = await file.read()
|
|
|
+ string = content.decode('utf-8')
|
|
|
+ root = ET.fromstring(string)
|
|
|
+ for child in root:
|
|
|
+ print(child)
|
|
|
+ if child.tag == "ZBFileContent":
|
|
|
+ base64_bytes = child.text.encode("ascii")
|
|
|
+
|
|
|
+ sample_string_bytes = base64.b64decode(base64_bytes)
|
|
|
+ tmp = str(uuid.uuid4())
|
|
|
+ os.mkdir(os.path.join("tmp", tmp))
|
|
|
+ with open(tmp, "wb") as file_:
|
|
|
+ file_.write(sample_string_bytes)
|
|
|
+ with zipfile.ZipFile(tmp, 'r') as zip_ref:
|
|
|
+ zip_ref.extractall(os.path.join("tmp", tmp))
|
|
|
+ dir_list = os.listdir(os.path.join("tmp", tmp))
|
|
|
+ print(dir_list)
|
|
|
+ for entry in dir_list:
|
|
|
+ if entry.endswith("13jz"):
|
|
|
+ with open(os.path.join("tmp", tmp, entry), 'r') as f:
|
|
|
+ await resolve(f.read())
|
|
|
+
|
|
|
+ return [file.filename]
|
|
|
+ else:
|
|
|
+ return [file.filename]
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
|
|
|
|
|
|
|
