python_client: add simple Python ClickHouse client

This commit is contained in:
Leopold Schabel 2022-08-12 19:32:37 +02:00
parent 7e8eaac300
commit 1fd274d895
1 changed files with 64 additions and 0 deletions

64
python_client/radiance.py Executable file
View File

@ -0,0 +1,64 @@
#!/usr/bin/env python3
import os
import requests
RADIANCE_HOST = os.getenv("RADIANCE_HOST")
RADIANCE_USER = os.getenv("RADIANCE_USER")
RADIANCE_PASSWORD = os.getenv("RADIANCE_PASSWORD")
LEADER_STATS = """
SELECT
leader,
median(diff),
count()
FROM
(
WITH
leader,
minIf(timestamp, type = 'firstShredReceived') AS tsReceived,
minIf(timestamp, type = 'completed') AS tsCompleted
SELECT
source,
slot,
leader,
toUnixTimestamp64Milli(tsCompleted) - toUnixTimestamp64Milli(tsReceived) AS diff
FROM slot_status
WHERE toDate(timestamp) = today() - 5
GROUP BY
slot,
leader,
source
)
GROUP BY leader
ORDER BY median(diff)
FORMAT JSON
"""
class RadianceClient():
def __init__(self, host, user, password):
self.host = host
self.user = user
self.password = password
self.session = requests.session()
self.session.user_agent = "radiance-client/v0.1"
self.session.auth = (self.user, self.password)
def query(self, query):
r = self.session.post(self.host, data=query)
if r.status_code != 200:
raise Exception("Failed to get leader stats: {}".format(r.text))
return r.json()['data']
def get_leader_stats(self):
return self.query(LEADER_STATS)
if __name__ == '__main__':
import pprint
c = RadianceClient(RADIANCE_HOST, RADIANCE_USER, RADIANCE_PASSWORD)
r = c.get_leader_stats()
pprint.pprint(r)