mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-14 03:25:15 +00:00
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
import argparse
|
||
|
|
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
|
||
|
|
import ssl
|
||
|
|
import csv
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def start_server(server_address, cert_path, data_path, schema):
|
||
|
|
class TSVHTTPHandler(BaseHTTPRequestHandler):
|
||
|
|
def _set_headers(self):
|
||
|
|
self.send_response(200)
|
||
|
|
self.send_header('Content-type', 'text/tsv')
|
||
|
|
self.end_headers()
|
||
|
|
|
||
|
|
def do_GET(self):
|
||
|
|
self._set_headers()
|
||
|
|
with open(data_path, 'r') as fl:
|
||
|
|
reader = csv.reader(fl, delimiter='\t')
|
||
|
|
for row in reader:
|
||
|
|
self.wfile.write('\t'.join(row) + '\n')
|
||
|
|
return
|
||
|
|
|
||
|
|
def do_POST(self):
|
||
|
|
return self.do_GET()
|
||
|
|
|
||
|
|
httpd = HTTPServer(server_address, TSVHTTPHandler)
|
||
|
|
if schema == 'https':
|
||
|
|
httpd.socket = ssl.wrap_socket(httpd.socket, certfile=cert_path, server_side=True)
|
||
|
|
httpd.serve_forever()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
parser = argparse.ArgumentParser(description="Simple HTTP server returns data from file")
|
||
|
|
parser.add_argument("--data-path", required=True)
|
||
|
|
parser.add_argument("--schema", choices=("http", "https"), required=True)
|
||
|
|
parser.add_argument("--host", default="localhost")
|
||
|
|
parser.add_argument("--port", default=5555, type=int)
|
||
|
|
parser.add_argument("--cert-path", default="./fake_cert.pem")
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
start_server((args.host, args.port), args.cert_path, args.data_path, args.schema)
|