ayuspie/VT-PR
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small change

Browse Source
This commit is contained in:
chu23465 2025-04-30 08:44:32 +05:30
parent 0ba20cf636
commit bfcb36734b
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
vinetrimmer/services/amazon.py
View File

@ -1109,7 +1109,7 @@ class Amazon(BaseService):
csrf_token = self.get_csrf_token() csrf_token = self.get_csrf_token()
# Code pair # Code pair
code_pair = self.get_code_pair(device) code_pair, referer = self.get_code_pair(device)
# Device link # Device link
response = self.session.post( response = self.session.post(
@ -1118,7 +1118,7 @@ class Amazon(BaseService):
"Accept": "*/*", "Accept": "*/*",
"Accept-Language": "en-US,en;q=0.9,es-US;q=0.8,es;q=0.7", # needed? "Accept-Language": "en-US,en;q=0.9,es-US;q=0.8,es;q=0.7", # needed?
"Content-Type": "application/x-www-form-urlencoded", "Content-Type": "application/x-www-form-urlencoded",
"Referer": self.endpoints["ontv"] "Referer": referer
}, },
params=urlencode({ params=urlencode({
# any reason it urlencodes here? requests can take a param dict... # any reason it urlencodes here? requests can take a param dict...
@ -1244,22 +1244,22 @@ class Amazon(BaseService):
prop = json.loads(match.group(1)) prop = json.loads(match.group(1))
prop = prop.get("props", {}).get("codeEntry", {}).get("token") prop = prop.get("props", {}).get("codeEntry", {}).get("token")
if prop: if prop:
return prop return prop, self.endpoints["ontv"]
raise self.log.exit(f"Unable to get ontv CSRF token - Navigate to {self.endpoints['mytv']}, login and save cookies from that page to default.txt") raise self.log.exit(f"Unable to get ontv CSRF token - Navigate to {self.endpoints['mytv']}, login and save cookies from code pair page to default.txt")
except: except:
res = self.session.get(self.endpoints["ontvold"]) res = self.session.get(self.endpoints["ontvold"])
response = res.text response = res.text
if 'input type="hidden" name="appAction" value="SIGNIN"' in response: if 'input type="hidden" name="appAction" value="SIGNIN"' in response:
raise self.log.exit( raise self.log.exit(
"Cookies are signed out, cannot get ontv CSRF token. " "Cookies are signed out, cannot get ontv CSRF token. "
f"Expecting profile to have cookies for: {self.endpoints['ontv']}" f"Expecting profile to have cookies for: {self.endpoints['ontvold']}"
) )
for match in re.finditer(r"<script type=\"text/template\">(.+)</script>", response): for match in re.finditer(r"<script type=\"text/template\">(.+)</script>", response):
prop = json.loads(match.group(1)) prop = json.loads(match.group(1))
prop = prop.get("props", {}).get("codeEntry", {}).get("token") prop = prop.get("props", {}).get("codeEntry", {}).get("token")
if prop: if prop:
return prop return prop, self.endpoints["ontvold"]
raise self.log.exit(f"Unable to get ontv CSRF token - Navigate to {self.endpoints['mytv']}, login and save cookies from that page to default.txt") raise self.log.exit(f"Unable to get ontv CSRF token - Navigate to {self.endpoints['mytv']}, login and save cookies from code pair page to default.txt")
def get_code_pair(self, device: dict) -> dict: def get_code_pair(self, device: dict) -> dict: