@@ -136,14 +136,14 @@ def _create_package(self) -> None:
136
136
# Package __init__.py
137
137
package_init = self .package_dir / "__init__.py"
138
138
139
- package_init_template = self .env .get_template ("package_init.pyi " )
139
+ package_init_template = self .env .get_template ("package_init.py.jinja " )
140
140
package_init .write_text (package_init_template .render (description = self .package_description ))
141
141
142
142
if self .meta != MetaType .NONE :
143
143
pytyped = self .package_dir / "py.typed"
144
144
pytyped .write_text ("# Marker file for PEP 561" )
145
145
146
- types_template = self .env .get_template ("types.py" )
146
+ types_template = self .env .get_template ("types.py.jinja " )
147
147
types_path = self .package_dir / "types.py"
148
148
types_path .write_text (types_template .render ())
149
149
@@ -157,7 +157,7 @@ def _build_metadata(self) -> None:
157
157
158
158
# README.md
159
159
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 " )
161
161
readme .write_text (
162
162
readme_template .render (
163
163
project_name = self .project_name , description = self .package_description , package_name = self .package_name
@@ -166,11 +166,11 @@ def _build_metadata(self) -> None:
166
166
167
167
# .gitignore
168
168
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 " )
170
170
git_ignore_path .write_text (git_ignore_template .render ())
171
171
172
172
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 "
174
174
pyproject_template = self .env .get_template (template )
175
175
pyproject_path = self .project_dir / "pyproject.toml"
176
176
pyproject_path .write_text (
@@ -183,7 +183,7 @@ def _build_pyproject_toml(self, *, use_poetry: bool) -> None:
183
183
)
184
184
185
185
def _build_setup_py (self ) -> None :
186
- template = self .env .get_template ("setup.py" )
186
+ template = self .env .get_template ("setup.py.jinja " )
187
187
path = self .project_dir / "setup.py"
188
188
path .write_text (
189
189
template .render (
@@ -201,15 +201,15 @@ def _build_models(self) -> None:
201
201
models_init = models_dir / "__init__.py"
202
202
imports = []
203
203
204
- model_template = self .env .get_template ("model.pyi " )
204
+ model_template = self .env .get_template ("model.py.jinja " )
205
205
for model in self .openapi .models .values ():
206
206
module_path = models_dir / f"{ model .reference .module_name } .py"
207
207
module_path .write_text (model_template .render (model = model ))
208
208
imports .append (import_string_from_reference (model .reference ))
209
209
210
210
# 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 " )
213
213
for enum in self .openapi .enums .values ():
214
214
module_path = models_dir / f"{ enum .reference .module_name } .py"
215
215
if enum .value_type is int :
@@ -218,13 +218,13 @@ def _build_models(self) -> None:
218
218
module_path .write_text (str_enum_template .render (enum = enum ))
219
219
imports .append (import_string_from_reference (enum .reference ))
220
220
221
- models_init_template = self .env .get_template ("models_init.pyi " )
221
+ models_init_template = self .env .get_template ("models_init.py.jinja " )
222
222
models_init .write_text (models_init_template .render (imports = imports ))
223
223
224
224
def _build_api (self ) -> None :
225
225
# Generate Client
226
226
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 " )
228
228
client_path .write_text (client_template .render ())
229
229
230
230
# Generate endpoints
@@ -233,7 +233,7 @@ def _build_api(self) -> None:
233
233
api_init = api_dir / "__init__.py"
234
234
api_init .write_text ('""" Contains methods for accessing the API """' )
235
235
236
- endpoint_template = self .env .get_template ("endpoint_module.pyi " )
236
+ endpoint_template = self .env .get_template ("endpoint_module.py.jinja " )
237
237
for tag , collection in self .openapi .endpoint_collections_by_tag .items ():
238
238
tag = utils .snake_case (tag )
239
239
tag_dir = api_dir / tag
0 commit comments