Allow setting POST headers

jjlawren · Nov 11, 2020 · 861fc49 · 861fc49
1 parent 12fd4a5
commit 861fc49
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/plexauth.py b/plexauth.py
@@ -13,10 +13,11 @@
 
 class PlexAuth():
 
-    def __init__(self, payload, session=None):
+    def __init__(self, payload, session=None, headers=None):
         '''Create PlexAuth instance.'''
         self.client_identifier = str(uuid.uuid4())
         self._code = None
+        self._headers = headers
         self._identifier = None
         self._payload = payload
         self._payload['X-Plex-Client-Identifier'] = self.client_identifier
@@ -29,7 +30,7 @@ def __init__(self, payload, session=None):
 
     async def initiate_auth(self):
         '''Request codes needed to create an auth URL. Starts external timeout.'''
-        async with self._session.post(CODES_URL, data=self._payload) as resp:
+        async with self._session.post(CODES_URL, data=self._payload, headers=self._headers) as resp:
             response = await resp.json()
             self._code = response['code']
             self._identifier = response['id']

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 with open('README.md') as f:
     long_description = f.read()
 
-VERSION="0.0.5"
+VERSION="0.0.6"
 
 setup(
     name='plexauth',