mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
import hashlib
|
|
import os
|
|
import threading
|
|
|
|
from dvc.utils.compat import HTTPServer, SimpleHTTPRequestHandler
|
|
|
|
|
|
class ETagHandler(SimpleHTTPRequestHandler):
|
|
def end_headers(self):
|
|
file = self.translate_path(self.path)
|
|
|
|
if not os.path.isdir(file) and os.path.exists(file):
|
|
with open(file, "r") as fd:
|
|
etag = hashlib.md5(fd.read().encode("utf8")).hexdigest()
|
|
self.send_header("ETag", etag)
|
|
|
|
SimpleHTTPRequestHandler.end_headers(self)
|
|
|
|
|
|
class ContentMD5Handler(SimpleHTTPRequestHandler):
|
|
def end_headers(self):
|
|
file = self.translate_path(self.path)
|
|
|
|
if not os.path.isdir(file) and os.path.exists(file):
|
|
with open(file, "r") as fd:
|
|
md5 = hashlib.md5(fd.read().encode("utf8")).hexdigest()
|
|
self.send_header("Content-MD5", md5)
|
|
|
|
SimpleHTTPRequestHandler.end_headers(self)
|
|
|
|
|
|
class StaticFileServer:
|
|
_server_lock = threading.Lock()
|
|
|
|
def __init__(self, handler="etag"):
|
|
self._server_lock.acquire()
|
|
handler_class = ETagHandler if handler == "etag" else ContentMD5Handler
|
|
self.bind_port_by_any_means(handler_class)
|
|
|
|
def bind_port_by_any_means(self, handler_class):
|
|
import time
|
|
|
|
# shutdowning/closing socket/server does not unbind port in time,
|
|
# locking the server also does not bring results, hence
|
|
# this method
|
|
for i in range(10000):
|
|
try:
|
|
self.httpd = HTTPServer(("localhost", 8000), handler_class)
|
|
except Exception:
|
|
time.sleep(0.01)
|
|
continue
|
|
break
|
|
|
|
def __enter__(self):
|
|
self.server_thread = threading.Thread(target=self.httpd.serve_forever)
|
|
self.server_thread.daemon = True
|
|
self.server_thread.start()
|
|
|
|
def __exit__(self, *args):
|
|
self.httpd.socket.close()
|
|
self.httpd.shutdown()
|
|
self.httpd.server_close()
|
|
self._server_lock.release()
|