fHDHR/fHDHR_NextPVR
1
0
Fork 0
mirror of https://github.com/fHDHR/fHDHR_NextPVR.git synced 2025-12-06 06:26:57 -05:00
Code Issues Releases Wiki Activity

Add Detection of Internal API calls

Browse Source
This commit is contained in:
deathbybadaid 2021-01-05 09:47:34 -05:00
parent e7e4ddcade
commit 2cf0f3d891
4 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fHDHR/api/__init__.py
View File

@ -19,6 +19,8 @@ class fHDHR_API_URLs():
self.config = settings
self.web = web
self.headers = {'User-Agent': "fHDHR/%s" % self.config.internal["versions"]["fHDHR"]}
# Replaced later
self.client = Fillin_Client(settings, web)

2
fHDHR/cli/run.py
View File

@ -73,7 +73,7 @@ def run(settings, logger, db, script_dir, fHDHR_web, origin, alternative_epg):
# Perform some actions now that HTTP Server is running
fhdhr.logger.info("Waiting 3 seconds to send startup tasks trigger.")
time.sleep(3)
fhdhr.api.client.get("/api/startup_tasks")
fhdhr.api.client.get("/api/startup_tasks", headers=fhdhr.api.headers)
# wait forever
while True:

13
fHDHR_web/__init__.py
View File

@ -63,6 +63,10 @@ class fHDHR_HTTP_Server():
def before_request(self):
session["is_internal_api"] = self.detect_internal_api(request)
if session["is_internal_api"]:
self.fhdhr.logger.debug("Client is using internal API call.")
session["is_mobile"] = self.detect_mobile(request)
if session["is_mobile"]:
self.fhdhr.logger.debug("Client is a mobile device.")
@ -79,6 +83,15 @@ class fHDHR_HTTP_Server():
self.fhdhr.logger.debug("Client %s requested %s Closing" % (request.method, request.path))
return response
def detect_internal_api(self, request):
user_agent = request.headers.get('User-Agent')
if not user_agent:
return False
elif str(user_agent).lower().startswith("fhdhr"):
return True
else:
return False
def detect_deviceauth(self, request):
return request.args.get('DeviceAuth', default=None, type=str)

4
fHDHR_web/api/startup_tasks.py
View File

@ -25,10 +25,10 @@ class Startup_Tasks():
updatechannels = True
if updatechannels:
self.fhdhr.api.client.get(self.channel_update_url)
self.fhdhr.api.client.get(self.channel_update_url, headers=self.fhdhr.api.headers)
# Hit EPG Update API
for epg_method in self.fhdhr.device.epg.epg_methods:
self.fhdhr.api.client.get("%s?sorurce=%s" % (self.epg_update_url, epg_method))
self.fhdhr.api.client.get("%s?sorurce=%s" % (self.epg_update_url, epg_method), headers=self.fhdhr.api.headers)
return "Success"