alessandro trinca tornidor
		
	commited on
		
		
					Commit 
							
							·
						
						5a8199e
	
1
								Parent(s):
							
							dbcbde8
								
ci: add HEALTH_MODELS env variable to make /health_models check optional
Browse files- scripts/healthcheck.py +13 -3
    	
        scripts/healthcheck.py
    CHANGED
    
    | @@ -1,3 +1,7 @@ | |
|  | |
|  | |
|  | |
|  | |
| 1 |  | 
| 2 | 
             
            def run_healthcheck():
         | 
| 3 | 
             
                import os
         | 
| @@ -14,9 +18,15 @@ def run_healthcheck(): | |
| 14 |  | 
| 15 | 
             
                r1 = requests.get(url1)
         | 
| 16 | 
             
                app_logger.info(r1.status_code)
         | 
| 17 | 
            -
                 | 
| 18 | 
            -
                 | 
| 19 | 
            -
             | 
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
| 20 |  | 
| 21 |  | 
| 22 | 
             
            if __name__ == "__main__":
         | 
|  | |
| 1 | 
            +
            import os
         | 
| 2 | 
            +
             | 
| 3 | 
            +
            HEALTH_MODELS = os.getenv("HEALTH_MODELS", "")
         | 
| 4 | 
            +
             | 
| 5 |  | 
| 6 | 
             
            def run_healthcheck():
         | 
| 7 | 
             
                import os
         | 
|  | |
| 18 |  | 
| 19 | 
             
                r1 = requests.get(url1)
         | 
| 20 | 
             
                app_logger.info(r1.status_code)
         | 
| 21 | 
            +
                msg = f"status health:{r1.status_code}!"
         | 
| 22 | 
            +
                if HEALTH_MODELS:
         | 
| 23 | 
            +
                    r2 = requests.get(url2)
         | 
| 24 | 
            +
                    msg += f" status health_models:{r2.status_code}!"
         | 
| 25 | 
            +
                    app_logger.info(msg)
         | 
| 26 | 
            +
                    sys.exit(0) if r1.status_code == 200 and r2.status_code == 200 else sys.exit(1)
         | 
| 27 | 
            +
                app_logger.info(msg)
         | 
| 28 | 
            +
                sys.exit(0) if r1.status_code == 200 else sys.exit(1)
         | 
| 29 | 
            +
             | 
| 30 |  | 
| 31 |  | 
| 32 | 
             
            if __name__ == "__main__":
         |