mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 03:22:14 +00:00
35 lines
988 B
Python
35 lines
988 B
Python
import os
|
|
import sys
|
|
import time
|
|
|
|
CURDIR = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
sys.path.insert(0, os.path.join(CURDIR))
|
|
|
|
import uexpect
|
|
|
|
class shell(object):
|
|
def __init__(self, command=None, name='', log=None, prompt='[#\$] '):
|
|
if command is None:
|
|
command = ['/bin/bash', '--noediting']
|
|
self.prompt = prompt
|
|
self.client = uexpect.spawn(command)
|
|
self.client.eol('\r')
|
|
self.client.logger(log, prefix=name)
|
|
self.client.timeout(20)
|
|
self.client.expect(prompt, timeout=60)
|
|
|
|
def __enter__(self):
|
|
io = self.client.__enter__()
|
|
io.prompt = self.prompt
|
|
return io
|
|
|
|
def __exit__(self, type, value, traceback):
|
|
self.client.reader['kill_event'].set()
|
|
# send Ctrl-C
|
|
self.client.send('\x03', eol='')
|
|
time.sleep(0.3)
|
|
self.client.send('exit', eol='\r')
|
|
self.client.send('\x03', eol='')
|
|
return self.client.__exit__(type, value, traceback)
|