Ensure timestamps are digits-only

jsvine · Jan 18, 2023 · 1da86a0 · 1da86a0
1 parent 5f82eba
commit 1da86a0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/test_bad_statuses.py b/tests/test_bad_statuses.py
@@ -10,7 +10,7 @@ class Test(unittest.TestCase):
     def test_503(self):
         asset = waybackpack.Asset(
             "https://www.amazon.com/Art-Gathering-How-Meet-Matters/dp/1594634920",
-            timestamp=20190506092829,
+            timestamp="20190506092829",
         )
 
         content = asset.fetch()

diff --git a/waybackpack/asset.py b/waybackpack/asset.py
@@ -33,6 +33,8 @@
 
 class Asset(object):
     def __init__(self, original_url, timestamp):
+        # Ensure timestamp is only numeric
+        assert re.match(r"^\d*$", timestamp) is not None
         self.timestamp = timestamp
         self.original_url = original_url