add the .md5 file extension before any URL parameters

jasonish · Feb 23, 2021 · d2988a4 · d2988a4
1 parent a97393d
commit d2988a4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/idstools/scripts/rulecat.py b/idstools/scripts/rulecat.py
@@ -256,6 +256,9 @@ def __init__(self, args):
     def check_checksum(self, tmp_filename, url):
         try:
             checksum_url = url + ".md5"
+            url_parts = url.split("?")
+            if len(url_parts) == 2:
+                checksum_url = ".md5?".join(url_parts)
             local_checksum = hashlib.md5(
                 open(tmp_filename, "rb").read()).hexdigest().strip()
             remote_checksum_buf = io.BytesIO()