add cnn

.gitea/workflows/ci.yml

@@ -85,5 +85,28 @@ jobs:
           shm-size: 2g
           ulimit: core=0:0
 
+      - name: Meta-cnn
+        id: meta-cnn
+        uses: docker/metadata-action@v4
+        with:
+          images: huoxue.link:12300/huoxue1/cnn
+          tags: |
+            type=raw,value=latest
+            type=ref,event=tag
+      - name: Build and push cnn
+        id: docker_build_cnn
+        uses: docker/build-push-action@v3
+        with:
+          context: .
+          push: ${{ github.event_name != 'pull_request' }}
+          file: ./Dockerfile.cnn
+          platforms: linux/amd64,linux/arm64
+          tags: ${{ steps.meta.outputs.tags }}
+          cache-from: type=gha
+          cache-to: type=gha,mode=max
+          shm-size: 2g
+          ulimit: core=0:0
+
+
 
 

Dockerfile

@@ -13,4 +13,4 @@ VOLUME /logs
 
 
 
-ENTRYPOINT [ "/min" ]
+ENTRYPOINT [ "/min_report" ]

Dockerfile.cnn

@@ -0,0 +1,9 @@
+FROM python:3.11.4
+
+ADD ./cnn.py  /home/cnn.py
+
+WORKDIR /home
+
+RUN pip3 install ddddocr fastapi uvicorn  pydantic pillow==9.5.0 -i https://pypi.mirrors.ustc.edu.cn/simple
+
+CMD ["python", "/home/cnn.py"]

cnn.py

