Feature/time coordinator (#297)

Author: TamarBerger

app/database/models.py

@@ -487,6 +487,14 @@ class Quote(Base):
     author = Column(String)
 
 
+class Country(Base):
+    __tablename__ = "countries"
+
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String, nullable=False, unique=True)
+    timezone = Column(String, nullable=False)
+
+
 class Comment(Base):
     __tablename__ = "comments"
 

app/internal/event.py

@@ -1,3 +1,4 @@
+import functools
 import logging
 import re
 from typing import List, NamedTuple, Set, Union
@@ -11,7 +12,8 @@
 from sqlalchemy.orm import Session
 from starlette.status import HTTP_400_BAD_REQUEST
 
-from app.database.models import Event
+from app.database.models import Country, Event
+from app.resources.countries import countries
 
 ZOOM_REGEX = re.compile(r"https://.*?\.zoom.us/[a-z]/.[^.,\b\s]+")
 
@@ -114,6 +116,43 @@ def get_messages(
     return messages
 
 
+def add_countries_to_db(session: Session) -> None:
+    """
+    Adding all new countries to the "Country" table in the database.
+    Information is based on the "countries" list.
+    (The list is located in app/resources/countries.py)
+    Names are described either as:
+    "Country Name, City Name" or
+    "Country Name" solely.
+    Timezones are described as "Continent/ City Name"
+    for example:
+        name: Israel, Jerusalem
+        timezone: Asia/Jerusalem
+    """
+    for country in countries:
+        partial_name = country["name"]
+        for capital_city in country["timezones"]:
+            capital_city_name = capital_city.split("/")[-1]
+            if partial_name != capital_city_name:
+                name = partial_name + ", " + capital_city_name
+            else:
+                name = capital_city_name
+            new_country = Country(name=name, timezone=str(capital_city))
+            session.merge(new_country)
+    session.commit()
+
+
+@functools.lru_cache
+def get_all_countries_names(session: Session) -> List[str]:
+    """
+    Returns a cached list of the countries names.
+    """
+    db_entity = session.query(Country).first()
+    if not db_entity:
+        add_countries_to_db(session=session)
+    return session.query(Country.name).all()
+
+
 async def get_location_coordinates(
     address: str,
 ) -> Union[Location, str]: