pyxatu 0.1__py3-none-any.whl

pyxatu/__init__.py

@@ -0,0 +1,3 @@
+from .core import PyXatu
+
+__all__ = ['PyXatu']

pyxatu/cli.py

@@ -0,0 +1,21 @@
+import argparse
+from .core import PyXatu
+
+def main():
+    parser = argparse.ArgumentParser(description="Run queries against the Xatu API")
+    parser.add_argument('query', type=str, help='The SQL query to execute')
+    parser.add_argument('--config', type=str, help='Path to the configuration file', default=None)
+    parser.add_argument('--columns', type=str, nargs='+', help='List of column names', default=None)
+    
+    args = parser.parse_args()
+    
+    xatu = PyXatu(config_path=args.config)
+    result = xatu.request_query(args.query, columns=args.columns)
+    
+    if result is not None:
+        print(result)
+    else:
+        print("Query failed")
+
+if __name__ == "__main__":
+    main()

pyxatu/config.json

@@ -0,0 +1,5 @@
+{
+    "CLICKHOUSE_USER": "YOUR-CLICKHOUSE-USER",
+    "CLICKHOUSE_PASSWORD": "YOUR-CLICKHOUSE-PASSPHRASE",
+    "CLICKHOUSE_URL": "https://clickhouse.analytics.production.platform.ethpandaops.io"
+}

pyxatu/core.py

@@ -0,0 +1,69 @@
+import os
+import json
+import requests
+from requests.auth import HTTPBasicAuth
+import pandas as pd
+from io import StringIO
+import time
+from pathlib import Path
+import shutil
+
+
+class PyXatu:
+    def __init__(self, config_path=None):
+        self.user_config_path = os.path.join(Path.home(), '.pyxatu_config.json')
+        
+        if config_path is None:
+            config_path = self.user_config_path
+
+        self.config_path = config_path
+        self.clickhouse_user, self.clickhouse_password, self.clickhouse_url = self.read_clickhouse_config()
+
+    def read_clickhouse_config(self):
+        with open(self.config_path, 'r') as file:
+            config = json.load(file)
+        clickhouse_user = config.get("CLICKHOUSE_USER")
+        clickhouse_password = config.get("CLICKHOUSE_PASSWORD")
+        url = config.get("CLICKHOUSE_URL")
+        print("Clickhouse configs set")
+        return clickhouse_user, clickhouse_password, url
+
+    def request_query(self, _query, columns=None, max_retries=5):
+        attempt = 0
+        while attempt < max_retries:
+            try:
+                if attempt > 0:
+                    print(f"Attempt {attempt + 1} of {max_retries}...")
+                response = requests.get(
+                    self.clickhouse_url,
+                    params={'query': _query},
+                    auth=HTTPBasicAuth(self.clickhouse_user, self.clickhouse_password),
+                    timeout=1500
+                )
+
+                if response.status_code == 200:
+                    data = StringIO(response.text)
+                    df = pd.read_csv(data, sep='\t', header=None)
+                    if columns:
+                        df.columns = columns
+                    return df
+                else:
+                    print(response.text)
+                    print(f"Query failed with status code {response.status_code}. Retrying...")
+
+            except requests.RequestException as e:
+                print(f"An error occurred: {e}. Retrying...")
+
+            attempt += 1
+            wait_time = 2 ** attempt
+            print(f"Waiting for {wait_time} seconds before next attempt...")
+            time.sleep(wait_time)
+
+        print("Max retries reached. Failed to retrieve data.")
+        return None
+    
+    def __str__(self):
+        return f"PyXatu(config_path={self.config_path}, user={self.clickhouse_user}, url={self.clickhouse_url})"
+
+    def __repr__(self):
+        return self.__str__()

pyxatu-0.1.dist-info/METADATA

@@ -0,0 +1,27 @@
+Metadata-Version: 2.1
+Name: pyxatu
+Version: 0.1
+Summary: A Python interface for the Xatu API
+Home-page: https://github.com/nerolation/pyxatu
+Author: Toni Wahrstätter
+Author-email: toni@ethereum.org
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+Requires-Dist: requests
+Requires-Dist: pandas
+
+# PyXatu Configuration
+
+On the first run, the default configuration file will be copied to your current working directory as `pyxatu_config.json`. Please modify this file with your actual ClickHouse credentials.
+
+## Steps to Modify Configuration
+
+1. **Run Your Script**:
+   On the first run, the default configuration file will be copied to your current working directory.
+   ```python
+   import pyxatu
+
+   xatu = pyxatu.PyXatu()

pyxatu-0.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+pyxatu/__init__.py,sha256=9YfZorNYLahJCdueB5RKLyF5JqFbWQsskLFZ8mvMJdI,46
+pyxatu/cli.py,sha256=b3w_V537a07Msf1tqp-Yj2RtBCkpY9mIMz0t1t-KHOo,697
+pyxatu/config.json,sha256=ijwidQo0w9TJtawKDtEY1cIWzqui4c7GoTlZPflCD68,196
+pyxatu/core.py,sha256=hRQxL9TxctduIIL7glPk5jR7GC1qVmMAvbmBSRkiuRI,2446
+pyxatu-0.1.dist-info/METADATA,sha256=fQJ-L5ykMTCUGRlPSxRujwgwVYgCSrUVFg78edW0o9s,885
+pyxatu-0.1.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+pyxatu-0.1.dist-info/entry_points.txt,sha256=l8MAh5m6Ym3axckVnkmAOvlsRw0e1yzw03gzyCeGTGo,47
+pyxatu-0.1.dist-info/top_level.txt,sha256=UT4eKtFyG7uKohc_waGd0o66Y9J3DxQrsM0o0nZC9UA,7
+pyxatu-0.1.dist-info/RECORD,,

pyxatu-0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (72.1.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

pyxatu-0.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+xatu-query = pyxatu.cli:main

pyxatu-0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+pyxatu