Merge pull request #6 from manueljgarciar/master

Run and Register functions doesn't work

Author: Meir Shpilraien (Spielrein)

gearsclient/redisgears_builder.py

@@ -112,12 +112,23 @@ def AddToGB(self, gb, globalsDict):
         gb.run(self.arg, self.convertToStr, self.collect)
 
 class GearsRemoteRegisterStep():
-    def __init__(self, arg):
-        self.arg = arg
+    def __init__(self, regex, mode, batch, duration,
+            eventTypes, keyTypes, onRegistered, onFailedPolicy,
+            onFailedRetryInterval):
+        self.regex = regex
+        self.mode = mode
+        self.batch = batch
+        self.duration = duration
+        self.eventTypes = eventTypes
+        self.keyTypes = keyTypes
+        self.onRegistered = onRegistered
+        self.onFailedPolicy = onFailedPolicy
+        self.onFailedRetryInterval = onFailedRetryInterval
 
     def AddToGB(self, gb, globalsDict):
-        gb.register(self.arg) if self.arg else gb.register()
-
+        gb.register(self.regex, self.mode, self.batch, self.duration,
+                    self.eventTypes, self.keyTypes, self.onRegistered, self.onFailedPolicy,
+                    self.onFailedRetryInterval)
 
 class GearsPipe():
     def __init__(self, reader='KeysReader', defaultArg='*'):
@@ -176,8 +187,12 @@ def limit(self, count, offset):
     def run(self, arg, convertToStr, collect):
         self.steps.append(GearsRemoteRunStep(arg, convertToStr, collect))
 
-    def register(self, arg):
-        self.steps.append(GearsRemoteRegisterStep(arg))
+    def register(self, regex, mode, batch, duration,
+            eventTypes, keyTypes, onRegistered, onFailedPolicy,
+            onFailedRetryInterval):
+        self.steps.append(GearsRemoteRegisterStep(regex, mode, batch, duration,
+            eventTypes, keyTypes, onRegistered, onFailedPolicy,
+            onFailedRetryInterval))
 
 
 class GearsRemoteBuilder():
@@ -240,11 +255,15 @@ def run(self, arg=None, convertToStr=False, collect=True):
         selfBytes = cloudpickle.dumps(self.pipe)
         results = self.r.execute_command('RG.PYEXECUTEREMOTE', selfBytes)
         res, errs = results
-        res = [cloudpickle.loads(record) for record in res]
         return res, errs
 
-    def register(self, arg=None):
-        self.pipe.register(arg)
+    def register(self, regex='*', mode='async', batch=1, duration=0,
+            eventTypes=None, keyTypes=None, onRegistered=None, onFailedPolicy="continue",
+            onFailedRetryInterval=1):
+        self.pipe.register(regex, mode, batch, duration,
+            eventTypes, keyTypes, onRegistered, onFailedPolicy,
+            onFailedRetryInterval)
         selfBytes = cloudpickle.dumps(self.pipe)
         res = self.r.execute_command('RG.PYEXECUTEREMOTE', selfBytes)
         return res
+