新增 IP 归属地查询，天气查询等功能

main.py

@@ -1,25 +1,90 @@
-# STM32Player 服务端
-
+import time
+import json
 import asyncio
 
+import aiohttp
+from fuzzywuzzy import fuzz
 from loguru import logger
 
 
-async def handler(reader, writer):
-    data = await reader.read(100)
-    message = data.decode('gbk')
-    addr = writer.get_extra_info('peername')
+citycode_list = json.load(open('./citycode.json', 'r'))
 
-    logger.info(f"从 {addr!r} 接收：{message!r}")
 
-    logger.info(f"发送：{message!r}")
-    writer.write(data)
+async def handler(reader: asyncio.StreamReader, writer: asyncio.StreamWriter):
+    data: bytes = await reader.read(100)
+    req_str: str = data.decode('gbk')
+
+    addr: tuple = writer.get_extra_info('peername')
+    logger.info(f"从 {addr!r} 接收：{req_str!r}")
+
+    # 处理请求
+    if (req_str.find('checknet') != -1):
+        resp_str = checknet_handler(req_str)
+    elif (req_str.find('time') != -1):
+        resp_str = time_handler(req_str)
+    elif (req_str.find('weather') != -1):
+        resp_str = await weather_handler(req_str)
+    elif (req_str.find('ipaddr') != -1):
+        resp_str = await ipaddr_handler(req_str, addr)
+    else:
+        resp_str = 'ERROR'
+
+    logger.info(f"发送：{resp_str!r}")
+    writer.write((resp_str + '\n').encode('gbk'))
     await writer.drain()
 
     logger.info("关闭连接")
     writer.close()
 
 
+def checknet_handler(req_str: str) -> str:
+    return 'OK'
+
+
+def time_handler(req_str: str) -> str:
+    return 'OK\n' + time.strftime("%Y%m%d%H%M%S", time.localtime())
+
+
+async def weather_handler(req_str: str) -> str:
+    city_str = req_str.split('=')[1]
+    city_code = city_to_code(city_str)
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get(f"http://t.weather.itboy.net/api/weather/city/{city_code}") as resp:
+            res = await resp.json()
+
+    if res['status'] != 200:
+        return 'ERROR'
+    else:
+        return 'OK\n' + str(res['cityInfo']['city']) + '&' + str(res['data']['forecast'][0]['type']) + '&' + str(res['data']['wendu']) + '&' + str(
+            res['data']['forecast'][0]['fx']) + '&' + str(res['data']['forecast'][0]['fl']) + '&' + str(res['data']['shidu']) + '&' + str(res['data']['quality'])
+
+
+def city_to_code(city_str: str) -> str:
+    ratio_list = []
+
+    for city in citycode_list:
+        ratio_list.append(fuzz.ratio(city_str, city['city_name']))
+
+    max_i = 0
+    for i in range(len(ratio_list)):
+        if ratio_list[i] > ratio_list[max_i] and citycode_list[i]['city_code'] != '':
+            max_i = i
+
+    return citycode_list[max_i]['city_code']
+
+
+async def ipaddr_handler(req_str: str, addr: tuple) -> str:
+    async with aiohttp.ClientSession() as session:
+        async with session.get(f"http://opendata.baidu.com/api.php?query={addr[0]}&co=&resource_id=6006&oe=utf8") as resp:
+            res = await resp.json()
+
+    if 'data' in res:
+        return f"OK\n{addr[0]}&{res['data'][0]['location']}"
+    else:
+        return 'FAIL'
+
+
 async def main():
     server = await asyncio.start_server(
         handler, '0.0.0.0', 2333)