add cli query

Author: billbsing

convex_api/tool/command/query_command.py

@@ -0,0 +1,51 @@
+"""
+
+    Command Query ..
+
+"""
+import json
+
+from .command_base import CommandBase
+
+
+class QueryCommand(CommandBase):
+
+    def __init__(self, sub_parser=None):
+        self._command_list = []
+        super().__init__('query', sub_parser)
+
+    def create_parser(self, sub_parser):
+
+        parser = sub_parser.add_parser(
+            self._name,
+            description='call a convex query',
+            help='Call a convex query command'
+
+        )
+
+        parser.add_argument(
+            'query',
+            help='query to perform'
+        )
+
+        parser.add_argument(
+            'name_address',
+            nargs='?',
+            help='account address or account name. Defaults: Address #1'
+        )
+
+        return parser
+
+    def execute(self, args, output):
+        convex = self.load_convex(args.url)
+
+        if args.name_address:
+            info = self.resolve_to_name_address(args.name_address, output)
+            if not info:
+                return
+            address = info['address']
+
+        address = 1
+        result = convex.query(args.query, address)
+        output.add_line(json.dumps(result))
+        output.set_values(result)

convex_api/tool/convex_tool.py

@@ -11,6 +11,7 @@
 
 
 from convex_api.tool.command.account_command import AccountCommand
+from convex_api.tool.command.query_command import QueryCommand
 from convex_api.tool.command.command_base import DEFAULT_CONVEX_URL
 from convex_api.tool.output import Output
 
@@ -81,6 +82,7 @@ def convex_tool():
 
     command_list = [
         AccountCommand(command_parser),
+        QueryCommand(command_parser)
     ]
 
     args = parser.parse_args()

tests/intergration/tool/test_query_command.py

@@ -0,0 +1,25 @@
+"""
+
+    Test Query
+
+"""
+from unittest.mock import Mock
+
+from convex_api.tool.command.query_command import QueryCommand
+from convex_api.tool.output import Output
+
+
+
+def test_query_command(convex_url):
+    args = Mock()
+
+    args.url = convex_url
+    args.query = '(address *registry*)'
+    args.name_address = None
+
+    command = QueryCommand()
+    output = Output()
+    command.execute(args, output)
+    assert(output.values['value'])
+
+