fix tests not running

Author: CodyKoInABox

cli/commands/analyze.py

@@ -23,11 +23,11 @@ def analyze_command(file, all, json_output, LANG_FILE):
 
     # dictionary for the stats
     stats_labels = {
-        "line_count": messages.get("line_count_option", "Line Count"),
-        "function_count": messages.get("function_count_option", "Function Count"),
-        "comment_line_count": messages.get("comment_line_count_option", "Comment Line Count"),
-        "inline_comment_count": messages.get("inline_comment_count_option", "Inline Comment Count"),
-        "indentation_level": messages.get("indentation_level_option", "Indentation Analysis")
+        "line_count": get_translation("line_count_option"),
+        "function_count": get_translation("function_count_option"),
+        "comment_line_count": get_translation("comment_line_count_option"),
+        "inline_comment_count": get_translation("inline_comment_count_option"),
+        "indentation_level": get_translation("indentation_level_option")
     }
 
     # If --all flag is used, skip the selection menu and use all stats

cli/commands/export/export.py

@@ -32,7 +32,7 @@ def export_results(results, format_type, output_file, messages):
             with open(output_file, "w", encoding="utf-8", newline="") as f:
                 writer = csv.writer(f)
                 # Write header
-                writer.writerow(["Metric", "Value"])
+                writer.writerow([get_translation("metric"), get_translation("value")])
                 # Write data
                 for key, value in results.items():
                     if isinstance(value, (int, float, str)):
@@ -80,22 +80,20 @@ def export_results(results, format_type, output_file, messages):
         return True
 
     except Exception as e:
-        print(f"{messages.get('export_error', 'Export error')}: {str(e)}")
+        print(f"[red]{get_translation('error')}[/]: {str(e)}")
         return False
 
 def export_command(file, format_type, output, LANG_FILE):
     """
     Export analysis results to a file.
     """
-    # Load translations
-    messages = get_translation(LANG_FILE)
     console = Console()
 
     # Validate format type
     valid_formats = ["json", "csv", "markdown", "html"]
     if format_type not in valid_formats:
-        console.print(f"[red]{messages.get('invalid_format', 'Invalid format')}[/] {format_type}")
-        console.print(f"{messages.get('valid_formats', 'Valid formats')}: {', '.join(valid_formats)}")
+        console.print(f"[red]{get_translation('invalid_format')}[/] {format_type}")
+        console.print(f"{get_translation('valid_formats')}: {', '.join(valid_formats)}")
         return
 
     try:
@@ -109,12 +107,12 @@ def export_command(file, format_type, output, LANG_FILE):
             output = f"{base_name}_analysis.{format_type}"
 
         # Export results
-        success = export_results(results, format_type, output, messages)
+        success = export_results(results, format_type, output, None)
 
         if success:
-            console.print(f"[green]{messages.get('export_success', 'Export successful')}[/]: {output}")
+            console.print(f"[green]{get_translation('export_success')}[/]: {output}")
         else:
-            console.print(f"[red]{messages.get('export_failed', 'Export failed')}[/]")
+            console.print(f"[red]{get_translation('export_failed')}[/]")
 
     except Exception as e:
-        console.print(f"[red]{messages.get('error', 'Error')}[/]: {str(e)}")
+        console.print(f"[red]{get_translation('error')}[/]: {str(e)}")

cli/commands/hello.py

@@ -4,10 +4,6 @@
 
 
 def hello_command(LANG_FILE):
-
-    # load translations
-    messages = get_translation(LANG_FILE)
-
     # print the hello message
-    print(messages["welcome"])
-    print(messages["description"])
+    print(get_translation("welcome"))
+    print(get_translation("description"))

cli/commands/version.py

@@ -9,16 +9,13 @@ def version_command(LANG_FILE, CURRENT_DIR):
     Display the current version of the application.
     """
 
-    # load translations
-    messages = get_translation(LANG_FILE)
-    
     try:
         # Get the path to setup.py in the parent directory
         setup_path = os.path.join(os.path.dirname(CURRENT_DIR), "setup.py")
 
         # Check if setup.py exists
         if not os.path.exists(setup_path):
-            print(f"[red]{messages.get('setup_not_found', 'Error: setup.py not found.')}")
+            print(f"[red]{get_translation('setup_not_found')}")
             return
 
         # Read setup.py to extract version
@@ -33,9 +30,9 @@ def version_command(LANG_FILE, CURRENT_DIR):
 
         # Display version information
         if version_info:
-            print(f"[green]{messages.get('version_info', 'SpiceCode Version:')}[/] {version_info}")
+            print(f"[green]{get_translation('version_info')}[/] {version_info}")
         else:
-            print(f"[yellow]{messages.get('version_not_found', 'Version information not found in setup.py')}")
+            print(f"[yellow]{get_translation('version_not_found')}")
 
     except Exception as e:
-        print(f"[red]{messages.get('error', 'Error:')}[/] {e}")
+        print(f"[red]{get_translation('error')}[/] {e}")

utils/get_translation.py

@@ -2,26 +2,35 @@
 import os
 
 # this will load the translations
-def get_translation(LANG_FILE):
+def get_translation(key):
+    """
+    Get a translated message for the given key.
+    
+    Args:
+        key (str): The message key to translate
+        
+    Returns:
+        str: The translated message
+    """
+    # read the lang file to see what language was set by user
+    LANG_FILE = os.path.join(os.path.dirname(os.path.dirname(__file__)), "cli", "lang.txt")
 
-    # read the lang file to see what langague was set by user
     if os.path.exists(LANG_FILE):
-
         # open the lang file
         with open(LANG_FILE, "r") as file:
-                
-                # read the lang file
-                lang = file.read().strip()
-
-              
-                if not lang:
-                    lang = 'en' # if file is empty, default to english
-        
+            # read the lang file
+            lang = file.read().strip()
+            
+            if not lang:
+                lang = 'en'  # if file is empty, default to english
     else:
-        lang = 'en'  # default to English if there is not file but there will always be a file this is just in case ALSO THIS IS SO @icrcode DOESNT COMPLAIN ABOUT MY CODE NOT BEING CLEAN AND WHATEVER
+        lang = 'en'  # default to English if there is no file
 
     # this is actually import the translations
     try:
-        return importlib.import_module(f"cli.translations.{lang}").messages
+        messages = importlib.import_module(f"cli.translations.{lang}").messages
     except ModuleNotFoundError:
-        return importlib.import_module("cli.translations.en").messages  # default to English if any errors
+        messages = importlib.import_module("cli.translations.en").messages  # default to English if any errors
+
+    # Return the specific message for the key, or the key itself if not found
+    return messages.get(key, key)