Fix incorrect merge of _public_module_end usage

gauge-sh · Feb 12, 2024 · c17c602 · c17c602
1 parent 6e6d8b2
commit c17c602
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/modguard/init.py b/modguard/init.py
@@ -100,7 +100,7 @@ def init_project(root: str, exclude_paths: Optional[list[str]] = None):
         try:
             if write_op.operation == WriteOperation.BOUNDARY:
                 add_boundary(write_op.location)
-            if write_op.operation == WriteOperation.PUBLIC:
+            elif write_op.operation == WriteOperation.PUBLIC:
                 mark_as_public(write_op.location, write_op.member_name)
         except errors.ModguardError:
             print(f"Error marking {write_op.operation} in {write_op.location}")
diff --git a/modguard/parsing/public.py b/modguard/parsing/public.py
@@ -227,8 +227,8 @@ def mark_as_public(file_path: str, member_name: str = ""):
     if not member_name:
         fs.write_file(
             file_path,
-            _public_module_end(should_import=not modguard_public_is_imported)
-            + file_content,
+            file_content
+            + _public_module_end(should_import=not modguard_public_is_imported),
         )
         return