Leftover from merge, updated readme
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@@ -2,12 +2,10 @@ from .file import FileHelper
|
||||
from .text import TextHelper
|
||||
from .token_counter import count_tokens
|
||||
from .exercises import ExercisesHelper
|
||||
from .logger import LoggerHelper
|
||||
|
||||
__all__ = [
|
||||
"FileHelper",
|
||||
"TextHelper",
|
||||
"count_tokens",
|
||||
"ExercisesHelper",
|
||||
"LoggerHelper"
|
||||
]
|
||||
|
||||
@@ -1,23 +0,0 @@
|
||||
import logging
|
||||
from functools import wraps
|
||||
|
||||
|
||||
class LoggerHelper:
|
||||
|
||||
@staticmethod
|
||||
def suppress_loggers():
|
||||
def decorator(f):
|
||||
@wraps(f)
|
||||
def wrapped(*args, **kwargs):
|
||||
root_logger = logging.getLogger()
|
||||
original_level = root_logger.level
|
||||
|
||||
root_logger.setLevel(logging.ERROR)
|
||||
|
||||
try:
|
||||
return f(*args, **kwargs)
|
||||
finally:
|
||||
root_logger.setLevel(original_level)
|
||||
|
||||
return wrapped
|
||||
return decorator
|
||||
@@ -8,11 +8,12 @@ import pdfplumber
|
||||
from fastapi import UploadFile
|
||||
|
||||
from app.services.abc import ILLMService
|
||||
from app.helpers import LoggerHelper, FileHelper
|
||||
from app.helpers import FileHelper
|
||||
from app.mappers import LevelMapper
|
||||
|
||||
from app.dtos.exams.level import Exam
|
||||
from app.dtos.sheet import Sheet
|
||||
from app.utils import suppress_loggers
|
||||
|
||||
|
||||
class UploadLevelModule:
|
||||
@@ -42,7 +43,7 @@ class UploadLevelModule:
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
@LoggerHelper.suppress_loggers()
|
||||
@suppress_loggers()
|
||||
def _check_pdf_for_images(pdf_path: str) -> bool:
|
||||
with pdfplumber.open(pdf_path) as pdf:
|
||||
for page in pdf.pages:
|
||||
|
||||
@@ -1,5 +1,7 @@
|
||||
from .handle_exception import handle_exception
|
||||
from .logger import suppress_loggers
|
||||
|
||||
__all__ = [
|
||||
"handle_exception"
|
||||
"handle_exception",
|
||||
"suppress_loggers"
|
||||
]
|
||||
|
||||
20
app/utils/logger.py
Normal file
20
app/utils/logger.py
Normal file
@@ -0,0 +1,20 @@
|
||||
import logging
|
||||
from functools import wraps
|
||||
|
||||
|
||||
def suppress_loggers():
|
||||
def decorator(f):
|
||||
@wraps(f)
|
||||
def wrapped(*args, **kwargs):
|
||||
root_logger = logging.getLogger()
|
||||
original_level = root_logger.level
|
||||
|
||||
root_logger.setLevel(logging.ERROR)
|
||||
|
||||
try:
|
||||
return f(*args, **kwargs)
|
||||
finally:
|
||||
root_logger.setLevel(original_level)
|
||||
|
||||
return wrapped
|
||||
return decorator
|
||||
Reference in New Issue
Block a user