Fetch data from Gmail API with time range

Author: vnepomuceno

gmail_fisher/api/gateway.py

@@ -65,11 +65,21 @@ def __authenticate(cls) -> Credentials:
 
 
 class GmailGateway:
-    """Maximum number of workers for thread pool executor"""
+    @staticmethod
+    def _get_search_query(sender_emails, keywords, start_time, end_time):
+        if start_time is None or end_time is None:
+            return f"from:{sender_emails} {keywords}"
+        else:
+            return f"from:{sender_emails} {keywords} after:{start_time} before:{end_time}"
 
     @classmethod
     def list_message_ids(
-        cls, sender_emails: str, keywords: str, max_results: int
+        cls,
+        sender_emails: str,
+        keywords: str,
+        max_results: int,
+        start_time: str = None,
+        end_time: str = None,
     ) -> Iterable[str]:
         """
         For a given sender email and comma-separated keywords, retrieve the matching
@@ -82,7 +92,7 @@ def list_message_ids(
             .messages()
             .list(
                 userId="me",
-                q=f"from:{sender_emails} {keywords}",
+                q=cls._get_search_query(sender_emails, keywords, start_time, end_time),
                 maxResults=max_results,
             )
             .execute(http=GmailClient.auth_http_request())

gmail_fisher/parsers/food.py

@@ -22,9 +22,17 @@
 logger = get_logger(__name__)
 
 
-def apply_restaurant_filter(func):
+def apply_restaurant_filter(method):
+    """
+    Decorator method that wraps the 'get_restaurant' method from the food parsers
+    and applies all restaurant filters to the output of that method.
+
+    :param method: Pass the function that is being wrapped
+    :return: Filtered restaurant name
+    """
+
     def wrapper(*args, **kwargs):
-        return FoodExpenseParser.apply_restaurant_filters(func(*args, **kwargs))
+        return FoodExpenseParser.apply_restaurant_filters(method(*args, **kwargs))
 
     return wrapper
 

gmail_fisher/utils/config.py

@@ -15,4 +15,5 @@
 AUTH_PATH: Final[Path] = Path("auth/")
 
 # CONCURRENCY
+# Maximum number of workers for thread pool executor
 THREAD_POOL_MAX_WORKERS: Final[int] = 200

tests/test_gmail_gateway.py

@@ -0,0 +1,6 @@
+from gmail_fisher.api.gateway import GmailGateway
+
+
+def test_get_email_messages():
+    messages = GmailGateway.get_email_messages(sender_emails="portugal-food@bolt.eu", keywords="Delivery", max_results=10)
+    print(len(messages))