Prevent autoflake to clean imports in module init files (#233)

* Prevent autoflake to clean imports in module init files

Co-authored-by: Simon Guigui <[email protected]>
Co-authored-by: Dylan Anthony <[email protected]>

Author: dbanty

CHANGELOG.md

@@ -12,6 +12,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Fixes
 
 - Prefix generated identifiers to allow leading digits in field names (#206 - @kalzoo).
+- Prevent autoflake from removing `__init__.py` imports during generation. (#223 - Thanks @fyhertz!)
 - Update minimum Pydantic version to support Python 3.9
 
 ### Additions

end_to_end_tests/golden-record/my_test_api_client/__init__.py

@@ -1 +1,2 @@
 """ A client library for accessing My Test API """
+from .client import AuthenticatedClient, Client

end_to_end_tests/golden-record/my_test_api_client/models/__init__.py

@@ -1 +1,9 @@
 """ Contains all the data models used in inputs/outputs """
+
+from .a_model import AModel
+from .an_enum import AnEnum
+from .an_int_enum import AnIntEnum
+from .body_upload_file_tests_upload_post import BodyUploadFileTestsUploadPost
+from .different_enum import DifferentEnum
+from .http_validation_error import HTTPValidationError
+from .validation_error import ValidationError

openapi_python_client/__init__.py

@@ -77,7 +77,7 @@ def update(self) -> Sequence[GeneratorError]:
 
     def _reformat(self) -> None:
         subprocess.run(
-            "autoflake -i -r --remove-all-unused-imports --remove-unused-variables .",
+            "autoflake -i -r --remove-all-unused-imports --remove-unused-variables --ignore-init-module-imports .",
             cwd=self.package_dir,
             shell=True,
             stdout=subprocess.PIPE,

tests/test___init__.py

@@ -382,7 +382,7 @@ def test__reformat(mocker):
     sub_run.assert_has_calls(
         [
             mocker.call(
-                "autoflake -i -r --remove-all-unused-imports --remove-unused-variables .",
+                "autoflake -i -r --remove-all-unused-imports --remove-unused-variables --ignore-init-module-imports .",
                 cwd=project.package_dir,
                 shell=True,
                 stdout=subprocess.PIPE,