Cleanly exit on SIGINT
This commit is contained in:
		
							parent
							
								
									1e958b397a
								
							
						
					
					
						commit
						1c6a45a1d2
					
				
							
								
								
									
										30
									
								
								main.py
									
									
									
									
									
								
							
							
						
						
									
										30
									
								
								main.py
									
									
									
									
									
								
							@ -1,4 +1,5 @@
 | 
			
		||||
"""GitMon"""
 | 
			
		||||
import signal
 | 
			
		||||
import time
 | 
			
		||||
from loguru import logger
 | 
			
		||||
from gitmon.api import GitHubApi
 | 
			
		||||
@ -21,21 +22,32 @@ def log_event(event):
 | 
			
		||||
        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 (
 | 
			
		||||
                isinstance(event, dict)
 | 
			
		||||
                and event.get('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(1)
 | 
			
		||||
    except KeyboardInterrupt:
 | 
			
		||||
        logger.info('Program terminating')
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
if __name__ == '__main__':
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user