tcl_ota_check/tcllib/__init__.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# pylint: disable=C0111,C0326,C0103

"""Library for TCL API work and related functions."""

import requests

from . import (dumpmgr, servervote, tclcheck, tclchecksum, tclencheader,
               tclrequest)


class FotaCheck(
        tclcheck.TclCheckMixin,
        tclrequest.TclRequestMixin,
        tclchecksum.TclChecksumMixin,
        tclencheader.TclEncHeaderMixin,
        servervote.ServerVoteMixin,
        dumpmgr.DumpMgrMixin
):
    """Main API handler class."""

    def __init__(self):
        """Handle mixins and populate variables."""
        super().__init__()
        self.reset_session()

    def reset_session(self, device=None):
        """Reset everything to default."""
        self.g2master = self.get_master_server()
        self.sess = requests.Session()
        if device:
            self.sess.headers.update({"User-Agent": device.ua})
        return self.sess