@dynamic decorator for methods used in pattern constraints and simu…

…lation

tests/test_attributes.py

@@ -11,6 +11,10 @@ class X(BaseClass):
     pi = PositiveIntegerAttribute()
     s = StringAttribute()
 
+    @dynamic
+    def product(f,i,factor=1):
+        return f*i*factor
+
 
 class TestAttributes(unittest.TestCase):
 
@@ -22,3 +26,12 @@ def test_attribute_defaults(self):
         self.assertEqual(x1.i,None)
         self.assertEqual(x1.pi,None)
         self.assertEqual(x1.s,None)
+
+    def test_dynamic_decorator(self):
+        x1 = X(id='idx1',b=True,f=0.5,i=100)
+
+        self.assertTrue(x1.product._isdynamic)
+        self.assertEqual(x1.product(),50)
+        self.assertEqual(x1.product(factor=3),150)
+        self.assertEqual(x1.product(f=0.25,factor=3),75)
+        self.assertEqual(x1.product(f=0.25,i=10,factor=3),7.5)

wc_rules/attributes.py

@@ -1,5 +1,6 @@
 from obj_model import core
 from obj_model import bio
+from functools import wraps
 
 # scalar attributes
 class IdAttribute(core.StringAttribute):
@@ -45,3 +46,15 @@ class ManyToManyAttribute(core.ManyToManyAttribute):pass
 class BioSeqAttribute(bio.BioSeqAttribute):
     def __init__(self):
         super().__init__(default=None)
+
+# wrapper for methods that can be called for setting pattern constraints & during simulation
+def dynamic(fn):
+    fnvars = fn.__code__.co_varnames
+    @wraps(fn)
+    def outer(self_obj,**kwargs):
+        populate_this = [x for x in fnvars if x in self_obj.get_literal_attrs() and x not in kwargs]
+        for var in populate_this:
+            kwargs[var] = getattr(self_obj,var)
+        return fn(**kwargs)
+    outer._isdynamic = True
+    return outer

wc_rules/pattern.py

@@ -92,7 +92,6 @@ def finalize(self):
         # get reference & defined dependencies
         compiled_deps = self.validate_dependencies()
 
-
         # here we build a networkx graph to evaluate symmetry
         g,helper = build_simple_graph(self)