[compat] Introduce compat_cookiejar_Cookie
authorSergey M․ <dstftw@gmail.com>
Mon, 4 May 2020 22:54:10 +0000 (05:54 +0700)
committerSergey M․ <dstftw@gmail.com>
Mon, 4 May 2020 22:54:10 +0000 (05:54 +0700)
youtube_dl/compat.py

index d1b86bd13df40676f76879e8e9ea314e1fb3c9be..0ee9bc76020377dc811c0d06736eba8082a53dbd 100644 (file)
@@ -57,6 +57,17 @@ try:
 except ImportError:  # Python 2
     import cookielib as compat_cookiejar
 
+if sys.version_info[0] == 2:
+    class compat_cookiejar_Cookie(compat_cookiejar.Cookie):
+        def __init__(self, version, name, value, *args, **kwargs):
+            if isinstance(name, compat_str):
+                name = name.encode()
+            if isinstance(value, compat_str):
+                value = value.encode()
+            compat_cookiejar.Cookie.__init__(self, version, name, value, *args, **kwargs)
+else:
+    compat_cookiejar_Cookie = compat_cookiejar.Cookie
+
 try:
     import http.cookies as compat_cookies
 except ImportError:  # Python 2
@@ -2987,6 +2998,7 @@ __all__ = [
     'compat_basestring',
     'compat_chr',
     'compat_cookiejar',
+    'compat_cookiejar_Cookie',
     'compat_cookies',
     'compat_ctypes_WINFUNCTYPE',
     'compat_etree_Element',