robotframework-openapitools 1.0.0b4__py3-none-any.whl → 1.0.1__py3-none-any.whl

openapi_libgen/generator.py

@@ -8,8 +8,6 @@ from openapi_libgen.spec_parser import get_keyword_data
 from OpenApiLibCore.models import OpenApiObject
 
 HERE = Path(__file__).parent.resolve()
-INIT_TEMPLATE_PATH = HERE / "templates/__init__.jinja"
-LIBRARY_TEMPLATE_PATH = HERE / "templates/library.jinja"
 
 
 def load_openapi_spec(
@@ -18,7 +16,7 @@ def load_openapi_spec(
     def recursion_limit_handler(
         limit: int, refstring: str, recursions: object
     ) -> object:  # pylint: disable=unused-argument
-        return recursion_default
+        return recursion_default  # pragma: no cover
 
     parser = ResolvingParser(
         source,
@@ -69,7 +67,7 @@ def generate(
     return f"Generated {library_name} at {output_folder.resolve().as_posix()}/{module_name}"
 
 
-if __name__ == "__main__":
+if __name__ == "__main__":  # pragma: no cover
     source = sys.argv[1]
     destination = Path(sys.argv[2])
     library_name = sys.argv[3]

openapi_libgen/parsing_utils.py

@@ -8,16 +8,20 @@ def remove_unsafe_characters_from_string(string: str) -> str:
         string_iterator = iter(string)
         capitalize_next_character = False
 
-        for character in string_iterator:
-            if character.isalpha():
-                yield character
-                break
+        # The first character must be A-z or _
+        first_character = next(string_iterator, "_")
+        if first_character.isalpha() or first_character == "_":
+            yield first_character
+        elif first_character.isnumeric():
+            yield "_" + first_character
 
         for character in string_iterator:
             if character.isalnum():
                 if capitalize_next_character:
                     capitalize_next_character = False
-                yield character
+                    yield character.upper()
+                else:
+                    yield character
 
             elif not capitalize_next_character:
                 capitalize_next_character = True

openapi_libgen/spec_parser.py

@@ -79,7 +79,7 @@ def get_keyword_signature(operation_details: OperationDetails) -> str:
     else:
         keyword_name = remove_unsafe_characters_from_string(
             f"{operation_details.method}_{operation_details.path}"
-        )
+        ).lower()
 
     parameters = operation_details.parameters or []
     path_parameters = [p for p in parameters if p.in_ == "path" and p.schema_]
@@ -93,7 +93,7 @@ def get_keyword_signature(operation_details: OperationDetails) -> str:
     keyword_argument_names: set[str] = set()
 
     for parameter in path_parameters:
-        annotation = f"{parameter.schema_.annotation_string} = UNSET"
+        annotation = f"{parameter.schema_.annotation_string} = UNSET"  # type: ignore[union-attr]
         safe_name = get_safe_name_for_oas_name(parameter.name)
         keyword_argument_names.add(safe_name)
         argument = f", {safe_name}: {annotation}"
@@ -120,7 +120,7 @@ def get_keyword_signature(operation_details: OperationDetails) -> str:
             arguments += argument
 
     for parameter in query_parameters:
-        annotation = f"{parameter.schema_.annotation_string} = UNSET"
+        annotation = f"{parameter.schema_.annotation_string} = UNSET"  # type: ignore[union-attr]
         safe_name = get_safe_name_for_oas_name(parameter.name)
         if safe_name in keyword_argument_names:
             safe_name = "query_" + safe_name
@@ -129,7 +129,7 @@ def get_keyword_signature(operation_details: OperationDetails) -> str:
         arguments += argument
 
     for parameter in header_parameters:
-        annotation = f"{parameter.schema_.annotation_string} = UNSET"
+        annotation = f"{parameter.schema_.annotation_string} = UNSET"  # type: ignore[union-attr]
         safe_name = get_safe_name_for_oas_name(parameter.name)
         if safe_name in keyword_argument_names:
             safe_name = "header_" + safe_name