René's URL Explorer Experiment

{"@context":"https://schema.org","@type":"DiscussionForumPosting","headline":"Execute statement and execute","articleBody":"```\n\tdef executeStmt(self, query):\n\t\t\"\"\"Executes a query using the StackQL instance and returns the output as a string.  \n\t\t\tThis is intended for operations which do not return a result set, for example a mutation \n\t\t\toperation such as an `INSERT` or a `DELETE` or life cycle method such as an `EXEC` operation\n\t\t\tor a `REGISTRY PULL` operation.\n\n\t\tThis method determines the mode of operation (server_mode or local execution) based \n\t\ton the `server_mode` attribute of the instance. If `server_mode` is True, it runs the query \n\t\tagainst the server. Otherwise, it executes the query using a subprocess.\n\n\t\t:param query: The StackQL query string to be executed.\n\t\t:type query: str, list of dict objects, or Pandas DataFrame\n\n\t\t:return: The output result of the query in string format. If in `server_mode`, it \n\t\t\t\treturns a JSON string representation of the result. \n\t\t:rtype: dict, Pandas DataFrame or str (for `csv` output)\n\n\t\tExample:\n\t\t\t\u003e\u003e\u003e from pystackql import StackQL\n\t\t\t\u003e\u003e\u003e stackql = StackQL()\n\t\t\t\u003e\u003e\u003e stackql_query = \"REGISTRY PULL okta\"\n\t\t\t\u003e\u003e\u003e result = stackql.executeStmt(stackql_query)\n\t\t\t\u003e\u003e\u003e result\n\t\t\"\"\"\n\t\tif self.server_mode:\n\t\t\t# Use server mode\n\t\t\tresult = self._run_server_query(query, True)\n\t\t\tif self.output == 'pandas':\n\t\t\t\treturn pd.DataFrame(result)\n\t\t\telif self.output == 'csv':\n\t\t\t\t# return the string representation of the result\n\t\t\t\treturn result[0]['message']\n\t\t\telse:\n\t\t\t\treturn result\n\t\telse:\n\t\t\tresult_msg = self._run_query(query)\n\t\t\tif self.output == 'pandas':\n\t\t\t\treturn pd.DataFrame({'message': [result_msg]})\n\t\t\telif self.output == 'csv':\n\t\t\t\treturn result_msg\n\t\t\telse:\n\t\t\t\treturn [{'message': result_msg}]\t\t\t\n\t\n\tdef execute(self, query):\n\t\t\"\"\"Executes a query using the StackQL instance and returns the output \n\t\tin the format specified by the `output` attribute.\n\n\t\tDepending on the `server_mode` and `output` attribute of the instance, \n\t\tthis method either runs the query against the StackQL server or executes \n\t\tit locally using a subprocess, returning the data in a dictionary, Pandas \n\t\tDataFrame, or CSV format.\n\n\t\t:param query: The StackQL query string to be executed.\n\t\t:type query: str\n\n\t\t:return: The output result of the query. Depending on the `output` attribute, \n\t\t\t\tthe result can be a dictionary, a Pandas DataFrame, or a raw CSV string.\n\t\t\t\tCSV output is currently not supported in `server_mode`.\n\t\t:rtype: dict, pd.DataFrame, or str\n\n\t\tExample:\n\t\t\t\u003e\u003e\u003e from pystackql import StackQL\n\t\t\t\u003e\u003e\u003e stackql = StackQL()\n\t\t\t\u003e\u003e\u003e stackql_query = \\\"\\\"\\\"SELECT SPLIT_PART(machineType, '/', -1) as machine_type, \n\t\t\t... status, COUNT(*) as num_instances\n\t\t\t... FROM google.compute.instances \n\t\t\t... WHERE project = 'stackql-demo' \n\t\t\t... AND zone = 'australia-southeast1-a'\n\t\t\t... GROUP BY machine_type, status\n\t\t\t... HAVING COUNT(*) \u003e 2\\\"\\\"\\\"\n\t\t\t\u003e\u003e\u003e result = stackql.execute(stackql_query)\n\t\t\"\"\"\n\t\tif self.server_mode:\n\t\t\t# Use server mode\n\t\t\tresult = self._run_server_query(query)\n\t\t\t\n\t\t\tif self.output == 'pandas':\n\t\t\t\tjson_str = json.dumps(result)\n\t\t\t\treturn pd.read_json(StringIO(json_str))\n\t\t\telif self.output == 'csv':\n\t\t\t\traise ValueError(\"CSV output is not supported in server_mode.\")\n\t\t\telse:  # Assume 'dict' output\n\t\t\t\treturn result\n\t\t\t\n\t\telse:\n\t\t\t# Local mode handling (existing logic)\n\t\t\toutput = self._run_query(query)\n\t\t\tif self.output == 'csv':\n\t\t\t\treturn output\n\t\t\telif self.output == 'pandas':\n\t\t\t\ttry:\n\t\t\t\t\treturn pd.read_json(StringIO(output))\n\t\t\t\texcept ValueError:\n\t\t\t\t\treturn pd.DataFrame([{\"error\": \"Invalid JSON output: {}\".format(output.strip())}])\n\t\t\telse:  # Assume 'dict' output\n\t\t\t\ttry:\n\t\t\t\t\treturn json.loads(output)\n\t\t\t\texcept ValueError:\n\t\t\t\t\treturn [{\"error\": \"Invalid JSON output: {}\".format(output.strip())}]\n```","author":{"url":"https://github.com/yunchengyang515","@type":"Person","name":"yunchengyang515"},"datePublished":"2023-12-09T04:15:01.000Z","interactionStatistic":{"@type":"InteractionCounter","interactionType":"https://schema.org/CommentAction","userInteractionCount":2},"url":"https://github.com/17/stackqljs/issues/17"}