@@ -136,14 +136,14 @@ def _create_package(self) -> None:
136136 # Package __init__.py
137137 package_init = self .package_dir / "__init__.py"
138138
139- package_init_template = self .env .get_template ("package_init.pyi " )
139+ package_init_template = self .env .get_template ("package_init.py.jinja " )
140140 package_init .write_text (package_init_template .render (description = self .package_description ))
141141
142142 if self .meta != MetaType .NONE :
143143 pytyped = self .package_dir / "py.typed"
144144 pytyped .write_text ("# Marker file for PEP 561" )
145145
146- types_template = self .env .get_template ("types.py" )
146+ types_template = self .env .get_template ("types.py.jinja " )
147147 types_path = self .package_dir / "types.py"
148148 types_path .write_text (types_template .render ())
149149
@@ -157,7 +157,7 @@ def _build_metadata(self) -> None:
157157
158158 # README.md
159159 readme = self .project_dir / "README.md"
160- readme_template = self .env .get_template ("README.md" )
160+ readme_template = self .env .get_template ("README.md.jinja " )
161161 readme .write_text (
162162 readme_template .render (
163163 project_name = self .project_name , description = self .package_description , package_name = self .package_name
@@ -166,11 +166,11 @@ def _build_metadata(self) -> None:
166166
167167 # .gitignore
168168 git_ignore_path = self .project_dir / ".gitignore"
169- git_ignore_template = self .env .get_template (".gitignore" )
169+ git_ignore_template = self .env .get_template (".gitignore.jinja " )
170170 git_ignore_path .write_text (git_ignore_template .render ())
171171
172172 def _build_pyproject_toml (self , * , use_poetry : bool ) -> None :
173- template = "pyproject.toml" if use_poetry else "pyproject_no_poetry.toml"
173+ template = "pyproject.toml.jinja " if use_poetry else "pyproject_no_poetry.toml.jinja "
174174 pyproject_template = self .env .get_template (template )
175175 pyproject_path = self .project_dir / "pyproject.toml"
176176 pyproject_path .write_text (
@@ -183,7 +183,7 @@ def _build_pyproject_toml(self, *, use_poetry: bool) -> None:
183183 )
184184
185185 def _build_setup_py (self ) -> None :
186- template = self .env .get_template ("setup.py" )
186+ template = self .env .get_template ("setup.py.jinja " )
187187 path = self .project_dir / "setup.py"
188188 path .write_text (
189189 template .render (
@@ -201,15 +201,15 @@ def _build_models(self) -> None:
201201 models_init = models_dir / "__init__.py"
202202 imports = []
203203
204- model_template = self .env .get_template ("model.pyi " )
204+ model_template = self .env .get_template ("model.py.jinja " )
205205 for model in self .openapi .models .values ():
206206 module_path = models_dir / f"{ model .reference .module_name } .py"
207207 module_path .write_text (model_template .render (model = model ))
208208 imports .append (import_string_from_reference (model .reference ))
209209
210210 # Generate enums
211- str_enum_template = self .env .get_template ("str_enum.pyi " )
212- int_enum_template = self .env .get_template ("int_enum.pyi " )
211+ str_enum_template = self .env .get_template ("str_enum.py.jinja " )
212+ int_enum_template = self .env .get_template ("int_enum.py.jinja " )
213213 for enum in self .openapi .enums .values ():
214214 module_path = models_dir / f"{ enum .reference .module_name } .py"
215215 if enum .value_type is int :
@@ -218,13 +218,13 @@ def _build_models(self) -> None:
218218 module_path .write_text (str_enum_template .render (enum = enum ))
219219 imports .append (import_string_from_reference (enum .reference ))
220220
221- models_init_template = self .env .get_template ("models_init.pyi " )
221+ models_init_template = self .env .get_template ("models_init.py.jinja " )
222222 models_init .write_text (models_init_template .render (imports = imports ))
223223
224224 def _build_api (self ) -> None :
225225 # Generate Client
226226 client_path = self .package_dir / "client.py"
227- client_template = self .env .get_template ("client.pyi " )
227+ client_template = self .env .get_template ("client.py.jinja " )
228228 client_path .write_text (client_template .render ())
229229
230230 # Generate endpoints
@@ -233,7 +233,7 @@ def _build_api(self) -> None:
233233 api_init = api_dir / "__init__.py"
234234 api_init .write_text ('""" Contains methods for accessing the API """' )
235235
236- endpoint_template = self .env .get_template ("endpoint_module.pyi " )
236+ endpoint_template = self .env .get_template ("endpoint_module.py.jinja " )
237237 for tag , collection in self .openapi .endpoint_collections_by_tag .items ():
238238 tag = utils .snake_case (tag )
239239 tag_dir = api_dir / tag
0 commit comments