import datetime

from src import definitions
from src.db.mongo_manager import MONGO_STORE_MANAGER
from src.pojo.ReserveResultPojo import ReserveResultPojo
from src.pojo.ResultEnum import ResultEnum
from src.pojo.accepted_appointment_pojo import AcceptedAppointmentPojo


def migre_accepted_item_from_firebase_to_mongo(day: str):
    # get successful item from firestore
    collection = definitions.firebase_store_manager.get_all_accepted_items_for_day(day).stream()
    for valid_appointment in collection:
        reserve_pojo = ReserveResultPojo.from_firestore_dict(valid_appointment.to_dict())
        if reserve_pojo.accepted == "ACCEPTED":
            MONGO_STORE_MANAGER.update_reserve_result_for_collection(reserve_pojo.id, ResultEnum.ACCEPTED, day)


def migre_accepted_appointment(day: str):
    # get successful item from firestore
    collection = MONGO_STORE_MANAGER.get_accepted_items_for_one_day(day)
    for valid_appointment in collection:
        accepted_pojo = AcceptedAppointmentPojo.from_reserve(valid_appointment)
        accepted_pojo.day = day
        accepted_pojo.accepted_at = datetime.datetime.strptime(day, '%Y-%m-%d').timestamp()
        MONGO_STORE_MANAGER.insert_accepted_reserve(accepted_pojo)


if __name__ == '__main__':
    # migre_accepted_item_from_firebase_to_mongo("2022-08-22")
    # days = ['2022-07-04', '2022-07-05', '2022-07-06', '2022-07-07', '2022-07-08', '2022-07-09', '2022-07-11',
    #         '2022-07-12', '2022-07-13', '2022-07-15', '2022-07-16', '2022-07-18''2022-07-19', '2022-07-20',
    #         '2022-07-21']
    # for day in days:
    migre_accepted_appointment("2022-08-22")