Feature/out of office

app/database/models.py

@@ -380,6 +380,16 @@ class Quote(Base):
     author = Column(String)
 
 
+class OutOfOffice(Base):
+    __tablename__ = "out_of_office"
+
+    id = Column(Integer, primary_key=True, index=True)
+    user_id = Column(Integer, ForeignKey("users.id"))
+    start_date = Column(DateTime, nullable=False)
+    end_date = Column(DateTime, nullable=False)
+    status = Column(String, nullable=False)
+
+
 class Comment(Base):
     __tablename__ = "comments"
 

app/internal/event.py

@@ -1,6 +1,6 @@
 import logging
 import re
-from typing import List, Set
+from typing import List, Set, Tuple
 
 from email_validator import EmailSyntaxError, validate_email
 from fastapi import HTTPException
@@ -87,6 +87,7 @@ def get_messages(
     session: Session,
     event: Event,
     uninvited_contacts: Set[str],
+    out_of_office_users: List[Tuple[str, str]],
 ) -> List[str]:
     messages = []
     if uninvited_contacts:
@@ -100,4 +101,14 @@ def get_messages(
             f"Same event happened {weeks_diff} weeks before too. "
             f"Want to create another one {weeks_diff} after too?",
         )
+
+    if out_of_office_users:
+        username = 0
+        email = 1
+        messages.append("Out of office:")
+        for user in out_of_office_users:
+            messages.append(
+                f"Username: {user[username]}, " f"Email: {user[email]}",
+            )
+
     return messages

app/internal/out_of_office.py

@@ -0,0 +1,125 @@
+from datetime import datetime
+from sqlalchemy.orm import Session
+from typing import List
+
+from app.database.models import User, OutOfOffice
+
+DATETIME_FORMAT = "%Y-%m-%d %H:%M"
+START_DATE = "start_date"
+END_DATE = "end_date"
+START_TIME = "start_time"
+END_TIME = "end_time"
+
+
+def get_who_is_out_of_office(
+    session: Session,
+    event_start_date: datetime,
+    invited_emails: List[str],
+):
+    """
+    Get who is out of office
+
+    Args:
+        session: db session
+        event_start_date: event start date
+        invited_emails: invited emails
+
+    Returns:
+        Users who are out of office at the event date
+    """
+    out_of_office_users = (
+        session.query(User.username, User.email)
+        .join(OutOfOffice)
+        .filter(User.email.in_(invited_emails))
+        .filter(
+            OutOfOffice.start_date <= event_start_date,
+            OutOfOffice.end_date >= event_start_date,
+        )
+        .filter(OutOfOffice.status == "On")
+        .all()
+    )
+    return out_of_office_users
+
+
+def insert_new_out_of_office(out_of_office_data, user: User, session: Session):
+    out = get_out_of_office_template(
+        user_id=user.id,
+        start_date=get_date_formatted(
+            out_of_office_data,
+            START_DATE,
+            START_TIME,
+            DATETIME_FORMAT,
+        ),
+        end_date=get_date_formatted(
+            out_of_office_data,
+            END_DATE,
+            END_TIME,
+            DATETIME_FORMAT,
+        ),
+        status="On",
+    )
+    session.add(out)
+
+
+def get_out_of_office_template(
+    user_id,
+    start_date=None,
+    end_date=None,
+    status="Off",
+):
+    return OutOfOffice(
+        user_id=user_id,
+        start_date=start_date,
+        end_date=end_date,
+        status=status,
+    )
+
+
+def get_date_formatted(out_of_office_data, date, time, date_format):
+    return datetime.strptime(
+        out_of_office_data[date] + " " + out_of_office_data[time],
+        date_format,
+    )
+
+
+def update_out_of_office(
+    out_of_office_data_from_req,
+    out_of_office_data_from_db,
+):
+    activate_out_of_office = "1"
+
+    if out_of_office_data_from_req["outOfOffice"] == activate_out_of_office:
+        out_of_office_data_from_db.start_date = get_date_formatted(
+            out_of_office_data_from_req,
+            START_DATE,
+            START_TIME,
+            DATETIME_FORMAT,
+        )
+        out_of_office_data_from_db.end_date = get_date_formatted(
+            out_of_office_data_from_req,
+            END_DATE,
+            END_TIME,
+            DATETIME_FORMAT,
+        )
+        out_of_office_data_from_db.status = "On"
+    else:
+        out_of_office_data_from_db.status = "Off"
+
+
+def update_out_of_office_status_to_off(out_of_office_data, session: Session):
+    """
+    Update out of office status to off if out of office date passed
+
+    Args:
+        out_of_office_data: Out of office data from db
+        session: db session
+
+    Returns:
+        out_of_office_data object
+    """
+    if out_of_office_data:
+        if out_of_office_data.status == "On":
+            if out_of_office_data.end_date < datetime.now():
+                out_of_office_data.status = "Off"
+                session.commit()
+    return out_of_office_data

app/routers/event.py

@@ -24,13 +24,15 @@
 )
 from app.internal import comment as cmt
 from app.internal.emotion import get_emotion
+from app.internal.out_of_office import get_who_is_out_of_office
 from app.internal.privacy import PrivacyKinds
 from app.internal.utils import create_model, get_current_user
 from app.routers.categories import get_user_categories
 
 EVENT_DATA = Tuple[Event, List[Dict[str, str]], str]
 TIME_FORMAT = "%Y-%m-%d %H:%M"
 START_FORMAT = "%A, %d/%m/%Y %H:%M"
+
 UPDATE_EVENTS_FIELDS = {
     "title": str,
     "start": dt,
@@ -44,7 +46,6 @@
     "category_id": (int, type(None)),
 }
 
-
 router = APIRouter(
     prefix="/event",
     tags=["event"],
@@ -153,7 +154,17 @@ async def create_new_event(
         privacy=privacy,
     )
 
-    messages = get_messages(session, event, uninvited_contacts)
+    out_of_office_users = get_who_is_out_of_office(
+        session,
+        start,
+        invited_emails,
+    )
+    messages = get_messages(
+        session,
+        event,
+        uninvited_contacts,
+        out_of_office_users,
+    )
     return RedirectResponse(
         router.url_path_for("eventview", event_id=event.id)
         + f'?messages={"---".join(messages)}',