timmy / 台灣假日 API 查詢與快取

import requests import os import json class HolidayAPI: def __init__(self, base_url, cache_dir="cache"): """ Initialize the HolidayAPI class. :param base_url: The base URL for the API endpoint. :param cache_dir: Directory to store cached data. """ self.base_url = base_url self.cached_holidays = {} self.cache_dir = cache_dir if not os.path.exists(self.cache_dir): os.makedirs(self.cache_dir) def _get_cache_path(self, page, size): """ Generate the cache file path for a given page and size. :param page: The page number. :param size: The size of the page. :return: Path to the cache file. """ return os.path.join(self.cache_dir, f"holidays_page{page}_size{size}.json") def get_holidays(self, page=0, size=10): """ Fetch holiday data from the API or cache. :param page: The page number to fetch. :param size: The number of records per page. :return: A list of holiday data dictionaries. """ cache_path = self._get_cache_path(page, size) if os.path.exists(cache_path): with open(cache_path, "r", encoding="utf-8") as cache_file: return json.load(cache_file) url = f"{self.base_url}?page={page}&size={size}" headers = { 'accept': 'application/json' } response = requests.get(url, headers=headers) if response.status_code == 200: data = response.json() with open(cache_path, "w", encoding="utf-8") as cache_file: json.dump(data, cache_file, ensure_ascii=False, indent=4) return data else: response.raise_for_status() def get_holidays_by_year(self, year): """ Fetch holiday data for a specific year. If the data is already cached, it returns from the cache. :param year: The year for which to fetch holidays. :return: A list of holiday data dictionaries for the given year. """ if year in self.cached_holidays: return self.cached_holidays[year] all_holidays = [] page = 0 while True: data = self.get_holidays(page=page, size=100) if not data: break all_holidays.extend(data) page += 1 filtered_holidays = [ holiday for holiday in all_holidays if holiday.get("year") == str(year) and holiday.get("date")[4:] != "0903" ] self.cached_holidays[year] = filtered_holidays return filtered_holidays def is_holiday(self, date): """ Check if a specific date is a holiday. :param date: The date to check in YYYYMMDD format. :return: True if the date is a holiday, False otherwise. """ year = date[:4] if year not in self.cached_holidays: self.get_holidays_by_year(year) for holiday in self.cached_holidays[year]: if holiday.get("date") == date and holiday.get("isholiday") == "是": return True return False class Holiday: def __init__(self, date, year, name, isholiday, holidaycategory, description): """ Represent a single holiday entry. :param date: The date of the holiday in YYYYMMDD format. :param year: The year of the holiday. :param name: The name of the holiday (if any). :param isholiday: Whether it is a holiday ('是' or '否'). :param holidaycategory: The category of the holiday. :param description: The description of the holiday. """ self.date = date self.year = year self.name = name self.isholiday = isholiday self.holidaycategory = holidaycategory self.description = description def __repr__(self): return f"Holiday(date={self.date}, year={self.year}, name={self.name}, " \ f"isholiday={self.isholiday}, holidaycategory={self.holidaycategory}, " \ f"description={self.description})" # Example usage if __name__ == "__main__": api = HolidayAPI("https://data.ntpc.gov.tw/api/datasets/308dcd75-6434-45bc-a95f-584da4fed251/json") holidays_2025 = api.get_holidays_by_year(2025) holidays = [] for entry in holidays_2025: holiday = Holiday( date=entry.get("date"), year=entry.get("year"), name=entry.get("name"), isholiday=entry.get("isholiday"), holidaycategory=entry.get("holidaycategory"), description=entry.get("description") ) holidays.append(holiday) for holiday in holidays: print(holiday) # Check if a specific date is a holiday specific_date = "20250102" print(f"Is {specific_date} a holiday? {api.is_holiday(specific_date)}")