Skip to content

Commit 882abe4

Browse files
committed
More work: instance_loaders
1 parent f57a99e commit 882abe4

File tree

2 files changed

+16
-12
lines changed

2 files changed

+16
-12
lines changed

stubs/django-import-export/import_export/fields.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ class Field:
2727
dehydrate_method: str | None = None,
2828
m2m_add: bool = False,
2929
) -> None: ...
30-
def clean(self, data: Mapping[str, Any], **kwargs: Any): ...
30+
def clean(self, data: Mapping[str, Any], **kwargs: Any) -> Any: ...
3131
def get_value(self, obj: Model) -> Any: ...
3232
def save(self, obj: Model, data: Mapping[str, Any], is_m2m: bool = False, **kwargs: Any) -> None: ...
3333
def export(self, obj: Model) -> str: ...
Lines changed: 15 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,21 @@
1-
from _typeshed import Incomplete
1+
from collections.abc import Mapping
2+
from typing import Any
3+
4+
from django.db.models import Model, QuerySet
5+
from tablib import Dataset
6+
7+
from .fields import Field
8+
from .resources import Resource
29

310
class BaseInstanceLoader:
4-
resource: Incomplete
5-
dataset: Incomplete
6-
def __init__(self, resource, dataset: Incomplete | None = None) -> None: ...
7-
def get_instance(self, row) -> None: ...
11+
resource: Resource
12+
dataset: Dataset | None
13+
def __init__(self, resource: Resource, dataset: Dataset | None = None) -> None: ...
14+
def get_instance(self, row: Mapping[str, Any]) -> Model | None: ...
815

916
class ModelInstanceLoader(BaseInstanceLoader):
10-
def get_queryset(self): ...
11-
def get_instance(self, row): ...
17+
def get_queryset(self) -> QuerySet: ...
1218

1319
class CachedInstanceLoader(ModelInstanceLoader):
14-
pk_field: Incomplete
15-
all_instances: Incomplete
16-
def __init__(self, *args, **kwargs) -> None: ...
17-
def get_instance(self, row): ...
20+
pk_field: Field
21+
all_instances: dict[Any, Model]

0 commit comments

Comments
 (0)