from contextlib import contextmanager | |
from .ftp_server import SimpleFtpServer | |
from .http_server import SimpleHttpServer | |
@contextmanager | |
def create_file_server(file_server_type): | |
if file_server_type == "FTP": | |
server = SimpleFtpServer() | |
elif file_server_type == "HTTP": | |
server = SimpleHttpServer() | |
else: | |
assert False | |
try: | |
yield server | |
finally: | |
server.stop() |