2020-04-10 15:23:04 +00:00
|
|
|
import os
|
|
|
|
import sys
|
|
|
|
import time
|
|
|
|
|
|
|
|
CURDIR = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
|
|
|
|
sys.path.insert(0, os.path.join(CURDIR))
|
|
|
|
|
2020-10-02 16:54:07 +00:00
|
|
|
from . import uexpect
|
2020-04-10 15:23:04 +00:00
|
|
|
|
2024-06-05 07:50:39 +00:00
|
|
|
prompt = ":\\) "
|
2022-03-22 16:39:58 +00:00
|
|
|
end_of_block = r".*\r\n.*\r\n"
|
2020-04-10 15:23:04 +00:00
|
|
|
|
2020-09-16 04:26:10 +00:00
|
|
|
|
2020-04-10 15:23:04 +00:00
|
|
|
class client(object):
|
2022-03-22 16:39:58 +00:00
|
|
|
def __init__(self, command=None, name="", log=None):
|
|
|
|
self.client = uexpect.spawn(["/bin/bash", "--noediting"])
|
2020-04-10 15:23:04 +00:00
|
|
|
if command is None:
|
2022-03-22 16:39:58 +00:00
|
|
|
command = "/usr/bin/clickhouse-client"
|
2020-04-10 15:23:04 +00:00
|
|
|
self.client.command = command
|
2022-03-22 16:39:58 +00:00
|
|
|
self.client.eol("\r")
|
2020-04-10 15:23:04 +00:00
|
|
|
self.client.logger(log, prefix=name)
|
|
|
|
self.client.timeout(20)
|
2024-06-05 07:50:39 +00:00
|
|
|
self.client.expect("[#\\$] ", timeout=2)
|
2020-04-10 15:23:04 +00:00
|
|
|
self.client.send(command)
|
|
|
|
|
|
|
|
def __enter__(self):
|
|
|
|
return self.client.__enter__()
|
|
|
|
|
|
|
|
def __exit__(self, type, value, traceback):
|
2022-03-22 16:39:58 +00:00
|
|
|
self.client.reader["kill_event"].set()
|
2020-04-10 15:23:04 +00:00
|
|
|
# send Ctrl-C
|
2022-03-22 16:39:58 +00:00
|
|
|
self.client.send("\x03", eol="")
|
2020-04-10 15:23:04 +00:00
|
|
|
time.sleep(0.3)
|
2022-03-22 16:39:58 +00:00
|
|
|
self.client.send("quit", eol="\r")
|
|
|
|
self.client.send("\x03", eol="")
|
2020-04-10 15:23:04 +00:00
|
|
|
return self.client.__exit__(type, value, traceback)
|