notion_api.py
import os
import requests
import json
from datetime import datetime
from dotenv import load_dotenv
import logging
# dotenv를 통해 환경 변수 로드
load_dotenv()
# 환경 변수에서 민감한 정보 가져오기
DATABASE_URL = os.getenv("NOTION_DATABASE_URL")
NOTION_TOKEN = os.getenv("NOTION_API_TOKEN")
NOTION_VERSION = os.getenv("NOTION_VERSION") # 기본값 설정
# 헤더에 환경 변수 적용
headers = {
"Authorization": f"Bearer {NOTION_TOKEN}",
"Content-Type": "application/json",
"Notion-Version": NOTION_VERSION
}
# 로깅 설정
logging.basicConfig(
filename='notion_api.log', # 로그 파일 이름
level=logging.ERROR, # 로그 수준
format='%(asctime)s - %(levelname)s - %(message)s', # 로그 포맷
)
# Notion API URL
def fetch_data_from_notion():
"""Notion API에서 데이터 가져오기"""
logging.info("Fetching data from Notion API.")
response = requests.post(DATABASE_URL, headers=headers)
if response.status_code == 200:
logging.info("Data fetched successfully.")
return response.json()
else:
logging.error(f"Request failed: {response.status_code}")
logging.error(response.text)
return None
def extract_dates():
"""오늘 날짜를 반환"""
today_date = datetime.today().strftime('%Y-%m-%d')
logging.info(f"Today's date: {today_date}")
return today_date
def process_page_data(data, today_date):
"""페이지 데이터를 순회하면서 어제 날짜에 해당하는 데이터를 찾고 처리"""
logging.info("Processing page data.")
for page in data.get('results', []):
created_date = page['created_time'].split('T')[0]
if created_date == today_date:
contents = page['properties'].get('contents', {}).get('rich_text', [{}])[0].get('text', {}).get('content', '')
learn = page['properties'].get('learn', {}).get('rich_text', [{}])[0].get('text', {}).get('content', '')
logging.info(f"Found data for today: {created_date}")
return created_date, contents, learn
logging.info("No data found for today.")
return None, None, None
def save_data_to_file(created_date, contents, learn, today_date):
"""파일로 데이터를 저장"""
logging.info(f"Saving data to file: {today_date.replace('-', '')}.json")
final_result = {created_date: {'contents': contents, 'learn': learn}}
file_name = f"{today_date.replace('-', '')}.json" # yymmdd 형태로 파일명 생성
with open(file_name, 'w') as json_file:
json.dump(final_result[created_date], json_file, indent=4, ensure_ascii=False)
logging.info(f"File '{file_name}' has been created/overwritten.")
def main():
# 데이터 가져오기
logging.info("Starting the process.")
data = fetch_data_from_notion()
if not data:
logging.error("No data fetched. Exiting the process.")
return
# 오늘과 어제 날짜 추출
today_date = extract_dates()
# 페이지 데이터 처리
created_date, contents, learn = process_page_data(data, today_date)
# 처리된 데이터가 있으면 파일로 저장
if created_date:
save_data_to_file(created_date, contents, learn, today_date)
else:
logging.warning("No data for today's date.")
if __name__ == "__main__":
main()
