Format code using black

globocom · May 11, 2023 · a4d6f22 · a4d6f22
1 parent 57d2547
commit a4d6f22
Show file tree

Hide file tree

Showing 14 changed files with 2,274 additions and 1,655 deletions.
diff --git a/m3u8/__init__.py b/m3u8/__init__.py
@@ -9,41 +9,87 @@
 from urllib.parse import urljoin, urlsplit
 
 from m3u8.httpclient import DefaultHTTPClient
-from m3u8.model import (M3U8, Segment, SegmentList, PartialSegment,
-                        PartialSegmentList, Key, Playlist, IFramePlaylist,
-                        Media, MediaList, PlaylistList, Start,
-                        RenditionReport, RenditionReportList, ServerControl,
-                        Skip, PartInformation, PreloadHint, DateRange,
-                        DateRangeList, ContentSteering)
+from m3u8.model import (
+    M3U8,
+    Segment,
+    SegmentList,
+    PartialSegment,
+    PartialSegmentList,
+    Key,
+    Playlist,
+    IFramePlaylist,
+    Media,
+    MediaList,
+    PlaylistList,
+    Start,
+    RenditionReport,
+    RenditionReportList,
+    ServerControl,
+    Skip,
+    PartInformation,
+    PreloadHint,
+    DateRange,
+    DateRangeList,
+    ContentSteering,
+)
 from m3u8.parser import parse, ParseError
 
 
-__all__ = ('M3U8', 'Segment', 'SegmentList', 'PartialSegment',
-            'PartialSegmentList', 'Key', 'Playlist', 'IFramePlaylist',
-            'Media', 'MediaList', 'PlaylistList', 'Start', 'RenditionReport',
-            'RenditionReportList', 'ServerControl', 'Skip', 'PartInformation',
-            'PreloadHint', 'DateRange', 'DateRangeList', 'ContentSteering',
-            'loads', 'load', 'parse', 'ParseError')
+__all__ = (
+    "M3U8",
+    "Segment",
+    "SegmentList",
+    "PartialSegment",
+    "PartialSegmentList",
+    "Key",
+    "Playlist",
+    "IFramePlaylist",
+    "Media",
+    "MediaList",
+    "PlaylistList",
+    "Start",
+    "RenditionReport",
+    "RenditionReportList",
+    "ServerControl",
+    "Skip",
+    "PartInformation",
+    "PreloadHint",
+    "DateRange",
+    "DateRangeList",
+    "ContentSteering",
+    "loads",
+    "load",
+    "parse",
+    "ParseError",
+)
+
 
 def loads(content, uri=None, custom_tags_parser=None):
-    '''
+    """
     Given a string with a m3u8 content, returns a M3U8 object.
     Optionally parses a uri to set a correct base_uri on the M3U8 object.
     Raises ValueError if invalid content
-    '''
+    """
 
     if uri is None:
         return M3U8(content, custom_tags_parser=custom_tags_parser)
     else:
-        base_uri = urljoin(uri, '.')
+        base_uri = urljoin(uri, ".")
         return M3U8(content, base_uri=base_uri, custom_tags_parser=custom_tags_parser)
 
 
-def load(uri, timeout=None, headers={}, custom_tags_parser=None, http_client=DefaultHTTPClient(), verify_ssl=True):
-    '''
+def load(
+    uri,
+    timeout=None,
+    headers={},
+    custom_tags_parser=None,
+    http_client=DefaultHTTPClient(),
+    verify_ssl=True,
+):
+    """
     Retrieves the content from a given URI and returns a M3U8 object.
     Raises ValueError if invalid content or IOError if request fails.
-    '''
+    """
     if urlsplit(uri).scheme:
         content, base_uri = http_client.download(uri, timeout, headers, verify_ssl)
         return M3U8(content, base_uri=base_uri, custom_tags_parser=custom_tags_parser)
@@ -52,7 +98,7 @@ def load(uri, timeout=None, headers={}, custom_tags_parser=None, http_client=Def
 
 
 def _load_from_file(uri, custom_tags_parser=None):
-    with open(uri, encoding='utf8') as fileobj:
+    with open(uri, encoding="utf8") as fileobj:
         raw_content = fileobj.read().strip()
     base_uri = os.path.dirname(uri)
     return M3U8(raw_content, base_uri=base_uri, custom_tags_parser=custom_tags_parser)
diff --git a/m3u8/httpclient.py b/m3u8/httpclient.py
@@ -5,7 +5,6 @@
 
 
 class DefaultHTTPClient:
-
     def __init__(self, proxies=None):
         self.proxies = proxies
 
@@ -15,15 +14,14 @@ def download(self, uri, timeout=None, headers={}, verify_ssl=True):
         opener = urllib.request.build_opener(proxy_handler, https_handler)
         opener.addheaders = headers.items()
         resource = opener.open(uri, timeout=timeout)
-        base_uri = urljoin(resource.geturl(), '.')
+        base_uri = urljoin(resource.geturl(), ".")
         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:

diff --git a/m3u8/mixins.py b/m3u8/mixins.py
@@ -3,7 +3,6 @@
 
 
 class BasePathMixin(object):
-
     @property
     def absolute_uri(self):
         if self.uri is None:
@@ -14,8 +13,8 @@ def absolute_uri(self):
             return ret
 
         if not urlsplit(ret).scheme:
-            raise ValueError('There can not be `absolute_uri` with no `base_uri` set')
-        
+            raise ValueError("There can not be `absolute_uri` with no `base_uri` set")
+
         return ret
 
     @property
@@ -38,7 +37,6 @@ def base_path(self, newbase_path):
 
 
 class GroupedBasePathMixin(object):
-
     def _set_base_uri(self, new_base_uri):
         for item in self:
             item.base_uri = new_base_uri