Expect proper operation_type

golemfactory · Jul 3, 2019 · 8fe47df · 8fe47df
1 parent 9a7caf4
commit 8fe47df
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/golem/rpc/api/ethereum_.py b/golem/rpc/api/ethereum_.py
@@ -75,7 +75,7 @@ def item(o):
     @rpc_utils.expose('pay.operations')
     @staticmethod
     def get_operations(
-            operation_type: typing.Optional[model.WalletOperation.TYPE],
+            operation_type: typing.Optional[str],
             page_num: int = 1,
             per_page: int = 20,
     ):
@@ -88,6 +88,11 @@ def get_operations(
         query = model.WalletOperation.select() \
             .order_by(model.WalletOperation.id.desc())
         if operation_type:
+            try:
+                operation_type = model.WalletOperation.TYPE(operation_type)
+            except ValueError:
+                logger.error('Invalid operation type: %r', operation_type)
+                return []
             query = query.where(
                 model.WalletOperation.operation_type == operation_type,
             )