* refactor receiverid into a separate controller base
* allow multiple headers to prepare for checking multiple claims
This commit is contained in:
@ -18,7 +18,11 @@ class Controller(object):
|
||||
if max_age is not None:
|
||||
headers["Cache-Control"] = "max-age: {0}".format(max_age)
|
||||
for key, value in headers.items():
|
||||
self.handler.send_header(key, value)
|
||||
if isinstance(value, list):
|
||||
for v in value:
|
||||
self.handler.send_header(key, v)
|
||||
else:
|
||||
self.handler.send_header(key, value)
|
||||
self.handler.end_headers()
|
||||
if type(content) == str:
|
||||
content = content.encode()
|
||||
|
21
owrx/controllers/receiverid.py
Normal file
21
owrx/controllers/receiverid.py
Normal file
@ -0,0 +1,21 @@
|
||||
from owrx.controllers import Controller
|
||||
from owrx.receiverid import ReceiverId
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
class ReceiverIdController(Controller):
|
||||
def __init__(self, handler, request, options):
|
||||
super().__init__(handler, request, options)
|
||||
self.authHeaders = []
|
||||
|
||||
def send_response(self, content, code=200, content_type="text/html", last_modified: datetime = None, max_age=None, headers=None):
|
||||
if headers is None:
|
||||
headers = {}
|
||||
headers['Authorization'] = self.authHeaders
|
||||
super().send_response(content, code=code, content_type=content_type, last_modified=last_modified, max_age=max_age, headers=headers)
|
||||
pass
|
||||
|
||||
def handle_request(self):
|
||||
headers = self.request.headers.get_all("Authorization", [])
|
||||
self.authHeaders = [ReceiverId.getResponseHeader(h) for h in headers]
|
||||
super().handle_request()
|
@ -1,8 +1,7 @@
|
||||
from . import Controller
|
||||
from .receiverid import ReceiverIdController
|
||||
from owrx.version import openwebrx_version
|
||||
from owrx.sdr import SdrService
|
||||
from owrx.config import Config
|
||||
from owrx.receiverid import ReceiverId, KeyException
|
||||
import json
|
||||
|
||||
import logging
|
||||
@ -10,7 +9,7 @@ import logging
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class StatusController(Controller):
|
||||
class StatusController(ReceiverIdController):
|
||||
def getProfileStats(self, profile):
|
||||
return {
|
||||
"name": profile["name"],
|
||||
@ -29,12 +28,6 @@ class StatusController(Controller):
|
||||
|
||||
def indexAction(self):
|
||||
pm = Config.get()
|
||||
headers = None
|
||||
if "Authorization" in self.request.headers:
|
||||
try:
|
||||
headers = ReceiverId.getResponseHeader(self.request.headers["Authorization"])
|
||||
except KeyException:
|
||||
logger.exception("error processing authorization header")
|
||||
status = {
|
||||
"receiver": {
|
||||
"name": pm["receiver_name"],
|
||||
@ -47,4 +40,4 @@ class StatusController(Controller):
|
||||
"version": openwebrx_version,
|
||||
"sdrs": [self.getReceiverStats(r) for r in SdrService.getSources().values()]
|
||||
}
|
||||
self.send_response(json.dumps(status), content_type="application/json", headers=headers)
|
||||
self.send_response(json.dumps(status), content_type="application/json")
|
||||
|
Reference in New Issue
Block a user