Check if extension is in counter dictionary in MOSS check

mrlvsb · Mar 25, 2024 · 4b1b643 · 4b1b643
1 parent 321547a
commit 4b1b643
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/common/moss/__init__.py b/common/moss/__init__.py
@@ -67,7 +67,8 @@ def is_source_valid(logger, source) -> bool:
 def add_file(logger, moss: mosspy.Moss, file_path: str, name: str, counters):
     logger.info(f"Adding file {name} from {file_path}")
     ext = name.split('.')[-1].lower()
-    counters[ALLOWED_EXTENSIONS[ext]] += 1
+    if ext in ALLOWED_EXTENSIONS:
+        counters[ALLOWED_EXTENSIONS[ext]] += 1
     moss.addFile(file_path, name)