Skip to content

Commit 573c55d

Browse files
committed
Added py.typed file in generated package
Closes #18
1 parent 770653c commit 573c55d

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

openapi_python_client/__init__.py

+3
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,9 @@ def _create_package(self) -> None:
8585
package_init_template = self.env.get_template("package_init.pyi")
8686
package_init.write_text(package_init_template.render(description=self.package_description))
8787

88+
pytyped = self.package_dir / "py.typed"
89+
pytyped.write_text("# Marker file for PEP 561")
90+
8891
def _build_metadata(self) -> None:
8992
# Create a pyproject.toml file
9093
pyproject_template = self.env.get_template("pyproject.toml")

tests/test___init__.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -184,14 +184,21 @@ def test__create_package(self, mocker):
184184
project.env = mocker.MagicMock()
185185
project.env.get_template.return_value = package_init_template
186186
package_init_path = mocker.MagicMock(autospec=pathlib.Path)
187-
project.package_dir.__truediv__.return_value = package_init_path
187+
pytyped_path = mocker.MagicMock(autospec=pathlib.Path)
188+
paths = {
189+
"__init__.py": package_init_path,
190+
"py.typed": pytyped_path,
191+
}
192+
193+
project.package_dir.__truediv__.side_effect = lambda x: paths[x]
188194

189195
project._create_package()
190196

191197
project.package_dir.mkdir.assert_called_once()
192198
project.env.get_template.assert_called_once_with("package_init.pyi")
193199
package_init_template.render.assert_called_once_with(description=project.package_description)
194200
package_init_path.write_text.assert_called_once_with(package_init_template.render())
201+
pytyped_path.write_text.assert_called_once_with("# Marker file for PEP 561")
195202

196203
def test__build_metadata(self, mocker):
197204
from openapi_python_client import _Project

0 commit comments

Comments
 (0)