Add support for getting links by index (#75)

pyndustri.pyi

@@ -4,6 +4,9 @@ from typing import Iterator, Optional, Union
 class Link:
     """Represents a link."""
 
+    def __getitem__(self, n) -> Building | None:
+        """Gets the nth link"""
+
 class Env:
     """
     Access to special environmental variables.

pyndustric/compiler.py

@@ -1195,6 +1195,11 @@ def as_value(self, node, output: str = None):
                 self.ins_append(f"read {output} {cell} {val}")
                 return output
 
+            if isinstance(node.value, ast.Name) and node.value.id == "Link":
+                val = self.as_value(node.slice)
+                self.ins_append(f"getlink {output} {val}")
+                return output
+
             # container1[dynamic_res]
             # 'some_object'['some_resource']
             obj = self.as_value(node.value).strip('"')

test_all.py

@@ -1164,3 +1164,11 @@ def test_None():
     set foo @null
     """
     foo = None
+
+
+@masm_test
+def test_Link():
+    """
+    getlink l 2
+    """
+    l = Link[2]