Merge pull request #401 from vil02/simplify_to_fortran90

style: shorten `to_fortran90`

Author: vil02

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "string-to-code"
-version = "0.4.0"
+version = "0.4.1"
 description = "Generates a piece of messy code in a given language displaying a given string "
 authors = ["piotr.idzik <[email protected]>"]
 readme = "./string_to_code/README.md"

string_to_code/to_fortran90.py

@@ -12,13 +12,13 @@
 def _atom_to_code(in_atom: core.Atom) -> str:
     if in_atom.atom_char == "\n":
         return "write (*, *)"
-    res = 'write (*, "(A)", advance="no") '
-    if in_atom.atom_char == '"':
-        return res + "'\"'"
-    if in_atom.atom_char == "\t":
-        return res + "char(9)"
-
-    return res + f'"{in_atom.atom_char}"'
+    special_chars = {
+        '"': "'\"'",
+        "\t": "char(9)",
+    }
+    return 'write (*, "(A)", advance="no") ' + special_chars.get(
+        in_atom.atom_char, f'"{in_atom.atom_char}"'
+    )
 
 
 _function_call_str = utils.get_function_call_str_fun(_get_function_name, "call ", "()")
@@ -32,9 +32,7 @@ def _atom_to_code(in_atom: core.Atom) -> str:
 
 
 def _merge_to_full_function(in_function_name: str, in_function_body: str) -> str:
-    body_str = ""
-    if in_function_body:
-        body_str = in_function_body + "\n"
+    body_str = in_function_body + "\n" if in_function_body else ""
     return f"""    subroutine {in_function_name}()
 {body_str}    end subroutine {in_function_name}
 """
@@ -52,12 +50,12 @@ def _main_call_to_code(in_initial_call: core.InitialCall, **kwargs) -> str:
 
 
 def _join_to_final(main_call: str, function_definitions: list[str], **_kwargs) -> str:
-    main_call_str = ""
-    if main_call:
-        main_call_str = "\n    " + main_call
-    contains_str = "\n"
-    if function_definitions:
-        contains_str = "\ncontains\n" + "\n".join(function_definitions)
+    main_call_str = "\n    " + main_call if main_call else ""
+    contains_str = (
+        "\ncontains\n" + "\n".join(function_definitions)
+        if function_definitions
+        else "\n"
+    )
     return f"""program main
     implicit none{main_call_str}{contains_str}end program main
 """