|
1 | 1 | from datetime import datetime |
2 | 2 | from operator import attrgetter |
3 | | -from typing import Dict, List, Optional |
| 3 | +from typing import Dict, List, Optional, Any |
4 | 4 |
|
5 | 5 | from app.database.models import Event, UserEvent |
6 | 6 | from app.dependencies import templates |
@@ -30,33 +30,49 @@ async def eventview(request: Request, id: int): |
30 | 30 |
|
31 | 31 | def by_id(db: Session, event_id: int) -> Event: |
32 | 32 | """Select event by id""" |
| 33 | + |
33 | 34 | return db.query(Event).filter(Event.id == event_id).first() |
34 | 35 |
|
35 | 36 |
|
36 | | -def validate_dates(start_date: datetime, end_date: datetime) -> bool: |
| 37 | +def is_date_before(start_date: datetime, end_date: datetime) -> bool: |
37 | 38 | """Check if the start date is earlier than the end date""" |
| 39 | + |
38 | 40 | return start_date < end_date |
39 | 41 |
|
40 | 42 |
|
| 43 | +def is_it_possible_to_change_dates( |
| 44 | + db: Session, old_event: Event, event: Dict[str, Any]) -> bool: |
| 45 | + return is_date_before( |
| 46 | + event.get('start', old_event.start), |
| 47 | + event.get('end', old_event.end)) |
| 48 | + |
| 49 | + |
| 50 | +def get_items_that_can_be_updated(event: Dict[str, Any]) -> Dict[str, Any]: |
| 51 | + """Extract only that keys to update""" |
| 52 | + |
| 53 | + return {i: event[i] for i in ( |
| 54 | + 'title', 'start', 'end', 'content', 'location') if i in event} |
| 55 | + |
| 56 | + |
41 | 57 | def update_event(event_id: int, event: Dict, db: Session |
42 | 58 | ) -> Optional[Event]: |
| 59 | + |
43 | 60 | # TODO Check if the user is the owner of the event. |
44 | 61 |
|
45 | | - # Extract only that keys to update |
46 | | - event_to_update = {i: event[i] for i in ( |
47 | | - 'title', 'start', 'end', 'content', 'location') if i in event} |
48 | | - if not bool(event_to_update): # Event items is empty |
| 62 | + event_to_update = get_items_that_can_be_updated(event) |
| 63 | + if not event_to_update: |
49 | 64 | return None |
50 | 65 | try: |
51 | 66 | old_event = by_id(db=db, event_id=event_id) |
52 | | - if old_event is None or not validate_dates( |
53 | | - event_to_update.get('start', old_event.start), |
54 | | - event_to_update.get('end', old_event.end)): |
| 67 | + if old_event is None or not is_it_possible_to_change_dates( |
| 68 | + db, old_event, event_to_update): |
55 | 69 | return None |
56 | | - |
| 70 | + |
| 71 | + # Update database |
57 | 72 | db.query(Event).filter(Event.id == event_id).update( |
58 | 73 | event_to_update, synchronize_session=False) |
59 | 74 | db.commit() |
| 75 | + |
60 | 76 | # TODO: Send emails to recipients. |
61 | 77 | except (AttributeError, SQLAlchemyError, TypeError): |
62 | 78 | return None |
|
0 commit comments