Leftover from merge, updated readme

This commit is contained in:
Carlos-Mesquita
2024-11-06 02:07:46 +00:00
parent f02a34fda2
commit e51cd891d2
8 changed files with 52 additions and 1316 deletions

File diff suppressed because it is too large Load Diff

View File

@@ -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"
]

View File

@@ -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

View File

@@ -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:

View File

@@ -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
View 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