def test_get_events_start_and_end_time_frame_failure( today_event, next_month_event, sender, session): start = today_event.start.date() + timedelta(days=1) end = next_month_event.start.date() - timedelta(days=1) events = get_all_user_events(session, sender.id) filtered_events = filter_dates(events, start, end) assert list(filtered_events) == []
def test_get_events_start_and_end_time_frame_success( today_event, next_month_event, sender, session): start = today_event.start.date() end = next_month_event.start.date() events = get_all_user_events(session, sender.id) filtered_events = filter_dates(events, start, end) assert set(filtered_events) == {today_event, next_month_event}
def test_get_events_no_time_frame(today_event, next_month_event, sender, session): start = None end = None events = get_all_user_events(session, sender.id) filtered_events = filter_dates(events, start, end) assert set(filtered_events) == {today_event, next_month_event}
def get_events_in_time_frame( start_date: Union[date, None], end_date: Union[date, None], user_id: int, db: Session ) -> Iterator[Event]: """Yields all user's events in a time frame.""" events = get_all_user_events(db, user_id) yield from filter_dates(events, start_date, end_date)
def get_all_day_events( day: datetime, session, user_id: int, ) -> Iterator[Event]: events = get_all_user_events(session, user_id) day_end = day + timedelta(hours=24) for event in events: if is_all_day_event_in_day(event=event, day=day, day_end=day_end): yield (event)
def get_events_and_attributes( day: datetime, session, user_id: int, ) -> Iterator[Tuple[Event, DivAttributes]]: events = get_all_user_events(session, user_id) day_end = day + timedelta(hours=24) for event in events: if is_specific_time_event_in_day(event, day, day_end): yield event, DivAttributes(event, day)
def get_events_per_dates(session: Session, user_id: int, start: Optional[date], end: Optional[date]) -> Union[Iterator[Event], list]: """Read from the db. Return a list of all the user events between the relevant dates.""" if start > end: return [] return (filter_dates( sort_by_date(get_all_user_events(session, user_id)), start, end, ))
def get_events_per_friend( session: Session, user_id: int, my_friend: str, ) -> List[Event]: """ My_friend is the name of a person that appears in the invite list of events. He is not necessarily a registered user×¥ The variable is used to show all events where we are both in the invitees list""" events_together = [] sorted_events = sort_by_date(get_all_user_events(session, user_id)) for event in sorted_events: if my_friend in event.invitees.split(','): events_together.append(event) return events_together
def test_get_events_end_time_frame_success(today_event, sender, session): start = None end = today_event.end.date() events = get_all_user_events(session, sender.id) filtered_events = filter_dates(events, start, end) assert list(filtered_events) == [today_event]