add protected_dirs to config

Author: adelevie

openapi_python_client/__init__.py

@@ -1,5 +1,6 @@
 """ Generate modern Python clients from OpenAPI """
 
+import os
 import json
 import mimetypes
 import shutil
@@ -127,7 +128,21 @@ def update(self) -> Sequence[GeneratorError]:
         if not self.package_dir.is_dir():
             return [GeneratorError(detail=f"Directory {self.package_dir} not found")]
         print(f"Updating {self.package_name}")
-        shutil.rmtree(self.package_dir)
+
+        if not self.config.protected_dirs:
+            shutil.rmtree(self.package_dir)
+        else:
+            files_and_dirs: List[str] = os.listdir(self.package_dir)
+            for protected_dir in self.config.protected_dirs:
+                files_and_dirs.remove(protected_dir)
+
+            for file_or_dir in files_and_dirs:
+                path = os.path.join(self.package_dir, file_or_dir)
+                if os.path.isfile(path):
+                    os.remove(path)
+                if os.path.isdir(path):
+                    shutil.rmtree(path)
+
         self._create_package()
         self._build_models()
         self._build_api()
@@ -170,7 +185,7 @@ def _get_errors(self) -> List[GeneratorError]:
         return errors
 
     def _create_package(self) -> None:
-        self.package_dir.mkdir()
+        self.package_dir.mkdir(exist_ok=True)
         # Package __init__.py
         package_init = self.package_dir / "__init__.py"
 

openapi_python_client/cli.py

@@ -1,7 +1,7 @@
 import codecs
 import pathlib
 from pprint import pformat
-from typing import Optional, Sequence
+from typing import Optional, List, Sequence
 
 import typer
 
@@ -159,7 +159,7 @@ def update(
     meta: MetaType = _meta_option,
     file_encoding: str = typer.Option("utf-8", help="Encoding used when writing generated"),
     config_path: Optional[pathlib.Path] = CONFIG_OPTION,
-    fail_on_warning: bool = False,
+    fail_on_warning: bool = False
 ) -> None:
     """Update an existing OpenAPI Client library
 
@@ -188,6 +188,6 @@ def update(
         meta=meta,
         custom_template_path=custom_template_path,
         file_encoding=file_encoding,
-        config=config,
+        config=config
     )
     handle_errors(errors, fail_on_warning)

openapi_python_client/config.py

@@ -27,6 +27,7 @@ class Config(BaseModel):
     project_name_override: Optional[str]
     package_name_override: Optional[str]
     package_version_override: Optional[str]
+    protected_dirs: Optional[List[str]]
     post_hooks: List[str] = [
         "autoflake -i -r --remove-all-unused-imports --remove-unused-variables --ignore-init-module-imports .",
         "isort .",