Use new Device classes in ALL the features!

tcllib/__init__.py

@@ -5,19 +5,12 @@
 
 """Library for TCL API work and related functions."""
 
-import enum
-
 import requests
 
 from . import (dumpmgr, servervote, tclcheck, tclchecksum, tclencheader,
                tclrequest)
 
 
-def default_enum(enumname, vardict, qualroot="tcllib.FotaCheck"):
-    """Enum with defaults set."""
-    return enum.IntEnum(enumname, vardict, module=__name__, qualname="{}.{}".format(qualroot, enumname))
-
-
 class FotaCheck(
         tclcheck.TclCheckMixin,
         tclrequest.TclRequestMixin,
@@ -28,35 +21,15 @@ class FotaCheck(
 ):
     """Main API handler class."""
 
-    CKTP = default_enum("CKTP", ["AUTO", "MANUAL"])
-    MODE = default_enum("MODE", {"OTA": 2, "FULL": 4})
-    RTD = default_enum("RTD", ["UNROOTED", "ROOTED"])
-    CHNL = default_enum("CHNL", ["3G", "WIFI"])
-    CLTP = default_enum("CLTP", {"MOBILE": 10, "DESKTOP": 2010})
-    CKOT = default_enum("CKOT", ["ALL", "AOTA_ONLY", "FOTA_ONLY"])
-
     def __init__(self):
         """Handle mixins and populate variables."""
         super().__init__()
-        self.serid = "543212345000000"
-        self.curef = "PRD-63117-011"
-        self.fv = "AAM481"
-        self.osvs = "7.1.1"
-        self.mode = self.MODE.FULL
-        self.ftype = "Firmware"
-        self.cltp = self.CLTP.MOBILE
-        self.cktp = self.CKTP.MANUAL
-        self.ckot = self.CKOT.ALL
-        self.rtd = self.RTD.UNROOTED
-        self.chnl = self.CHNL.WIFI
         self.reset_session()
 
-    def reset_session(self):
+    def reset_session(self, device=None):
         """Reset everything to default."""
         self.g2master = self.get_master_server()
         self.sess = requests.Session()
-        if self.mode == self.MODE.FULL:
-            self.sess.headers.update({"User-Agent": "com.tcl.fota/5.1.0.2.0029.0, Android"})
-        else:
-            self.sess.headers.update({"User-Agent": "tcl"})
+        if device:
+            self.sess.headers.update({"User-Agent": device.ua})
         return self.sess