Merge pull request #18 from lynnporu/dev

Forbid to change global variables

annotatec/declarations.py

@@ -226,28 +226,31 @@ def __getattr__(self, key):
         return self.members[key]
 
 
-class MembersWrapper:
+class DeclarationWrapper:
 
     def __init__(self, subject, wrapper):
         self.subject = subject
         self.wrapper = wrapper
 
-    def __getattr__(self, key):
-        try:
-            return getattr(self.wrapper, key)
-        except KeyError:
-            return getattr(self.subject, key)
-
     def __call__(self, *args, **kwargs):
-        return self.subject(*args, **kwargs)
+        return self.subject.__call__(*args, **kwargs)
 
     def __mul__(self, amount):
         return MembersWrapper(
-            subject=(self.subject * amount),
+            subject=(self.subject.__mul__(amount)),
             wrapper=self.wrapper
         )
 
 
+class MembersWrapper(DeclarationWrapper):
+
+    def __getattr__(self, key):
+        try:
+            return getattr(self.wrapper, key)
+        except KeyError:
+            return getattr(self.subject, key)
+
+
 class StructDeclaration(Declaration, MembersDeclaration):
     type_name: str = "struct"
     plural_units = ["member"]

annotatec/loader.py

@@ -3,6 +3,7 @@
 
 from . import libtypes
 from . import parser
+from . import declarations
 
 
 class Loader:
@@ -48,6 +49,15 @@ def __init__(
     def __getattr__(self, key):
         return self.parser.declarations.compile(key)
 
+    def __setattr__(self, key, value):
+        compiled = self.parser.declarations.compile(key)
+        if isinstance(compiled, declarations.VariableDeclaration):
+            raise ValueError(
+                f"Better implement setter function for `{key}` global "
+                "variable.")
+        else:
+            raise RuntimeError(f"`{key}` is not a variable")
+
     def parse_sources(
         self,
         sources: typing.Union[