Let-K-Pop-Wake-You-Up/weather.py at master · NCNU-OpenSource/Let-K-Pop-Wake-You-Up · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
import requests
import os
import re
from datetime import datetime
from dotenv import load_dotenv

load_dotenv()

def get_weather_data(location):
        url = "https://opendata.cwa.gov.tw/api/v1/rest/datastore/F-C0032-001"
        params = {
            "Authorization": os.getenv('OPEN_API'),
            "format": "JSON",
            "locationName": location
        }
        headers = {
            "accept": "application/json"
        }

        response = requests.get(url, params=params, headers=headers)
        # 將回應的JSON內容轉換為Python字典
        print(response)
        data = response.json()

        return data


def simplify_data(data):
        location_data = data['records']['location'][0]
        weather_elements = location_data['weatherElement']

        simplified_data = {
            'location': location_data['locationName'],
        }

        for element in weather_elements:
            element_name = element['elementName']
            for time in element['time']:
                # 使用完整的開始時間作為鍵
                start_time = time['startTime']
                if start_time not in simplified_data:
                    simplified_data[start_time] = {}

                parameter = time['parameter']
                parameter_str = parameter['parameterName']
                if 'parameterUnit' in parameter:
                    parameter_str += f" {parameter['parameterUnit']}"

                # 尋找或創建對應時間的字典
                end_time = time['endTime']
                if end_time not in simplified_data[start_time]:
                    simplified_data[start_time][end_time] = {}

                simplified_data[start_time][end_time][element_name] = parameter_str

        return simplified_data


def get_current_weather(simplified_data):
        try:
            # 獲取當前的日期和時間
            now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')

            # 遍歷所有的時間段
            for start_time in simplified_data:
                if start_time == 'location':
                    continue
                for end_time in simplified_data[start_time]:
                    # 如果當前時間在這個時間段內，則返回對應的天氣資訊
                    if start_time <= now <= end_time:
                        return simplified_data[start_time][end_time]
                    else:
                        # 如果沒有找到符合的時間段，則返回第一個天氣資訊
                        return simplified_data[start_time][end_time]
        except Exception as e:
            print(f"An error occurred: {e}")

        # 如果沒有找到任何天氣資訊，則返回None
        return None

def check_location_in_message(message):
        locations = [
            "臺北市", "臺中市", "臺南市", "高雄市",
            "新北市", "桃園市", "新竹市", "苗栗縣",
            "彰化縣", "南投縣", "雲林縣", "嘉義市",
            "嘉義縣", "屏東縣", "宜蘭縣", "花蓮縣",
            "臺東縣", "澎湖縣"
        ]

        # 將訊息中的 "台" 替換為 "臺"
        corrected_message = re.sub("台", "臺", message)
        local = corrected_message.split("_")

        for location in locations:
            if re.search(local[0], location):
                return location
            else:
                location

        return locations[0]

def fetch_weather():
    text = '南投縣'

    location = check_location_in_message(text)
    print('Location is: ' + location)
    weather_data = get_weather_data(location)
    simplified_data = simplify_data(weather_data)
    current_weather = get_current_weather(simplified_data)

    print('The Data is: ' + str(current_weather))
    if current_weather is not None:
            text = f'☝️「{text}」...\n位置: {location}\n氣候: {current_weather["Wx"]}\n降雨機率: {current_weather["PoP"]}\n體感: {current_weather["CI"]}'

    print(text)