openm3u8 7.0.0__cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

openm3u8/httpclient.py

@@ -0,0 +1,36 @@
+import gzip
+import ssl
+import urllib.request
+from urllib.parse import urljoin
+
+
+class DefaultHTTPClient:
+    def __init__(self, proxies=None):
+        self.proxies = proxies
+
+    def download(self, uri, timeout=None, headers={}, verify_ssl=True):
+        proxy_handler = urllib.request.ProxyHandler(self.proxies)
+        https_handler = HTTPSHandler(verify_ssl=verify_ssl)
+        opener = urllib.request.build_opener(proxy_handler, https_handler)
+        opener.addheaders = headers.items()
+        resource = opener.open(uri, timeout=timeout)
+        base_uri = urljoin(resource.geturl(), ".")
+
+        if resource.info().get("Content-Encoding") == "gzip":
+            content = gzip.decompress(resource.read()).decode(
+                resource.headers.get_content_charset(failobj="utf-8")
+            )
+        else:
+            content = resource.read().decode(
+                resource.headers.get_content_charset(failobj="utf-8")
+            )
+        return content, base_uri
+
+
+class HTTPSHandler:
+    def __new__(self, verify_ssl=True):
+        context = ssl.create_default_context()
+        if not verify_ssl:
+            context.check_hostname = False
+            context.verify_mode = ssl.CERT_NONE
+        return urllib.request.HTTPSHandler(context=context)

openm3u8/mixins.py

@@ -0,0 +1,52 @@
+from os.path import dirname
+from urllib.parse import urljoin, urlsplit
+
+
+class BasePathMixin:
+    @property
+    def absolute_uri(self):
+        if self.uri is None:
+            return None
+
+        ret = urljoin(self.base_uri, self.uri)
+        if self.base_uri:
+            base_uri_parts = urlsplit(self.base_uri)
+            if (not base_uri_parts.scheme) and (not base_uri_parts.netloc):
+                return ret
+
+        if not urlsplit(ret).scheme:
+            raise ValueError("There can not be `absolute_uri` with no `base_uri` set")
+
+        return ret
+
+    @property
+    def base_path(self):
+        if self.uri is None:
+            return None
+        return dirname(self.get_path_from_uri())
+
+    def get_path_from_uri(self):
+        """Some URIs have a slash in the query string."""
+        return self.uri.split("?")[0]
+
+    @base_path.setter
+    def base_path(self, newbase_path):
+        if self.uri is not None:
+            if not self.base_path:
+                self.uri = f"{newbase_path}/{self.uri}"
+            else:
+                self.uri = self.uri.replace(self.base_path, newbase_path)
+
+
+class GroupedBasePathMixin:
+    def _set_base_uri(self, new_base_uri):
+        for item in self:
+            item.base_uri = new_base_uri
+
+    base_uri = property(None, _set_base_uri)
+
+    def _set_base_path(self, newbase_path):
+        for item in self:
+            item.base_path = newbase_path
+
+    base_path = property(None, _set_base_path)