Feature/international days (#210)

Author: liaarbel

.gitignore

@@ -1,5 +1,6 @@
 dev.db
 test.db
+.idea
 config.py
 
 # Byte-compiled / optimized / DLL files

app/database/models.py

@@ -422,6 +422,15 @@ class Joke(Base):
     text = Column(String, nullable=False)
 
 
+class InternationalDays(Base):
+    __tablename__ = "international_days"
+
+    id = Column(Integer, primary_key=True, index=True)
+    day = Column(Integer, nullable=False)
+    month = Column(Integer, nullable=False)
+    international_day = Column(String, nullable=False)
+
+
 # insert language data
 
 # Credit to adrihanu   https://stackoverflow.com/users/9127249/adrihanu

app/internal/international_days.py

@@ -0,0 +1,40 @@
+from datetime import datetime
+from typing import Optional, Dict, Union
+
+from sqlalchemy.orm import Session
+from sqlalchemy.sql.expression import func
+
+from app.database.models import InternationalDays
+
+
+def get_international_day(
+        international_day: Dict[str, Union[str, int]]
+) -> InternationalDays:
+    """Returns an international day object from the dictionary data.
+
+        Args:
+            international_day: A dictionary international day
+                related information.
+
+        Returns:
+            A new international day object.
+    """
+    return InternationalDays(
+        day=international_day["day"],
+        month=international_day["month"],
+        international_day=international_day["international_day"],
+    )
+
+
+def get_international_day_per_day(
+        session: Session, date: datetime
+) -> Optional[InternationalDays]:
+    day_num = date.day
+    month = date.month
+    international_day = (session.query(InternationalDays)
+                         .filter(InternationalDays.day == day_num)
+                         .filter(InternationalDays.month == month)
+                         .order_by(func.random())
+                         .first()
+                         )
+    return international_day

app/internal/json_data_loader.py

@@ -5,8 +5,8 @@
 from loguru import logger
 from sqlalchemy.orm import Session
 
-from app.database.models import Base, Joke, Quote, Zodiac
-from app.internal import daily_quotes, jokes, zodiac
+from app.database.models import Base, InternationalDays, Joke, Quote, Zodiac
+from app.internal import daily_quotes, international_days, jokes, zodiac
 
 
 def load_to_database(session: Session) -> None:
@@ -35,6 +35,13 @@ def load_to_database(session: Session) -> None:
         daily_quotes.get_quote,
     )
 
+    _insert_into_database(
+        session,
+        'app/resources/international_days.json',
+        InternationalDays,
+        international_days.get_international_day,
+    )
+
     _insert_into_database(
         session,
         'app/resources/jokes.json',