Adjust sleep back to 5

gitmon/api.py

@@ -72,9 +72,12 @@ class GitHubApi:
             headers (dict): The response headers from the GitHub API.
 
         """
-        self.rate_reset = datetime.fromtimestamp(int(headers['x-ratelimit-reset']))
-        self.rate_remaining = int(headers['x-ratelimit-remaining'])
-        logger.info(f'{self.rate_remaining} calls remaining until {self.rate_reset} ({self.rate_reset - datetime.now()})')
+        try:
+            self.rate_reset = datetime.fromtimestamp(int(headers['x-ratelimit-reset']))
+            self.rate_remaining = int(headers['x-ratelimit-remaining'])
+            logger.info(f'{self.rate_remaining} calls remaining until {self.rate_reset} ({self.rate_reset - datetime.now()})')
+        except KeyError as e:
+            logger.warning(f'Could not parse headers: {str(e)}')
 
     def get_events(self, page=None):
         """Fetch GitHub events.

gitmon/logger.py

@@ -2,6 +2,9 @@
 import sys
 from loguru import logger
 
+logger.level('CREATE', no=34, color='<green>')
+logger.level('DELETE', no=35, color='<red>')
+
 
 def setup_logger():
     logger.remove()

main.py

@@ -1,4 +1,5 @@
 """GitMon"""
+import signal
 import time
 from loguru import logger
 from gitmon.api import GitHubApi
@@ -16,23 +17,37 @@ def log_event(event):
     actor = event.get('actor')
 
     if event_type == 'DeleteEvent':
-        logger.error(f'{event_type}:\thttps://github.com/{repo["name"]}, {actor["login"]}')
+        logger.log('DELETE', f'{event_type}:\thttps://github.com/{repo["name"]}, {actor["login"]}')
     else:
-        logger.info(f'{event_type}:\thttps://github.com/{repo["name"]}, {actor["login"]}')
+        logger.log('CREATE', f'{event_type}:\thttps://github.com/{repo["name"]}, {actor["login"]}')
+
+
+def sigint_handler(signum, frame):
+    """Signal handler for Ctrl+C."""
+    raise KeyboardInterrupt
 
 
 def main():
     """Main entry point."""
+    # Setup logger and signal handling
     setup_logger()
+    signal.signal(signal.SIGINT, sigint_handler)
+
     api = GitHubApi(token)
 
-    while True:
-        events = api.get_events()
-        for event in events:
-            if event['type'] in ['CreateEvent', 'DeleteEvent']:
-                log_event(event)
+    try:
+        while True:
+            events = api.get_events()
+            for event in events:
+                if (
+                    isinstance(event, dict)
+                    and event.get('type') in ['CreateEvent', 'DeleteEvent']
+                ):
+                    log_event(event)
 
-        time.sleep(5)
+            time.sleep(5)
+    except KeyboardInterrupt:
+        logger.info('Program terminating')
 
 
 if __name__ == '__main__':