From 28ed2365ab13b162928ade4e3c92d5fb00b81ea8 Mon Sep 17 00:00:00 2001 From: thurask Date: Tue, 6 Feb 2018 20:04:21 -0500 Subject: [PATCH] checksum tweak --- tcllib/tclchecksum.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tcllib/tclchecksum.py b/tcllib/tclchecksum.py index 66306f8..84b281f 100644 --- a/tcllib/tclchecksum.py +++ b/tcllib/tclchecksum.py @@ -15,15 +15,19 @@ 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.""" + @staticmethod + def prep_checksum(encslave, address, uri): + """Prepare URL and parameters for checksum request.""" url = "http://" + encslave + "/checksum.php" params = credentials.get_creds2() - payload = {address: uri} payload_json = json.dumps(payload) params[b"address"] = bytes(payload_json, "utf-8") + return url, params + def do_checksum(self, encslave, address, uri): + """Perform checksum request with given parameters.""" + url, params = self.prep_checksum(encslave, address, uri) # print(repr(dict(params))) req = self.sess.post(url, data=params) if req.status_code == 200: