From 190c36c456974fd72661be2c4a6a314854e93474 Mon Sep 17 00:00:00 2001 From: Felix Stupp Date: Fri, 3 Apr 2020 09:45:50 +0000 Subject: [PATCH] Connection: Added support for different procotols (http, https) --- tinytinypy/main.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/tinytinypy/main.py b/tinytinypy/main.py index aa6f8c8..c8e85f0 100644 --- a/tinytinypy/main.py +++ b/tinytinypy/main.py @@ -92,11 +92,19 @@ class Headline(JsonClass): class Connection: - def __init__(self, host, endpoint="/api/"): + SUPPORTED_PROTO = { + "http": http.client.HTTPConnection, + "https": http.client.HTTPSConnection, + } + + def __init__(self, proto, host, endpoint="/api/"): + if proto not in SUPPORTED_PROTO: + raise ValueError(f"Protocol '{proto}' not supported") + self._proto = proto self._host = host self._endpoint = endpoint self._sid = None - self.__conn = http.client.HTTPSConnection(host) + self.__conn = SUPPORTED_PROTO[proto](host=host) atexit.register(lambda: self.__conn.close()) def __raiseError(self, op, info):