Add effect.index, input.index and output.index methods

PedalPi · Feb 23, 2017 · 905f16d · 905f16d
1 parent c988bf9
commit 905f16d
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 4 deletions.
diff --git a/pluginsmanager/model/effect.py b/pluginsmanager/model/effect.py
@@ -124,3 +124,13 @@ def json(self):
     @abstractmethod
     def __dict__(self):
         pass
+
+    @property
+    def index(self):
+        """
+        Returns the first occurrence of the effect in your pedalboard
+        """
+        if self.pedalboard is None:
+            raise IndexError('Effect not contains a pedalboard')
+
+        return self.pedalboard.effects.index(self)
diff --git a/pluginsmanager/model/input.py b/pluginsmanager/model/input.py
@@ -68,7 +68,14 @@ def json(self):
     @property
     def __dict__(self):
         return {
-            'effect': self.effect.pedalboard.effects.index(self.effect),
+            'effect': self.effect.index,
             'symbol': self.symbol,
-            'index': self.effect.inputs.index(self),
+            'index': self.index,
         }
+
+    @property
+    def index(self):
+        """
+        :return Input index in the your effect
+        """
+        return self.effect.inputs.index(self)
diff --git a/pluginsmanager/model/output.py b/pluginsmanager/model/output.py
@@ -124,7 +124,14 @@ def json(self):
     @property
     def __dict__(self):
         return {
-            'effect': self.effect.pedalboard.effects.index(self.effect),
+            'effect': self.effect.index,
             'symbol': self.symbol,
-            'index': self.effect.outputs.index(self),
+            'index': self.index,
         }
+
+    @property
+    def index(self):
+        """
+        :return Output index in the your effect
+        """
+        return self.effect.outputs.index(self)