tcllib.requests.downloadrequest

Complexity

Total Complexity

11

Size/Duplication

Total Lines	83
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	DownloadRequest.get_params()	0	18	2
A	DownloadRequest.is_done()	0	13	3
A	DownloadRequest.__init__()	0	7	1
A	DownloadRequest.get_headers()	0	2	1

# -*- coding: utf-8 -*-

This module should have a docstring.

import binascii
import hashlib
import random
import time
import zlib
from collections import OrderedDict
from math import floor
from .. import devices
from .tclrequest import TclRequest
from .tclresult import DownloadResult


VDKEY_B64Z = b"eJwdjwEOwDAIAr8kKFr//7HhmqXp8AIIDrYAgg8byiUXrwRJRXja+d6iNxu0AhUooDCN9rd6rDLxmGIakUVWo3IGCTRWqCAt6X4jGEIUAxgN0eYWnp+LkpHQAg/PsO90ELsy0Npm/n2HbtPndFgGEV31R9OmT4O4nrddjc3Qt6nWscx7e+WRHq5UnOudtjw5skuV09pFhvmqnOEIs4ljPeel1wfLYUF4\n"


def get_salt():
    """Generate cryptographic salt."""
    millis = floor(time.time() * 1000)
    tail = "{:06d}".format(random.randint(0, 999999))
    return "{}{}".format(str(millis), tail)

def get_vk2(params_dict, cltp):
    """Generate salted hash of API parameters."""
    params_dict["cltp"] = cltp
    query = ""
    for key, val in params_dict.items():
        if query:
            query += "&"
        query += key + "=" + str(val)
    vdk = zlib.decompress(binascii.a2b_base64(VDKEY_B64Z))
    query += vdk.decode("utf-8")
    engine = hashlib.sha1()
    engine.update(bytes(query, "utf-8"))
    hexhash = engine.hexdigest()
    return hexhash

class DownloadRequest(TclRequest):

This class should have a docstring.

Too many instance attributes (9/7)

The variable __class__ seems to be unused.

    def __init__(self, device: devices.Device, tvver: str, fw_id: str):

The variable devices does not seem to be defined.

        super().__init__()
        self.uri = "/download_request.php"
        self.method = "POST"
        self.device = device
        self.tvver = tvver
        self.fw_id = fw_id

    def get_headers(self):
        return {"User-Agent": self.device.ua}

    def get_params(self):
        params = OrderedDict()
        params["id"] = self.device.imei
        params["salt"] = get_salt()
        params["curef"] = self.device.curef
        params["fv"] = self.device.fwver
        params["tv"] = self.tvver
        params["type"] = self.device.type
        params["fw_id"] = self.fw_id
        params["mode"] = self.device.mode
        params["vk"] = get_vk2(params, self.device.cltp)
        params["cltp"] = self.device.cltp
        params["cktp"] = self.device.cktp
        params["rtd"] = self.device.rtd
        if self.device.mode == self.device.MODE_STATES["FULL"]:
            params["foot"] = 1
        params["chnl"] = self.device.chnl
        return params

    def is_done(self, http_status: int, contents: str) -> bool:
        if http_status == 200:
            self.response = contents
            self.result = DownloadResult(contents)
            self.success = True
            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