Merge pull request #242 from webtech-network/229-td-replace-print-calls-with-logging

229 td replace print calls with logging

Author: ArthurCRodrigues

autograder/services/template_library_service.py

@@ -7,10 +7,13 @@
 This is a singleton service that should be instantiated once at application startup.
 """
 
+import logging
 from typing import Dict, List, Optional
 from autograder.models.abstract.template import Template
 from autograder.template_library import TEMPLATE_REGISTRY, get_template_instance
 
+logger = logging.getLogger(__name__)
+
 
 class TemplateLibraryService:
     """
@@ -50,7 +53,7 @@ def _load_all_templates(self):
                 self._templates[template_name] = get_template_instance(template_name)
             except Exception as e:
                 # Log the error but continue loading other templates
-                print(f"Warning: Failed to load template '{template_name}': {e}")
+                logger.warning("Failed to load template '%s': %s", template_name, e)
 
     @classmethod
     def get_instance(cls) -> 'TemplateLibraryService':

autograder/services/upstash_driver.py

@@ -1,8 +1,11 @@
 import json
+import logging
 import os
 from dotenv import load_dotenv
 from upstash_redis import Redis
 
+logger = logging.getLogger(__name__)
+
 load_dotenv() #TODO: place this in application startup
 class UpstashDriver:
     def __init__(self):
@@ -49,11 +52,11 @@ def create_user(self, username: str):
             "quota": 10,
             "score": -1.0
         })
-        print(f"User '{username}' created.")
+        logger.info("User '%s' created.", username)
 
     def set_score(self, username: str, score: float):
         """Function to set the score of a user"""
         key = f"user:{username}"
         self.redis.hset(key, "score", score)
-        print(f"Score '{score}' set for user '{username}'.")
+        logger.info("Score '%s' set for user '%s'.", score, username)
 

autograder/utils/executors/ai_executor.py

@@ -1,11 +1,14 @@
 import json
+import logging
 from typing import List, Dict
 from openai import OpenAI
 from autograder.models.dataclass.test_result import TestResult
 from pydantic import BaseModel, Field
 import dotenv
 from autograder.utils.secrets_fetcher import get_secret
 
+logger = logging.getLogger(__name__)
+
 dotenv.load_dotenv()  # Load environment variables from .env file
 
 class TestInput(BaseModel):
@@ -73,20 +76,20 @@ def mapback(self):
            It finds the corresponding TestResult by matching the test name.
         """
         if not self.test_results:
-            print("No test results to map back.")
+            logger.warning("No test results to map back.")
             return
 
         for ai_result in self.test_results:
             # Find the corresponding TestResult reference by matching the test name
             matching_refs = [ref for ref in self.test_result_references if ref.test_name == ai_result.title]
             if matching_refs:
                 ref = matching_refs[0]
-                print("Found matching TestResult for AI result:",ref)
+                logger.debug("Found matching TestResult for AI result: %s", ref)
                 ref.score = ai_result.score
                 ref.report = ai_result.feedback
-                print(f"Mapped AI result '{ai_result.title}' with score {ai_result.score} to TestResult.")
+                logger.info("Mapped AI result '%s' with score %s to TestResult.", ai_result.title, ai_result.score)
             else:
-                print(f"No matching TestResult found for AI result '{ai_result.title}'.")
+                logger.warning("No matching TestResult found for AI result '%s'.", ai_result.title)
 
 
 
@@ -151,10 +154,10 @@ def stop(self):
                                 lembre-se: os nomes dos testes devem corresponder exatamente aos nomes fornecidos na lista de testes. não mude a formatação ou a estrutura dos titulos dos tests.
 
                               """
-        print("System Prompt:\n", system_prompt)
-        print("User Prompt:\n", user_prompt)
+        logger.debug("System Prompt:\n%s", system_prompt)
+        logger.debug("User Prompt:\n%s", user_prompt)
         try:
-            print("Sending AI engine batch request...\n")
+            logger.info("Sending AI engine batch request...")
             response = self.client.responses.parse(
                 model="o4-mini-2025-04-16",
                 input=[
@@ -167,12 +170,12 @@ def stop(self):
             # Extracts and logs the results
             self.test_results = response.output[1].content[0].parsed.results
             for test_result in self.test_results:
-                print(f"""{test_result}\n""")
+                logger.debug("AI test result: %s", test_result)
             self.mapback()
             return self.test_result_references
 
         except Exception as e:
-            print(f"An error occurred while running the AI tests: {e}")
+            logger.error("An error occurred while running the AI tests: %s", e)
             return []
 
 

autograder/utils/secrets_fetcher.py

@@ -3,10 +3,13 @@
 # https://aws.amazon.com/developer/language/python/
 
 import boto3
-import os
 import json
+import logging
+import os
 from botocore.exceptions import ClientError
 
+logger = logging.getLogger(__name__)
+
 def get_secret(secret_key : str, secret_name : str = None, region : str = "us-east-1"):
     """
     Fetches a secret value, adapting to the environment.
@@ -32,7 +35,7 @@ def get_secret(secret_key : str, secret_name : str = None, region : str = "us-ea
         # In production, get the key from Secrets Manager
         return _get_secret_from_manager(secret_name, secret_key, region)
     # In development, get the key from a local environment variable
-    print("Fetching secret from local environment variable...")
+    logger.debug("Fetching secret from local environment variable...")
     api_key = os.environ.get(secret_key)
     if not api_key:
         raise ValueError("Environment variable not set for development.")