Turned credentials into namespaced methods.

tcllib/__init__.py

@@ -9,7 +9,7 @@ import enum
 
 import requests
 
-from . import (credentials, devlist, dumpmgr, servervote, tclcheck,
+from . import (dumpmgr, servervote, tclcheck,
                tclchecksum, tclencheader, tclrequest, xmltools)
 
 
@@ -24,7 +24,6 @@ class FotaCheck(
         tclchecksum.TclChecksumMixin,
         tclencheader.TclEncHeaderMixin,
         servervote.ServerVoteMixin,
-        credentials.CredentialsMixin,
         dumpmgr.DumpMgrMixin
 ):
     """Main API handler class."""

tcllib/credentials.py

@@ -8,24 +8,20 @@
 import base64
 
 
-class CredentialsMixin:
+def get_creds():
-    """A mixin component to provide authentication."""
+    """Return main authentication."""
-    @staticmethod
+    creds = {
-    def get_creds():
+        b"YWNjb3VudA==": b"emhlbmdodWEuZ2Fv",
-        """Return main authentication."""
+        b"cGFzc3dvcmQ=": b"cWFydUQ0b2s=",
-        creds = {
+    }
-            b"YWNjb3VudA==": b"emhlbmdodWEuZ2Fv",
+    params = {base64.b64decode(key): base64.b64decode(val) for key, val in creds.items()}
-            b"cGFzc3dvcmQ=": b"cWFydUQ0b2s=",
+    return params
-        }
-        params = {base64.b64decode(key): base64.b64decode(val) for key, val in creds.items()}
-        return params
 
-    @staticmethod
+def get_creds2():
-    def get_creds2():
+    """Return alternate authentication."""
-        """Return alternate authentication."""
+    creds = {
-        creds = {
+        b"YWNjb3VudA==": b"VGVsZUV4dFRlc3Q=",
-            b"YWNjb3VudA==": b"VGVsZUV4dFRlc3Q=",
+        b"cGFzc3dvcmQ=": b"dDA1MjM=",
-            b"cGFzc3dvcmQ=": b"dDA1MjM=",
+    }
-        }
+    params = {base64.b64decode(key): base64.b64decode(val) for key, val in creds.items()}
-        params = {base64.b64decode(key): base64.b64decode(val) for key, val in creds.items()}
+    return params
-        return params

tcllib/tclchecksum.py

@@ -8,14 +8,14 @@
 import json
 
 from defusedxml import ElementTree
-
+from . import credentials
 
 class TclChecksumMixin:
     """A mixin component for TCL's checksum API."""
     def do_checksum(self, encslave, address, uri):
         """Perform checksum request with given parameters."""
         url = "http://" + encslave + "/checksum.php"
-        params = self.get_creds2()
+        params = credentials.get_creds2()
 
         payload = {address: uri}
         payload_json = json.dumps(payload)

tcllib/tclencheader.py

@@ -5,12 +5,14 @@
 
 """Tools to interface with TCL's encrypted header API."""
 
+from . import credentials
+
 
 class TclEncHeaderMixin:
     """A mixin component for TCL's encrypted header API.."""
     def do_encrypt_header(self, encslave, address):
         """Perform encrypted header request with given parameters."""
-        params = self.get_creds2()
+        params = credentials.get_creds2()
         params[b"address"] = bytes(address, "utf-8")
         url = "http://" + encslave + "/encrypt_header.php"
         req = self.sess.post(url, data=params, verify=False)