@@ -0,0 +1,144 @@
+import uvicorn
+from pydantic import BaseModel
+
+import os
+from io import BytesIO
+from PIL import Image
+from ddddocr import DdddOcr
+
+string1 = "KLMNOPQRSTABCDEFGHIJUVWXYZabcdopqrstuvwxefghijklmnyz0123456789+/"
+string2 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
+
+def randomstr(num):
+    randomstr = ''.join(str(uuid.uuid4()).split('-'))[num:]
+    return randomstr
+
+def randomstr1(num):
+    randomstr = ""
+    for i in range(num):
+        randomstr = randomstr + random.choice("abcdefghijklmnopqrstuvwxyz0123456789")
+    return randomstr
+
+def sign_core(inarg):
+    key = b'80306f4370b39fd5630ad0529f77adb6'
+    mask = [0x37, 0x92, 0x44, 0x68, 0xA5, 0x3D, 0xCC, 0x7F, 0xBB, 0xF, 0xD9, 0x88, 0xEE, 0x9A, 0xE9, 0x5A]
+    array = [0 for _ in range(len(inarg))]
+    for i in range(len(inarg)):
+        r0 = int(inarg[i])
+        r2 = mask[i & 0xf]
+        r4 = int(key[i & 7])
+        r0 = r2 ^ r0
+        r0 = r0 ^ r4
+        r0 = r0 + r2
+        r2 = r2 ^ r0
+        r1 = int(key[i & 7])
+        r2 = r2 ^ r1
+        array[i] = r2 & 0xff
+    return bytes(array)
+
+def base64Encode(string):
+    return base64.b64encode(string.encode("utf-8")).decode('utf-8').translate(str.maketrans(string1, string2))
+
+def base64Decode(string):
+    return base64.b64decode(string.translate(str.maketrans(string1, string2))).decode('utf-8')
+
+def randomeid():
+    return 'eidAaf8081218as20a2GM%s7FnfQYOecyDYLcd0rfzm3Fy2ePY4UJJOeV0Ub840kG8C7lmIqt3DTlc11fB/s4qsAP8gtPTSoxu' % randomstr1(
+        20)
+
+def get_ep(jduuid : str=''):
+    if not jduuid:
+        jduuid = randomstr(16)
+    # jduuid=base64Decode('EQHtZNG2CJu5CzcnCzu0CK==')
+    # print(jduuid)
+    ts = str(int(time.time() * 1000))
+    bsjduuid = base64Encode(jduuid)
+    # ts='1643792319938'
+    area = base64Encode('%s_%s_%s_%s' % (
+        random.randint(1, 10000), random.randint(1, 10000), random.randint(1, 10000), random.randint(1, 10000)))
+    d_model = random.choice(['Mi11Ultra', 'Mi11', 'Mi10'])
+    d_model = base64Encode(d_model)
+    return '{"hdid":"JM9F1ywUPwflvMIpYPok0tt5k9kW4ArJEU3lfLhxBqw=","ts":%s,"ridx":-1,"cipher":{"area":"%s","d_model":"%s","wifiBssid":"dW5hbw93bq==","osVersion":"CJS=","d_brand":"WQvrb21f","screen":"CtS1DIenCNqm","uuid":"%s","aid":"%s","openudid":"%s"},"ciphertype":5,"version":"1.2.0","appname":"com.jingdong.app.mall"}' % (
+        int(ts) - random.randint(100, 1000), area, d_model, bsjduuid, bsjduuid, bsjduuid), jduuid, ts
+
+def get_sign(functionId, body, client : str="android", clientVersion : str='11.2.8',jduuid : str='') -> dict:
+    if isinstance(body,dict):#判断body数据类型是否为dict 是就转化为json
+        d=body
+        body=json.dumps(body)
+    else:
+        d=json.loads(body)
+
+    if "eid" in d:
+        eid=d["eid"]
+    else:
+        eid=randomeid()
+
+    ep, suid, st = get_ep(jduuid)
+    # print(ep)
+    sv = random.choice(["102", "111", "120"])
+    # sv = '102'
+    all_arg = "functionId=%s&body=%s&uuid=%s&client=%s&clientVersion=%s&st=%s&sv=%s" % (
+        functionId, body, suid, client, clientVersion, st, sv)
+    #print(all_arg)
+    back_bytes = sign_core(str.encode(all_arg))
+    sign = hashlib.md5(base64.b64encode(back_bytes)).hexdigest()
+    # print(sign)
+    data={"functionId":functionId,"body":body,"clientVersion":clientVersion,"client":client,"uuid":suid,"eid":eid,"ep":ep,"st":st,"sign":sign,"sv":sv}
+    data["convertUrl"]='functionId=%s&body=%s&clientVersion=%s&client=%s&sdkVersion=31&lang=zh_CN&harmonyOs=0&networkType=wifi&oaid=%s&eid=%s&ef=1&ep=%s&st=%s&sign=%s&sv=%s' % (
+        functionId,body, clientVersion, client, suid, eid, urllib.parse.quote(ep), st, sign, sv)
+    data["url"]='https://api.m.jd.com?%s' % (data["convertUrl"])
+    return data
+
+# print(get_sign("initForFarm",{ "babelChannel": "121", "sid": "3c52b5f17ab2a42398939a27887eaf8w", "version": 18, "channel": 1 },"android","11.2.8"))
+
+
+
+
+
+app = FastAPI()
+
+
+
+ocr = DdddOcr()
+
+
+class Req(BaseModel):
+    img: str
+    host: str
+    ip: str
+    port: str
+
+
+@app.post("/data")
+async def _(req: Req):
+    img = req.dict().get("img")
+    host = req.dict().get("host")
+    ip = req.dict().get("ip")
+    port = req.dict().get("port")
+    print("host: {}, ip:{}, listen_port: {}".format(host,ip,port))
+    img = bytes(base64.b64decode(img))
+    s = time.time()
+    value = ocr.classification(img)
+    e = time.time()
+    print("识别结果: {}".format(value))
+    result = {
+        'value': value,  # 预测的结果
+        'speed_time(ms)': int((e - s) * 1000)  # 识别耗费的时间
+    }
+    return result
+
+class SignReq(BaseModel):
+    functionId: str
+    body: str
+
+@app.post("/sign")
+def http_sign(req:SignReq):
+    return get_sign(req.functionId,req.body)
+
+@app.get("/sign")
+def http_get_sign(functionId:str,body:str):
+    return get_sign(functionId,body)
+
+
+if __name__ == '__main__':
+    uvicorn.run(app=app, host='0.0.0.0', port=7000)