Fix static analysis issues in arxiv_category_converter.py - formatting and line length

Author: Goziee-git

dev/arxiv_category_converter.py

@@ -3,59 +3,71 @@
 ArXiv category code to user-friendly name converter.
 Called by arxiv_fetch.py to convert category codes to readable names.
 """
+# Standard library
 import csv
 import os
+
+# Third-party
 import yaml
 
+
 def load_category_mapping(data_dir):
     """Load category code to label mapping from YAML file."""
     mapping_file = os.path.join(data_dir, "arxiv_category_map.yaml")
-    
+
     if not os.path.exists(mapping_file):
         return {}
-    
+
     try:
-        with open(mapping_file, 'r') as f:
+        with open(mapping_file, "r") as f:
             return yaml.safe_load(f) or {}
     except Exception:
         return {}
 
+
 def convert_categories_to_friendly_names(input_file, output_file, data_dir):
     """
     Convert category codes in CSV to user-friendly names.
-    
+
     Args:
         input_file: Path to input CSV with category codes
         output_file: Path to output CSV with friendly names
         data_dir: Directory containing arxiv_category_map.yaml
     """
     if not os.path.exists(input_file):
         return
-    
+
     # Load category mapping
     category_mapping = load_category_mapping(data_dir)
-    
-    with open(input_file, 'r') as infile, open(output_file, 'w', newline='') as outfile:
+
+    with (
+        open(input_file, "r") as infile,
+        open(output_file, "w", newline="") as outfile,
+    ):
         reader = csv.DictReader(infile)
-        
+
         # Create new fieldnames with both code and label
         fieldnames = []
         for field in reader.fieldnames:
             fieldnames.append(field)
-            if field == 'CATEGORY':
-                fieldnames.append('CATEGORY_LABEL')
-        
-        writer = csv.DictWriter(outfile, fieldnames=fieldnames, dialect='unix')
+            if field == "CATEGORY":
+                fieldnames.append("CATEGORY_LABEL")
+
+        writer = csv.DictWriter(outfile, fieldnames=fieldnames, dialect="unix")
         writer.writeheader()
-        
+
         for row in reader:
-            if 'CATEGORY' in row:
-                category_code = row['CATEGORY']
-                # Convert code to label, fallback to uppercase first part if not found
+            if "CATEGORY" in row:
+                category_code = row["CATEGORY"]
+                # Convert code to label, fallback to uppercase first part
                 category_label = category_mapping.get(
                     category_code,
-                    category_code.split('.')[0].upper() if category_code and '.' in category_code else category_code
+                    (
+                        category_code.split(".")[0].upper()
+                        if category_code and "." in category_code
+                        else category_code
+                    ),
                 )
-                row['CATEGORY_LABEL'] = category_label
-            
+                row["CATEGORY_LABEL"] = category_label
+
             writer.writerow(row)