Fix the previous commit. Set cookie after login

Tkd-Alex · Jan 22, 2021 · c4e9466 · c4e9466
1 parent 87da61b
commit c4e9466
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/TwitchChannelPointsMiner/classes/Twitch.py b/TwitchChannelPointsMiner/classes/Twitch.py
@@ -33,7 +33,7 @@ def __init__(self, username):
         cookies_path = os.path.join(Path().absolute(), "cookies")
         Path(cookies_path).mkdir(parents=True, exist_ok=True)
         self.cookies_file = os.path.join(cookies_path, f"{username}.pkl")
-        self.twitch_login = TwitchLogin(TWITCH_CLIENT_ID, username)
+        self.twitch_login = TwitchLogin(TWITCH_CLIENT_ID, username, USER_AGENT)
         self.running = True
 
     def login(self):
@@ -42,6 +42,7 @@ def login(self):
                 self.twitch_login.save_cookies(self.cookies_file)
         else:
             self.twitch_login.load_cookies(self.cookies_file)
+            self.twitch_login.set_token(self.twitch_login.get_auth_token())
 
     def update_minute_watched_event_request(self, streamer):
         event_properties = {

diff --git a/TwitchChannelPointsMiner/classes/TwitchLogin.py b/TwitchChannelPointsMiner/classes/TwitchLogin.py
@@ -15,12 +15,15 @@
 
 
 class TwitchLogin:
-    def __init__(self, client_id, username):
+    def __init__(self, client_id, username, user_agent):
         self.client_id = client_id
         self.token = None
         self.login_check_result = False
         self.session = requests.session()
-        self.session.headers.update({"Client-ID": self.client_id})
+        self.session.headers.update({
+            "Client-ID": self.client_id,
+            "User-Agent": user_agent
+        })
         self.username = username
         self.user_id = None
         self.email = None