From af916a0f09ec09e8fd8b06fa5bb872d66cfc617e Mon Sep 17 00:00:00 2001 From: REDNBLACK Date: Wed, 30 Nov 2016 22:27:58 +0300 Subject: [PATCH] [WIP] #21:2 --- src/service/chance_manager.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 src/service/chance_manager.py diff --git a/src/service/chance_manager.py b/src/service/chance_manager.py new file mode 100644 index 0000000..13d802d --- /dev/null +++ b/src/service/chance_manager.py @@ -0,0 +1,16 @@ +from src.config import config + + +class ChanceManager: + def __init__(self, redis): + self.redis = redis + self.key = "chance:{}" + self.default_chance = config['bot']['default_chance'] + + def get_chance(self, chat_id): + result = self.redis.instance().get(self.key.format(chat_id)) + + return result if result is not None else self.default_chance + + def set_chance(self, chat_id, chance): + self.redis.instance.set(self.key.format(chat_id), chance)