From 97a8370caea2fae4a0a67165036aae2dc2d05900 Mon Sep 17 00:00:00 2001 From: Jacob McDonnell Date: Wed, 3 May 2023 13:13:47 -0400 Subject: Rewrite of the website in Flask --- db.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100755 db.py (limited to 'db.py') diff --git a/db.py b/db.py new file mode 100755 index 0000000..ea42c73 --- /dev/null +++ b/db.py @@ -0,0 +1,22 @@ +import redis + +class DB: + def __init__(self, host='localhost', port=6379): + self.db = redis.Redis(host=host, port=port, decode_responses=True) + + def add_article(r, article): + r.db.hset(f"articles:{article['url']}", mapping=article) + + def get_article(r, name): + return r.db.hgetall(f"articles:{name}") + + def get_all_articles(r): + keys = r.get_all_articles_keys() + return [r.db.hgetall(key) for key in keys] + + def get_all_articles_keys(r): + return [key for key in r.db.scan_iter(f"articles:*")] + + def get_all_keys(r): + return [key for key in r.db.scan_iter("*")] + -- cgit v1.2.3