retry

requirements.txt

@@ -5,5 +5,7 @@ numpy
 pycryptodome
 boto3
 redis
+retry
+
 
 #tomli

src/feedback.py

@@ -2,6 +2,7 @@
 import hashlib
 import redis
 import os
+from retry import retry
 
 def hexdigest(text):
 	return hashlib.md5(text.encode('utf8')).hexdigest()
@@ -28,6 +29,7 @@ def __init__(self, user):
 		self.db = redis.Redis.from_url(REDIS_URL)
 		self.user = user
 
+	@retry(tries=5, delay=0.1)
 	def send(self, score, ctx, details=False):
 		p = self.db.pipeline()
 		dist_list = ctx.get('debug',{}).get('model.query.resp',{}).get('dist_list',[])
@@ -73,6 +75,7 @@ def send(self, score, ctx, details=False):
 		p.sadd(key3, fb_hash)
 		p.execute()
 
+	@retry(tries=5, delay=0.1)
 	def get_score(self):
 		key = f'feedback-score:v2:{self.user}'
 		return self.db.scard(key)

src/stats.py

@@ -1,6 +1,7 @@
 import redis
 from time import strftime
 import os
+from retry import retry
 
 class Stats:
 	def __init__(self):
@@ -44,6 +45,7 @@ def __init__(self):
 		self.db = redis.Redis.from_url(REDIS_URL)
 		self.config = {}
 
+	@retry(tries=5, delay=0.1)
 	def incr(self, key, kv_dict):
 		# TODO: non critical code -> safe exceptions
 		key = self.render(key)
@@ -53,6 +55,7 @@ def incr(self, key, kv_dict):
 			self.db.zincrby(key, val, member)
 		p.execute()
 
+	@retry(tries=5, delay=0.1)
 	def get(self, key):
 		# TODO: non critical code -> safe exceptions
 		key = self.render(key)