forked from JOJ/Joint-Teapot
		
	feat: InterceptHandler with diagnose and backtrace
This commit is contained in:
		
							parent
							
								
									cae195a4b5
								
							
						
					
					
						commit
						04995f1934
					
				|  | @ -11,6 +11,11 @@ from joint_teapot.config import settings | ||||||
| 
 | 
 | ||||||
| # recipe from https://loguru.readthedocs.io/en/stable/overview.html#entirely-compatible-with-standard-logging | # recipe from https://loguru.readthedocs.io/en/stable/overview.html#entirely-compatible-with-standard-logging | ||||||
| class InterceptHandler(logging.Handler): | class InterceptHandler(logging.Handler): | ||||||
|  |     def __init__(self, diagnose: bool = True, backtrace: bool = True): | ||||||
|  |         super().__init__() | ||||||
|  |         self.diagnose = diagnose | ||||||
|  |         self.backtrace = backtrace | ||||||
|  | 
 | ||||||
|     def emit(self, record: logging.LogRecord) -> None: |     def emit(self, record: logging.LogRecord) -> None: | ||||||
|         try: |         try: | ||||||
|             level = logger.level(record.levelname).name |             level = logger.level(record.levelname).name | ||||||
|  | @ -25,7 +30,7 @@ class InterceptHandler(logging.Handler): | ||||||
|             depth += 1 |             depth += 1 | ||||||
| 
 | 
 | ||||||
|         logger.opt(depth=depth, exception=record.exc_info).log( |         logger.opt(depth=depth, exception=record.exc_info).log( | ||||||
|             level, record.getMessage() |             level, record.getMessage(), diagnose=self.diagnose, backtrace=self.backtrace | ||||||
|         ) |         ) | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
|  | @ -35,7 +40,9 @@ def set_logger( | ||||||
|     diagnose: bool = True, |     diagnose: bool = True, | ||||||
|     backtrace: bool = True, |     backtrace: bool = True, | ||||||
| ) -> None: | ) -> None: | ||||||
|     logging.basicConfig(handlers=[InterceptHandler()], level=0, force=True) |     logging.basicConfig( | ||||||
|  |         handlers=[InterceptHandler(diagnose, backtrace)], level=0, force=True | ||||||
|  |     ) | ||||||
|     logger.remove() |     logger.remove() | ||||||
|     logger.add( |     logger.add( | ||||||
|         stderr, |         stderr, | ||||||
|  |  | ||||||
		Loading…
	
		Reference in New Issue
	
	Block a user