added rewrite methods to element to separate from validation logic

[debian/gnuradio] / grc / python / Port.py

diff --git a/grc/python/Port.py b/grc/python/Port.py
index bfbac72374f11eac6a51576ede21654210939be1..dde736dbaae4e720bfa9d7dc9d52cfe446c0e2e2 100644 (file)
--- a/grc/python/Port.py
+++ b/grc/python/Port.py
@@ -57,17 +57,19 @@ class Port(_Port):
                except AssertionError: self.add_error_message('Port is not connected.')
                try: assert self.is_source() or len(self.get_enabled_connections()) <= 1
                except AssertionError: self.add_error_message('Port has too many connections.')
-               ################################################################
-               # message port logic
-               ################################################################
+               #message port logic
                if self.get_type() == 'msg':
                        try: assert not self.get_nports()
                        except AssertionError: self.add_error_message('A port of type "msg" cannot have "nports" set.')
                        try: assert self.get_vlen() == 1
                        except AssertionError: self.add_error_message('A port of type "msg" must have a "vlen" of 1.')
-               ################################################################
-               # virtual sink logic
-               ################################################################
+
+       def rewrite(self):
+               """
+               Handle the port cloning for virtual blocks.
+               """
+               _Port.rewrite(self)
+               #virtual sink logic
                if self.get_parent().get_key() == 'virtual_sink':
                        if self.get_enabled_connections(): #clone type and vlen
                                source = self.get_enabled_connections()[0].get_source()