import streamlit as st
from pymongo import MongoClient
from datetime import datetime, time

# MongoDB에 연결
client = MongoClient('mongodb://localhost:27017/')

# 데이터베이스 선택
db = client['db_schedules']

# 컬렉션 선택
collection = db['schedules']

def create_schedule(title, date, task, priority):
    # datetime.combine()을 사용하여 date 객체를 datetime 객체로 변환
    datetime_obj = datetime.combine(date, time.min)
    schedule = {
        'title': title,
        'date': datetime_obj,
        'task': task,
        'priority': priority
    }
    collection.insert_one(schedule)

def get_schedules():
    return collection.find()

# 스케줄 입력
st.header('새로운 스케줄 추가')
title = st.text_input('제목')
date = st.date_input('날짜')
task = st.text_area('해야 할 일')
priority = st.selectbox('우선순위', ['낮음', '보통', '높음'])
if st.button('저장'):
    create_schedule(title, date, task, priority)
    st.success('스케줄이 저장되었습니다.')

# 저장된 스케줄 조회
st.header('저장된 스케줄 목록')
schedules = get_schedules()
for schedule in schedules:
    # datetime 객체를 날짜 형식으로 변환하여 출력
    schedule_date = schedule['date'].date()
    st.write('제목:', schedule['title'])
    st.write('날짜:', schedule_date)
    st.write('해야 할 일:', schedule['task'])
    st.write('우선순위:', schedule['priority'])
    st.write('---')