From e885b5b8a8db1df3e2e9caedea1a12ce35941fc6 Mon Sep 17 00:00:00 2001
From: Ivan Puchalka <ivanpuchalka@gmail.com>
Date: Sun, 14 Apr 2024 23:36:02 -0300
Subject: [PATCH] Add valid class name generation

---
 har2locust/__main__.py   | 45 +++++++++++++++++++++++++++++++++++++++-
 tests/test_har2locust.py |  8 ++++++-
 2 files changed, 51 insertions(+), 2 deletions(-)

diff --git a/har2locust/__main__.py b/har2locust/__main__.py
index 73574a8..d1e056d 100644
--- a/har2locust/__main__.py
+++ b/har2locust/__main__.py
@@ -5,6 +5,7 @@
 import os
 import pathlib
 import sys
+import unicodedata
 from argparse import Namespace
 
 import jinja2
@@ -20,7 +21,7 @@ def __main__(arguments=None):
         args.plugins.split(",") if args.plugins else [], args.disable_plugins.split(",") if args.disable_plugins else []
     )
     har_path = pathlib.Path(args.input)
-    name = har_path.stem.replace("-", "_").replace(".", "_")  # build class name from filename
+    name = generate_class_name(har_path.stem)  # build class name from filename
     with open(har_path, encoding="utf8") as f:
         har = json.load(f)
     logging.debug(f"loaded {har_path}")
@@ -123,5 +124,47 @@ def render(name: str, values: dict) -> str:
     return py
 
 
+# Generate a valid identifier (https://docs.python.org/3.8/reference/lexical_analysis.html#identifiers) by replacing
+# invalid characters with "_".
+def generate_class_name(file_name: str) -> str:
+    VALID_STARTING_CHARACTER_CATEGORIES = ["Lu", "Ll", "Lt", "Lm", "Lo", "Nl"]
+    OTHER_ID_START_CHARACTERS = ["\u1885", "\u1886", "\u2118", "\u212e", "\u309b", "\u309c"]
+    VALID_CONTINUATION_CHARACTER_CATEGORIES = VALID_STARTING_CHARACTER_CATEGORIES + ["Mn", "Mc", "Nd", "Pc"]
+    OTHER_ID_CONTINUE_CHARACTERS = ["\u00b7", "\u0387", "\u1369", "\u1370", "\u1371", "\u19da"]
+
+    def valid_continuation_character(character: str) -> bool:
+        normalized_character = unicodedata.normalize("NFKC", character)
+        for character in normalized_character:
+            if (
+                unicodedata.category(character) not in VALID_CONTINUATION_CHARACTER_CATEGORIES
+                and character not in OTHER_ID_START_CHARACTERS + OTHER_ID_CONTINUE_CHARACTERS
+                and character != "_"
+            ):
+                return False
+        return True
+
+    def valid_starting_character(character: str) -> bool:
+        normalized_character = unicodedata.normalize("NFKC", character)
+        first = normalized_character[0]
+        if (
+            unicodedata.category(first) not in VALID_STARTING_CHARACTER_CATEGORIES
+            and first not in OTHER_ID_START_CHARACTERS
+            and first != "_"
+        ):
+            return False
+        for character in normalized_character[1:]:
+            if not valid_continuation_character(character):
+                return False
+        return True
+
+    first_character = file_name[0]
+    name = first_character if valid_starting_character(first_character) else "_"
+
+    for character in file_name[1:]:
+        name += character if valid_continuation_character(character) else "_"
+
+    return name
+
+
 if __name__ == "__main__":
     __main__()
diff --git a/tests/test_har2locust.py b/tests/test_har2locust.py
index 2b23add..e24b070 100644
--- a/tests/test_har2locust.py
+++ b/tests/test_har2locust.py
@@ -6,7 +6,7 @@
 
 import pytest
 
-from har2locust.__main__ import __main__
+from har2locust.__main__ import __main__, generate_class_name
 
 inputs_dir = pathlib.Path(__file__).parents[0] / "inputs"
 outputs_dir = pathlib.Path(__file__).parents[0] / "outputs"
@@ -186,3 +186,9 @@ def test_locust_run():
     _, stderr = proc.communicate()
     assert proc.returncode == 0, f"Bad return code {proc.returncode}, stderr: {stderr}"
     assert "--run-time limit reached" in stderr, stderr
+
+
+def test_generate_class_name_with_invalid_characters():
+    file_name = "0invalid(characters)"
+    generated_name = generate_class_name(file_name)
+    assert generated_name == "_invalid_characters_"