Files
appointment_tool/src/db/mirgration/migration_tools.py
T

36 lines
1.6 KiB
Python

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")