Merge pull request #2777 from lbryio/claim_list_multi_type

`claim_list` argument `--claim_type` can now be repeated to pass a list of claim_types

lbry/extras/daemon/daemon.py

@@ -2154,7 +2154,7 @@ class Daemon(metaclass=JSONRPCServerType):
         List my stream and channel claims.
 
         Usage:
-            claim_list [--claim_type=<claim_type>]
+            claim_list [--claim_type=<claim_type>...]
                        [--account_id=<account_id>] [--wallet_id=<wallet_id>]
                        [--page=<page>] [--page_size=<page_size>]
                        [--resolve]

lbry/wallet/database.py

@@ -739,9 +739,11 @@ class Database(SQLiteMixin):
 
     @staticmethod
     def constrain_claims(constraints):
-        claim_type = constraints.pop('claim_type', None)
-        if claim_type is not None:
-            constraints['txo_type'] = TXO_TYPES[claim_type]
+        claim_types = constraints.pop('claim_type', None)
+        if isinstance(claim_types, str) and claim_types:
+            claim_types = [claim_types]
+        if isinstance(claim_types, list) and claim_types:
+            constraints['txo_type__in'] = [TXO_TYPES[ct] for ct in claim_types]
         else:
             constraints['txo_type__in'] = CLAIM_TYPES
 

tests/integration/blockchain/test_claim_commands.py

@@ -384,6 +384,22 @@ class ClaimSearchCommand(ClaimTestCase):
 
 class ClaimCommands(ClaimTestCase):
 
+    async def test_claim_list_type_filtering(self):
+        await self.channel_create()
+        await self.stream_create()
+
+        r = await self.claim_list(claim_type='channel')
+        self.assertEqual(1, len(r))
+        self.assertEqual('channel', r[0]['value_type'])
+
+        r = await self.claim_list(claim_type='stream')
+        self.assertEqual(1, len(r))
+        self.assertEqual('stream', r[0]['value_type'])
+
+        r = await self.claim_list(claim_type=['stream', 'channel'])
+        self.assertEqual(2, len(r))
+        self.assertEqual({'stream', 'channel'}, {c['value_type'] for c in r})
+
     async def test_claim_stream_channel_list_with_resolve(self):
         await self.channel_create()
         await self.stream_create()