@@ -27,53 +27,54 @@ def __init__(self, path: Optional[Path] = None, expiry: Optional[int] = 14):
27
27
self .connection = sqlite3 .connect (path or get_cache_root () / "cache.sqlite" )
28
28
self .connection .row_factory = sqlite3 .Row
29
29
30
- self .connection .execute ("CREATE TABLE IF NOT EXISTS queries (url , response, query_date);" )
30
+ self .connection .execute ("CREATE TABLE IF NOT EXISTS queries (query , response, query_date);" )
31
31
self .cleanup ()
32
32
33
- def select (self , url : str ) -> dict [str , Any ]:
33
+ def select (self , query : str ) -> dict [str , Any ]:
34
34
"""Retrieve data from the cache database.
35
35
36
36
Args:
37
- url : Url string used as key.
37
+ query : Url string used as key.
38
38
39
39
Returns:
40
40
Empty dict or select results.
41
41
"""
42
42
if self .expiry :
43
43
expiry = datetime .now (tz = timezone .utc ).astimezone ().date () - timedelta (days = self .expiry )
44
44
cursor = self .connection .execute (
45
- "SELECT * FROM queries WHERE url = ? and query_date > ?;" , (url , expiry .isoformat ())
45
+ "SELECT * FROM queries WHERE query = ? and query_date > ?;" ,
46
+ (query , expiry .isoformat ()),
46
47
)
47
48
else :
48
- cursor = self .connection .execute ("SELECT * FROM queries WHERE url = ?;" , (url ,))
49
+ cursor = self .connection .execute ("SELECT * FROM queries WHERE query = ?;" , (query ,))
49
50
if results := cursor .fetchone ():
50
51
return json .loads (results ["response" ])
51
52
return {}
52
53
53
- def insert (self , url : str , response : dict [str , Any ]) -> None :
54
+ def insert (self , query : str , response : dict [str , Any ]) -> None :
54
55
"""Insert data into the cache database.
55
56
56
57
Args:
57
- url : Url string used as key.
58
+ query : Url string used as key.
58
59
response: Response dict from url.
59
60
"""
60
61
self .connection .execute (
61
- "INSERT INTO queries (url , response, query_date) VALUES (?, ?, ?);" ,
62
+ "INSERT INTO queries (query , response, query_date) VALUES (?, ?, ?);" ,
62
63
(
63
- url ,
64
+ query ,
64
65
json .dumps (response ),
65
66
datetime .now (tz = timezone .utc ).astimezone ().date ().isoformat (),
66
67
),
67
68
)
68
69
self .connection .commit ()
69
70
70
- def delete (self , url : str ) -> None :
71
+ def delete (self , query : str ) -> None :
71
72
"""Remove entry from the cache with the provided url.
72
73
73
74
Args:
74
- url : Url string used as key.
75
+ query : Url string used as key.
75
76
"""
76
- self .connection .execute ("DELETE FROM queries WHERE url = ?;" , (url ,))
77
+ self .connection .execute ("DELETE FROM queries WHERE query = ?;" , (query ,))
77
78
self .connection .commit ()
78
79
79
80
def cleanup (self ) -> None :
0 commit comments