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:34:34 +05:30
parent 34bf93c2d5
commit 3dc940ac53
1 changed files with 25 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
vinetrimmer/services/amazon.py
View File

@ -1234,6 +1234,18 @@ class Amazon(BaseService):
""" """
try: try:
res = self.session.get(self.endpoints["ontv"]) res = self.session.get(self.endpoints["ontv"])
response = res.text
if 'input type="hidden" name="appAction" value="SIGNIN"' in response:
raise self.log.exit(
"Cookies are signed out, cannot get ontv CSRF token. "
f"Expecting profile to have cookies for: {self.endpoints['ontv']}"
)
for match in re.finditer(r"<script type=\"text/template\">(.+)</script>", response):
prop = json.loads(match.group(1))
prop = prop.get("props", {}).get("codeEntry", {}).get("token")
if prop:
return prop
raise self.log.exit(f"Unable to get ontv CSRF token - Navigate to {self.endpoints['ontv']}, login and save cookies from that 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
@ -1247,7 +1259,8 @@ class Amazon(BaseService):
prop = prop.get("props", {}).get("codeEntry", {}).get("token") prop = prop.get("props", {}).get("codeEntry", {}).get("token")
if prop: if prop:
return prop return prop
raise self.log.exit(f"Unable to get ontv CSRF token - Navigate to {self.endpoints['ontv']}, 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['ontvold']}, login and save cookies from that page to default.txt")
def get_code_pair(self, device: dict) -> dict: def get_code_pair(self, device: dict) -> dict:
""" """