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

37
vinetrimmer/services/amazon.py
View File

@ -1234,20 +1234,33 @@ class Amazon(BaseService):
"""
try:
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:
res = self.session.get(self.endpoints["ontvold"])
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")
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['ontvold']}, login and save cookies from that page to default.txt")
def get_code_pair(self, device: dict) -> dict:
"""