fix new

main.py

@@ -69,7 +69,6 @@ async def root(request: Request):
         return {"message": "kajsk"}
 
 
-
 # @app.post("/send-notification/{email}")
 # async def send_notification(request: Request, email: str, background_tasks: BackgroundTasks, q: str = Depends(get_query)):
 #     redis_client = request.app.state.redis
@@ -82,5 +81,5 @@ async def root(request: Request):
 
 if __name__ == '__main__':
     import uvicorn
-    uvicorn.run(app='main:app', host="127.0.0.1",
+    uvicorn.run(app='main:app', host="0.0.0.0",
-                port=8010, reload=True, debug=True)
+                port=8020, reload=True, debug=True)

ops/imgsmall/route.py

@@ -9,12 +9,15 @@ router = APIRouter()
 
 
 @router.get("/img")
-async def get_img(request: Request, url: str) -> JSONResponse:
+async def get_imgs(request: Request, keys: str) -> JSONResponse:
-    key = url[24:]
     redis_client = request.app.state.redis
-    values = await redis_client.get(key)
+    values = await redis_client.get(keys)
+    print(keys, values)
     if values:
+        print(f"get {keys}")
         return {"url": values}
     else:
-        vv = await get_img(key)
+        print(f"new down {keys}")
+        vv = await get_img(keys)
+        await redis_client.set(keys, vv)
         return {"url": vv}

scripts/common/get_img.py

@@ -1,14 +1,20 @@
 import shutil
+from pydantic.types import NoneBytes
 import requests
 from starlette.requests import Request
 import datetime
 import oss2
 import os
 
-async def get_img(request:Request,key:str)->bool:
+
-    url = f'http://imgsmall.dmzj.com/{key}'
+async def get_img(keys: str) -> bool:
+    print(f"get {keys}")
+    url = keys
     headers = {'Referer': 'https://www.dmzj.com/'}
     response = requests.get(url=url, headers=headers, stream=True)
+    key = keys[24:]
+    if not key:
+        return None
     name = key.split('/')[-1]
     paths = '/'.join(key.split('/')[:-1])
     project = 'imgsmall'
@@ -17,11 +23,10 @@ async def get_img(request:Request,key:str)->bool:
     with open(local_filename, 'wb') as out_file:
         shutil.copyfileobj(response.raw, out_file)
     del response
-    remote_filename=f"{[project]}/{paths}/{name}"
+    remote_filename = f"{project}{paths}/{name}"
     await simple_upload(local_filename, remote_filename)
     remote_url = f"https://h5games-al.kingsome.cn/{remote_filename}"
-    redis_client = request.app.state.redis
+
-    await redis_client.set(url,remote_url)
     return remote_url