def create_user_schedule(self, user):
"""Create user schedule directly from db"""
user_schedule = UserSchedule(user_id=user.id)
for item in self.schedule_items:
user_schedule.add_schedule_item(**item)
user_schedule.save()
return user_schedule
def save_user_schedule(user_id):
data = request.json
user_schedule = UserSchedule(user_id=user_id)
for item in data['schedule_items']:
user_schedule.add_schedule_item(**item)
user_schedule.save()
return (jsonify({
'id': str(user_schedule.id),
}), 201)
def edit_user_schedule(user_id, user_schedule_id):
user_schedule = UserSchedule.objects(id=user_schedule_id).first()
# If the schedule doesn't exist or the user is mismatched,
# create a new one with the same items.
if user_schedule is None:
user_schedule = UserSchedule(user_id=user_id)
elif str(user_schedule.user_id.id) != user_id:
name = f'{user_schedule.name} (copied)'
user_schedule = UserSchedule(user_id=user_id, name=name)
data = request.json
user_schedule.clear_schedule_item()
for editedScheduleItem in data['schedule_items']:
user_schedule.add_schedule_item(**editedScheduleItem)
user_schedule.save()
return (jsonify({'user_schedule': user_schedule.serialize()}), 200)
def test_add_schedule_item_should_increase_the_schedule_item(self):
user_schedule = UserSchedule(user_id=self.generate_random_user_item(),
name="User Schedule A")
assert len(user_schedule.schedule_items) == 0
inserted_schedule = user_schedule.add_schedule_item(name="Schedule A",
day="Day A",
room="Room A",
start="Start",
end="End")
assert len(user_schedule.schedule_items) == 1
assert inserted_schedule.name == "Schedule A"
assert inserted_schedule.day == "Day A"
assert inserted_schedule.room == "Room A"
assert inserted_schedule.start == "Start"
assert inserted_schedule.end == "End"