style: shorten to_fortran90

Author: vil02

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}
 """
@@ -46,18 +44,20 @@ def _merge_to_full_function(in_function_name: str, in_function_body: str) -> str
 
 
 def _main_call_to_code(in_initial_call: core.InitialCall, **kwargs) -> str:
-    if in_initial_call is None:
-        return ""
-    return _call_function_or_atom(in_initial_call, **kwargs)
+    return (
+        ""
+        if in_initial_call is None
+        else _call_function_or_atom(in_initial_call, **kwargs)
+    )
 
 
 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
 """