From 2fda2f58b0529d95d716e5543eb29c7ef9c9310d Mon Sep 17 00:00:00 2001 From: deathbybandaid Date: Sun, 6 Dec 2020 08:50:41 -0500 Subject: [PATCH] test --- fHDHR/origin/origin_service.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/fHDHR/origin/origin_service.py b/fHDHR/origin/origin_service.py index ca54ae7..c82af60 100644 --- a/fHDHR/origin/origin_service.py +++ b/fHDHR/origin/origin_service.py @@ -1,3 +1,7 @@ +import json +import re + +import fHDHR.exceptions class OriginService(): @@ -5,6 +9,28 @@ class OriginService(): def __init__(self, fhdhr): self.fhdhr = fhdhr + self.base_url = 'http://www.xumo.tv' + + self.login() + + def login(self): + self.fhdhr.logger.info("Fetching XUMO token") + self.geoID, self.geoLST = self.getID() + if not self.geoID or self.geoLST: + raise fHDHR.exceptions.OriginSetupError("XUMO Setup Failed") + else: + self.fhdhr.logger.info("XUMO Setup Success") + self.status_dict["Login"] = "Success" + self.fhdhr.config.write(self.fhdhr.config.dict["main"]["dictpopname"], 'token', self.token) + return True + + def getID(self): + pagereq = self.fhdhr.web.session.get(self.base_url) + results = json.loads(re.findall('__JOBS_REHYDRATE_STATE__=(.+?);', (pagereq), flags=re.DOTALL)[0]) + print(results["jobs"]["1"]["data"]["geoId"]) + print(results["jobs"]["1"]["data"]["channelListId"]) + return results["jobs"]["1"]["data"]["geoId"], results["jobs"]["1"]["data"]["channelListId"] + def get_status_dict(self): ret_status_dict = { "Login": "Success",