do not access annotations too early

Author: skogsbaer

python/code/wypp/records.py

@@ -17,13 +17,6 @@ def init(enableTypeChecking=True):
     global _typeCheckingEnabled
     _typeCheckingEnabled = enableTypeChecking
 
-def _collectDataClassAttributes(cls):
-    result = dict()
-    for c in cls.mro():
-        if hasattr(c, '__kind') and c.__kind == 'record' and hasattr(c, '__annotations__'):
-            result = c.__annotations__ | result
-    return result
-
 def _checkRecordAttr(cls: typing.Any,
                      ns: myTypeguard.Namespaces,
                      name: str,
@@ -48,11 +41,8 @@ def _patchDataClass(cls, mutable: bool, ns: myTypeguard.Namespaces):
     fieldNames = [f.name for f in dataclasses.fields(cls)]
     setattr(cls, EQ_ATTRS_ATTR, fieldNames)
 
-    if hasattr(cls, '__annotations__'):
-        # add annotions for type checked constructor.
-        cls.__kind = 'record'
-        cls.__init__.__annotations__ = _collectDataClassAttributes(cls)
-        cls.__init__ = typecheck.wrapTypecheckRecordConstructor(cls, ns)
+    cls.__kind = 'record'
+    cls.__init__ = typecheck.wrapTypecheckRecordConstructor(cls, ns)
 
     if mutable:
         # prevent new fields being added

python/file-test-data/basics/recursive2_ok.err

@@ -0,0 +1 @@
+Kinzig

python/file-test-data/basics/recursive2_ok.out

@@ -0,0 +1,38 @@
+from __future__ import annotations
+from wypp import *
+
+# Ein Flussabschnitt ist entweder
+# - ein Bach mit Namen und Quelle, oder
+# - ein Zusammenfluss eines Haupt- und Nebenflussabschnitts an einem bestimmten Ort.
+
+@record
+class Creek:
+    origin: str
+    name: str
+
+@record
+class Confluence:
+    location: str
+    mainStem: RiverSection
+    tributary: RiverSection
+
+type RiverSection = Union[Creek, Confluence]
+
+kinzig1 = Creek('Loßburg', 'Kinzig')
+gutach = Creek('Schönwald', 'Gutach')
+kinzig2 = Confluence('Hausach', kinzig1, gutach)
+schutter1 = Creek('Schweighausen', 'Schutter')
+heidengraben = Creek('Lahr', 'Heidengraben')
+schutter2 = Confluence('Lahr', schutter1, heidengraben)
+kinzig3 = Confluence('Kehl', kinzig2, schutter2)
+
+# Name eines Flussabschnitts bestimmen
+# Eingabe: den Flussabschnitt (Typ: RiverSection)
+# Ergebnis: der Name (Typ: str)
+def riverName(r: RiverSection) -> str:
+    if isinstance(r, Creek):
+        return r.name
+    elif isinstance(r, Confluence):
+        return riverName(r.mainStem)
+
+print(riverName(kinzig3))

python/file-test-data/basics/recursive2_ok.py

@@ -0,0 +1 @@
+Kinzig

python/file-test-data/basics/recursive_ok.err

@@ -0,0 +1,37 @@
+from wypp import *
+
+# Ein Flussabschnitt ist entweder
+# - ein Bach mit Namen und Quelle, oder
+# - ein Zusammenfluss eines Haupt- und Nebenflussabschnitts an einem bestimmten Ort.
+
+@record
+class Creek:
+    origin: str
+    name: str
+
+@record
+class Confluence:
+    location: str
+    mainStem: RiverSection
+    tributary: RiverSection
+
+type RiverSection = Union[Creek, Confluence]
+
+kinzig1 = Creek('Loßburg', 'Kinzig')
+gutach = Creek('Schönwald', 'Gutach')
+kinzig2 = Confluence('Hausach', kinzig1, gutach)
+schutter1 = Creek('Schweighausen', 'Schutter')
+heidengraben = Creek('Lahr', 'Heidengraben')
+schutter2 = Confluence('Lahr', schutter1, heidengraben)
+kinzig3 = Confluence('Kehl', kinzig2, schutter2)
+
+# Name eines Flussabschnitts bestimmen
+# Eingabe: den Flussabschnitt (Typ: RiverSection)
+# Ergebnis: der Name (Typ: str)
+def riverName(r: RiverSection) -> str:
+    if isinstance(r, Creek):
+        return r.name
+    elif isinstance(r, Confluence):
+        return riverName(r.mainStem)
+
+print(riverName(kinzig3))

python/file-test-data/basics/recursive_ok.out

@@ -0,0 +1,6 @@
+Traceback (most recent call last):
+  File "file-test-data/basics/recursive_old_fail.py", line 13, in <module>
+    class Confluence:
+  File "file-test-data/basics/recursive_old_fail.py", line 15, in Confluence
+    mainStem: RiverSection
+NameError: name 'RiverSection' is not defined

python/file-test-data/basics/recursive_ok.py

@@ -0,0 +1,37 @@
+from wypp import *
+
+# Ein Flussabschnitt ist entweder
+# - ein Bach mit Namen und Quelle, oder
+# - ein Zusammenfluss eines Haupt- und Nebenflussabschnitts an einem bestimmten Ort.
+
+@record
+class Creek:
+    origin: str
+    name: str
+
+@record
+class Confluence:
+    location: str
+    mainStem: RiverSection
+    tributary: RiverSection
+
+type RiverSection = Union[Creek, Confluence]
+
+kinzig1 = Creek('Loßburg', 'Kinzig')
+gutach = Creek('Schönwald', 'Gutach')
+kinzig2 = Confluence('Hausach', kinzig1, gutach)
+schutter1 = Creek('Schweighausen', 'Schutter')
+heidengraben = Creek('Lahr', 'Heidengraben')
+schutter2 = Confluence('Lahr', schutter1, heidengraben)
+kinzig3 = Confluence('Kehl', kinzig2, schutter2)
+
+# Name eines Flussabschnitts bestimmen
+# Eingabe: den Flussabschnitt (Typ: RiverSection)
+# Ergebnis: der Name (Typ: str)
+def riverName(r: RiverSection) -> str:
+    if isinstance(r, Creek):
+        return r.name
+    elif isinstance(r, Confluence):
+        return riverName(r.mainStem)
+
+print(riverName(kinzig3))