From d1591e065b455f94f6ebdd8ba2c4947cad8845d5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B0=B9=E8=88=9F?= <13007110208@163.com>
Date: Tue, 11 Feb 2025 18:03:58 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dockerfile             |  2 +-
 run.py => main.py      | 28 ++++++++--------------------
 utils/query_weather.py |  4 +++-
 3 files changed, 12 insertions(+), 22 deletions(-)
 rename run.py => main.py (51%)

diff --git a/dockerfile b/dockerfile
index 23cf465..44f6eaa 100644
--- a/dockerfile
+++ b/dockerfile
@@ -17,4 +17,4 @@ RUN pip install --no-cache-dir -r requirements.txt -i https://pypi.tuna.tsinghua
 COPY . .
 
 # 运行应用程序
-ENTRYPOINT ["python3", "run.py"]
\ No newline at end of file
+ENTRYPOINT ["python3", "main.py"]
\ No newline at end of file
diff --git a/run.py b/main.py
similarity index 51%
rename from run.py
rename to main.py
index 9660b51..8f55cfa 100644
--- a/run.py
+++ b/main.py
@@ -3,40 +3,28 @@ import uvicorn
 from utils import query_weather
 from utils.log import Log
 
-app = FastAPI()
+app = FastAPI(title="yin_home文档")
 
 
-@app.get('/weather')
+@app.get('/weather', tags=["获取天气数据"], summary="返回实时天气数据")
 def weather():
-    log = Log().getlog()
     response = query_weather.query_weather()
-    log.info(response)
-
-    return response, 200
+    return response
 
 
-@app.get('/weather_all')
+@app.get('/weather_all', tags=["获取天气数据"], summary="返回天气预报数据")
 def weather_all():
-    log = Log().getlog()
     response = query_weather.query_weather(extensions='all')
-    log.info(response)
-
-    return response, 200
+    return response
 
 
-@app.get('/weather_flag')
+@app.get('/weather_flag', tags=["获取天气数据"], summary="返回实时天气数据状态码")
 def weather_flag():
     log = Log().getlog()
     response = query_weather.query_weather()
-    log.info(response)
     weather = response['lives'][0]['weather']
-    # weather包含雨字符串
-    if '雨' in weather:
-        log.info({"weather": 1})
-        return {"weather": 1}, 200
-    else:
-        log.info({"weather": 0})
-        return {"weather": 0}, 200
+    log.info(weather)
+    return 1 if '雨' in weather else 0
 
 
 if __name__ == '__main__':
diff --git a/utils/query_weather.py b/utils/query_weather.py
index 053ed94..ddc9bfd 100644
--- a/utils/query_weather.py
+++ b/utils/query_weather.py
@@ -1,12 +1,14 @@
 import requests
+from utils.log import Log
 
 
 def query_weather(key="3add99f10c5144ebd64bfa0bfb6b1c20", city="420115", extensions="base", output="JSON"):
+    log = Log().getlog()
     url = "https://restapi.amap.com/v3/weather/weatherInfo"
 
     querystring = {"key": key, "city": city, "extensions": extensions,
                    "output": output}
 
     response = requests.request("GET", url, params=querystring)
-
+    log.info(response.json())
     return response.json()