fix: Module import in modules with spaces

SEKOIA-IO · Oct 20, 2023 · 8206564 · 8206564
1 parent 96aa443
commit 8206564
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sekoia_automation/scripts/new_module/template/{{cookiecutter.module_dir}}/main.py b/sekoia_automation/scripts/new_module/template/{{cookiecutter.module_dir}}/main.py
@@ -1,4 +1,4 @@
-from {{cookiecutter.module_name.lower()}}_modules import {{cookiecutter.module_name.capitalize()}}Module
+from {{cookiecutter.module_name.lower().replace(" ", "_")}}_modules import {{cookiecutter.module_name.title().replace(" ", "")}}Module
 
 if __name__ == "__main__":
     module = {{cookiecutter.module_name.title().replace(" ", "")}}Module()