Skip to content

Commit ea39278

Browse files
committed
fix(type_checking): fix error on sync/client_only
1 parent ff452ff commit ea39278

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/trame_dataclass/v2.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -450,8 +450,8 @@ def __set_name__(self, owner, name):
450450
owner.TYPE_CHECKING = {}
451451

452452
self._name = name
453-
owner.FIELD_NAMES.add(name)
454453
owner.TYPE_CHECKING[name] = self._type_checking
454+
owner.FIELD_NAMES.add(name)
455455

456456
def __get__(self, instance, owner):
457457
if self._name not in instance._server_state:
@@ -481,6 +481,9 @@ def __set_name__(self, owner, name):
481481
if not hasattr(owner, "FIELD_NAMES"):
482482
owner.FIELD_NAMES = set()
483483

484+
if not hasattr(owner, "TYPE_CHECKING"):
485+
owner.TYPE_CHECKING = {}
486+
484487
if not hasattr(owner, "CLIENT_NAMES"):
485488
owner.CLIENT_NAMES = set()
486489

@@ -497,6 +500,7 @@ def __set_name__(self, owner, name):
497500
owner.ENCODERS[name] = self._convert
498501

499502
self._name = name
503+
owner.TYPE_CHECKING[name] = self._type_checking
500504
owner.FIELD_NAMES.add(name)
501505
owner.CLIENT_NAMES.add(name)
502506

@@ -506,13 +510,17 @@ def __set_name__(self, owner, name):
506510
if not hasattr(owner, "FIELD_NAMES"):
507511
owner.FIELD_NAMES = set()
508512

513+
if not hasattr(owner, "TYPE_CHECKING"):
514+
owner.TYPE_CHECKING = {}
515+
509516
if not hasattr(owner, "CLIENT_NAMES"):
510517
owner.CLIENT_NAMES = set()
511518

512519
if not hasattr(owner, "CLIENT_ONLY_NAMES"):
513520
owner.CLIENT_ONLY_NAMES = set()
514521

515522
self._name = name
523+
owner.TYPE_CHECKING[name] = self._type_checking
516524
owner.FIELD_NAMES.add(name)
517525
owner.CLIENT_NAMES.add(name)
518526
owner.CLIENT_ONLY_NAMES.add(name)

0 commit comments

Comments
 (0)