Fix error when inserting without an expiry (#37)

Metron-Project · Sep 13, 2021 · 8d92274 · 8d92274
1 parent 9121a50
commit 8d92274
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 4 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -16,6 +16,6 @@ repos:
     hooks:
       - id: flake8
   - repo: https://github.com/asottile/pyupgrade
-    rev: v2.25.0
+    rev: v2.26.0
     hooks:
       - id: pyupgrade
diff --git a/Simyan/sqlite_cache.py b/Simyan/sqlite_cache.py
@@ -32,7 +32,10 @@ def get(self, key: str) -> Optional[Dict[str, Any]]:
         return self.select(query=key) or None
 
     def insert(self, query: str, response: str):
-        expiry = datetime.now() + timedelta(days=self.expiry)
+        if self.expiry:
+            expiry = datetime.now() + timedelta(days=self.expiry)
+        else:
+            expiry = datetime.now()
         self.cur.execute(
             "INSERT INTO queries(query, response, expiry) VALUES(?, ?, ?);",
             (query, json.dumps(response), expiry.strftime("%Y-%m-%d")),

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "Simyan"
-version = "0.5.0"
+version = "0.5.1"
 description = "A Python wrapper for the Comicvine API."
 license = "GPL-3.0-or-later"
 authors = ["Buried-In-Code <6057651+Buried-In-Code@users.noreply.github.com>"]

diff --git a/tests/Simyan-Cache.sqlite b/tests/Simyan-Cache.sqlite