format: make passing protocol/protodecl optional

It is usually not used internally to allow "1 function, more roles"
anyway.

Signed-off-by: Jan Pokorný <jpokorny@redhat.com>

2 files changed, 5 insertions, 5 deletions

diff --git a/formats/command.py b/formats/command.py
index 6613dde..f38897a 100644
--- a/formats/command.py
+++ b/formats/command.py
@@ -16,7 +16,7 @@ class command(SimpleFormat):
     MERGED = Protocol('merged')
 
     @SimpleFormat.producing(BYTESTRING)
-    def get_bytestring(self, protocol):
+    def get_bytestring(self, *protodecl):
         """Return command as canonical single string"""
         # try to look (indirectly) if we have a file at hand first
         ret = super(command, self).get_bytestring(self.BYTESTRING)
@@ -27,7 +27,7 @@ class command(SimpleFormat):
         return ' '.join(self.MERGED(protect_safe=True))
 
     @SimpleFormat.producing(SEPARATED, protect=True)
-    def get_separated(self, protocol):
+    def get_separated(self, *protodecl):
         ret = self.MERGED(protect_safe=True)
         newret, acc = [], []
         for i in ret:
@@ -42,7 +42,7 @@ class command(SimpleFormat):
         return newret
 
     @SimpleFormat.producing(MERGED, protect=True)
-    def get_merged(self, protocol):
+    def get_merged(self, *protodecl):
         # try to look (indirectly) if we have "separated" at hand first
         if self.BYTESTRING in self._representations:  # break the possible loop
             from shlex import split
diff --git a/formats/simpleconfig.py b/formats/simpleconfig.py
index 57408a1..04df34b 100644
--- a/formats/simpleconfig.py
+++ b/formats/simpleconfig.py
@@ -45,7 +45,7 @@ class simpleconfig(SimpleFormat):
     BYTESTRING = SimpleFormat.BYTESTRING
 
     @SimpleFormat.producing(BYTESTRING)
-    def get_bytestring(self, protocol):
+    def get_bytestring(self, *protodecl):
         """Externalize 'struct', that is basically, pretty print it
 
         For example above, the result is something like:
@@ -115,6 +115,6 @@ class simpleconfig(SimpleFormat):
         return ret
 
     @SimpleFormat.producing(STRUCT, protect=True)
-    def get_struct(self, protocol):
+    def get_struct(self, *protodecl):
         # TODO parsing struct from string
         raise NotImplementedError