use CaseInsensitiveDict for headers

HHongSeungWoo · Dec 12, 2021 · 0cefc97 · 0cefc97
1 parent afad9d2
commit 0cefc97
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/ytmusicapi/mixins/uploads.py b/ytmusicapi/mixins/uploads.py
@@ -203,7 +203,7 @@ def upload_song(self, filepath: str) -> Union[str, requests.Response]:
                 + ', '.join(supported_filetypes))
 
         headers = self.headers.copy()
-        upload_url = "https://upload.youtube.com/upload/usermusic/http?authuser=%s" % headers['X-Goog-AuthUser']
+        upload_url = "https://upload.youtube.com/upload/usermusic/http?authuser=%s" % headers['x-goog-authuser']
         filesize = os.path.getsize(filepath)
         body = ("filename=" + ntpath.basename(filepath)).encode('utf-8')
         headers.pop('content-encoding', None)

diff --git a/ytmusicapi/ytmusic.py b/ytmusicapi/ytmusic.py
@@ -1,6 +1,7 @@
 import requests
 import gettext
 import os
+from requests.structures import CaseInsensitiveDict
 from functools import partial
 from contextlib import suppress
 from typing import Dict
@@ -74,15 +75,14 @@ def __init__(self,
         self.proxies = proxies
 
         # prepare headers
-        self.headers = {}
         if auth:
             try:
                 if os.path.isfile(auth):
                     file = auth
                     with open(file) as json_file:
-                        self.headers = json.load(json_file)
+                        self.headers = CaseInsensitiveDict(json.load(json_file))
                 else:
-                    self.headers = json.loads(auth)
+                    self.headers = CaseInsensitiveDict(json.loads(auth))
 
             except Exception as e:
                 print(
@@ -120,7 +120,7 @@ def __init__(self,
         # verify authentication credentials work
         if auth:
             try:
-                cookie = self.headers.get('cookie', self.headers.get('Cookie'))
+                cookie = self.headers.get('cookie')
                 self.sapisid = sapisid_from_cookie(cookie)
             except KeyError:
                 raise Exception("Your cookie is missing the required value __Secure-3PAPISID")