Adding minor updates

lcsrodriguez · Feb 16, 2024 · 0bd5b90 · 0bd5b90
1 parent 08d6d7b
commit 0bd5b90
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/earnings/constants.py b/earnings/constants.py
@@ -1,4 +1,5 @@
 from enum import Enum
+
 MAIN_URL: str = "https://www.earningswhispers.com"
 DEFAULT_POST_OPTIONS: dict = {"expect": "Meet", "vote": "flat"}
 

diff --git a/earnings/main.py b/earnings/main.py
@@ -113,7 +113,7 @@ def __new__(cls, *args, **kwargs):
         return cls.instance
 
     def __init__(self) -> None:
-        self.outputType: Output = Output.DICT
+        self.outputType: Output = Output.DICT  # Default output type
 
     @outputFormat
     def getEarningsByDay(self, day: Union[str, datetime.datetime, datetime.date]) -> dict:
@@ -125,6 +125,8 @@ def getEarningsByDay(self, day: Union[str, datetime.datetime, datetime.date]) ->
         if day.weekday() in [5, 6]:
             raise Exception("weekend")
         r = get(f"{MAIN_URL}/api/caldata/{day.isoformat().replace('-', '')}")
+        if r.status_code:
+            return None
         return r.json()
 
     @outputFormat
@@ -161,8 +163,8 @@ def _premCheck(self) -> bool:
     def __str__(self) -> str:
         return f"Earnings ({self._sym})"
 
-    def __repr__(self) -> None:
-        print(self.__str__())
+    def __repr__(self) -> str:
+        return self.__str__()
 
     def __hash__(self) -> int:
         return hash(self._sym)