app/redis_worker.py

@@ -0,0 +1,36 @@
+import os
+from datetime import datetime
+
+import redis
+from dotenv import load_dotenv
+
+
+class RedisWorker:
+    def __init__(self):
+        load_dotenv()
+        self.client = redis.Redis(
+            host=os.getenv('REDIS_ADDRESS'),
+            port=int(os.getenv('REDIS_PORT')),
+            username=os.getenv('REDIS_USERNAME'),
+            password=os.getenv('REDIS_PASSWORD'),
+            decode_responses=True
+        )
+        self.files_ttl = int(os.getenv('FILES_TTL'))
+
+    def create_record(self, user_ip: str, file_name: str, file_uuid: str, file_type: str, add_date):
+        self.client.hset(file_uuid, mapping={
+            "file_name": file_name,
+            "user_ip": user_ip,
+            "file_type": file_type,
+            "add_date": add_date
+        })
+        self.client.expire(file_uuid, self.files_ttl)
+
+    def get_record(self, file_uuid:str):
+        return self.client.hgetall(file_uuid)
+
+
+if __name__ == "__main__":
+    worker = RedisWorker()
+    worker.create_record("127.0.0.1", "some.jpg", "QWERTY", "image/jpeg", datetime.now().isoformat())
+    print(worker.get_record("QWERTY"))