mirror of
https://github.com/mbirth/tcl_ota_check.git
synced 2024-09-20 06:43:26 +01:00
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from .. import devices
|
|
from . import tclresult
|
|
from collections import OrderedDict
|
|
from defusedxml import ElementTree
|
|
|
|
class TclRequest:
|
|
def __init__(self):
|
|
self.uri = ""
|
|
self.response = None
|
|
self.result = None
|
|
self.error = None
|
|
self.success = False
|
|
|
|
def get_headers(self):
|
|
return {}
|
|
|
|
def get_params(self):
|
|
return {}
|
|
|
|
def is_done(self, http_status: int, contents: str):
|
|
"""Checks if query is done or needs retry."""
|
|
return False
|
|
|
|
def get_result(self):
|
|
"""Returns Result object."""
|
|
return self.result
|
|
|
|
class CheckRequest(TclRequest):
|
|
def __init__(self, device: devices.Device):
|
|
super().__init__()
|
|
self.uri = "/check.php"
|
|
self.method = "GET"
|
|
self.device = device
|
|
|
|
def get_headers(self):
|
|
return {"User-Agent": self.device.ua}
|
|
|
|
def get_params(self):
|
|
params = OrderedDict()
|
|
params["id"] = self.device.imei
|
|
params["curef"] = self.device.curef
|
|
params["fv"] = self.device.fwver
|
|
params["mode"] = self.device.mode
|
|
params["type"] = self.device.type
|
|
params["cltp"] = self.device.cltp
|
|
params["cktp"] = self.device.cktp
|
|
params["rtd"] = self.device.rtd
|
|
params["chnl"] = self.device.chnl
|
|
#params["osvs"] = self.device.osvs
|
|
#params["ckot"] = self.device.ckot
|
|
return params
|
|
|
|
def is_done(self, http_status: int, contents: str) -> bool:
|
|
ok_states = {
|
|
204: "No update available.",
|
|
404: "No data for requested CUREF/FV combination.",
|
|
}
|
|
if http_status == 200:
|
|
self.response = contents
|
|
self.result = tclresult.CheckResult(contents)
|
|
self.success = True
|
|
return True
|
|
elif http_status in ok_states:
|
|
self.error = ok_states[http_status]
|
|
self.success = False
|
|
return True
|
|
elif http_status not in [500, 502, 503]:
|
|
# Errors OTHER than 500, 502 or 503 are probably
|
|
# errors where we don't need to retry
|
|
self.error ="HTTP {}.".format(http_status)
|
|
self.success = False
|
|
return True
|
|
return False
|
|
|
|
# Check requests have 4 possible outcomes:
|
|
# 1. HTTP 200 with XML data - our desired info
|
|
# 2. HTTP 204 - means: no newer update available
|
|
# 3. HTTP 404 - means: invalid device or firmware version
|
|
# 4. anything else: server problem (esp. 500, 502, 503)
|
|
|
|
|
|
|
|
class DownloadRequest(TclRequest):
|
|
def __init__(self, device: devices.Device):
|
|
super().__init__()
|
|
self.uri = "/download_request.php"
|
|
self.method = "POST"
|
|
self.device = device
|
|
|
|
def get_headers(self):
|
|
return {"User-Agent": self.device.ua}
|