Skip to content

Commit d01535d

Browse files
committed
refactor: convert _SchemaContext to dataclass
1 parent e693bb0 commit d01535d

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

marshmallow_dataclass/__init__.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -433,17 +433,13 @@ def class_schema(
433433
return _internal_class_schema(clazz, base_schema)
434434

435435

436+
@dataclasses.dataclass
436437
class _SchemaContext:
437438
"""Global context for an invocation of class_schema."""
438439

439-
def __init__(
440-
self,
441-
globalns: Optional[Dict[str, Any]] = None,
442-
localns: Optional[Dict[str, Any]] = None,
443-
):
444-
self.seen_classes: Dict[type, str] = {}
445-
self.globalns = globalns
446-
self.localns = localns
440+
globalns: Optional[Dict[str, Any]] = None
441+
localns: Optional[Dict[str, Any]] = None
442+
seen_classes: Dict[type, str] = dataclasses.field(default_factory=dict)
447443

448444
def __enter__(self) -> "_SchemaContext":
449445
_schema_ctx_stack.push(self)

0 commit comments

Comments
 (0)