Skip to content

Commit 2048697

Browse files
authored
ASGI template MyPy fixes (piccolo-orm#224)
* add __init__.py file to asgi template * fix mypy warning with `create_pydantic_model`
1 parent 621c2aa commit 2048697

File tree

4 files changed

+10
-7
lines changed

4 files changed

+10
-7
lines changed

piccolo/apps/asgi/commands/new.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def new(root: str = ".", name: str = "piccolo_project"):
7575
os.mkdir(sub_dir_path)
7676

7777
for file_name in file_names:
78-
if file_name.startswith("_"):
78+
if file_name.startswith("_") and file_name != "__init__.py.jinja":
7979
continue
8080

8181
extension = file_name.rsplit(".")[0]

piccolo/apps/asgi/commands/templates/app/_blacksheep_app.py.jinja

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ from blacksheep.server.openapi.v3 import OpenAPIHandler
1111
from openapidocs.v3 import Info
1212

1313
from home.endpoints import home
14-
from home.piccolo_app import APP_CONFIG
14+
from home.piccolo_app import APP_CONFIG
1515
from home.tables import Task
1616

1717

@@ -36,11 +36,11 @@ app.serve_files("static", root_path="/static")
3636
app.router.add_get("/", home)
3737

3838

39-
TaskModelIn = create_pydantic_model(table=Task, model_name="TaskModelIn")
40-
TaskModelOut = create_pydantic_model(
39+
TaskModelIn: t.Any = create_pydantic_model(table=Task, model_name="TaskModelIn")
40+
TaskModelOut: t.Any = create_pydantic_model(
4141
table=Task, include_default_columns=True, model_name="TaskModelOut"
4242
)
43-
TaskModelPartial = create_pydantic_model(
43+
TaskModelPartial: t.Any = create_pydantic_model(
4444
table=Task, model_name="TaskModelPartial", all_optional=True
4545
)
4646

piccolo/apps/asgi/commands/templates/app/_fastapi_app.py.jinja

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,11 @@ app = FastAPI(
2929
)
3030

3131

32-
TaskModelIn = create_pydantic_model(table=Task, model_name='TaskModelIn')
33-
TaskModelOut = create_pydantic_model(
32+
TaskModelIn: t.Any = create_pydantic_model(
33+
table=Task,
34+
model_name='TaskModelIn'
35+
)
36+
TaskModelOut: t.Any = create_pydantic_model(
3437
table=Task,
3538
include_default_columns=True,
3639
model_name='TaskModelOut'

piccolo/apps/asgi/commands/templates/app/home/__init__.py.jinja

Whitespace-only changes.

0 commit comments

Comments
 (0)