diff options
| -rw-r--r-- | wqflask/utility/redis_tools.py | 153 |
1 files changed, 80 insertions, 73 deletions
diff --git a/wqflask/utility/redis_tools.py b/wqflask/utility/redis_tools.py index ffd263e8..0754e16f 100644 --- a/wqflask/utility/redis_tools.py +++ b/wqflask/utility/redis_tools.py @@ -1,73 +1,80 @@ -from __future__ import print_function, division, absolute_import - -import simplejson as json - -import redis # used for collections -Redis = redis.StrictRedis() - -import logging - -from flask import (render_template, flash) - -from utility.logger import getLogger -logger = getLogger(__name__) - -def get_user_id(column_name, column_value): - user_list = Redis.hgetall("users") - for key in user_list: - user_ob = json.loads(user_list[key]) - if user_ob[column_name] == column_value: - return key - - return None - -def get_user_by_unique_column(column_name, column_value): - item_details = None - - user_list = Redis.hgetall("users") - if column_name != "user_id": - for key in user_list: - user_ob = json.loads(user_list[key]) - if user_ob[column_name] == column_value: - item_details = user_ob - else: - item_details = json.loads(user_list[column_value]) - - return item_details - -def set_user_attribute(user_id, column_name, column_value): - user_info = json.loads(Redis.hget("users", user_id)) - user_info[column_name] = column_value - - Redis.hset("users", user_id, json.dumps(user_info)) - -def get_user_collections(user_id): - collections = None - collections = Redis.hget("collections", user_id) - - if collections: - return json.loads(collections) - else: - return [] - -def save_user(user, user_id): - Redis.hset("users", user_id, json.dumps(user)) - -def save_collections(user_id, collections_ob): - Redis.hset("collections", user_id, collections_ob) - -def save_verification_code(user_email, code): - Redis.hset("verification_codes", code, user_email) - -def check_verification_code(code): - email_address = None - user_details = None - email_address = Redis.hget("verification_codes", code) - return email_address - - if email_address: - user_details = get_user_by_unique_column('email_address', email_address) - return user_details - else: - return None - #flash("Invalid code: Password reset code does not exist or might have expired!", "error") +from __future__ import print_function, division, absolute_import + +import simplejson as json + +import redis # used for collections +Redis = redis.StrictRedis() + +import logging + +from flask import (render_template, flash) + +from utility.logger import getLogger +logger = getLogger(__name__) + +def is_redis_available(): + try: + Redis.ping() + except: + return False + return True + +def get_user_id(column_name, column_value): + user_list = Redis.hgetall("users") + for key in user_list: + user_ob = json.loads(user_list[key]) + if column_name in user_ob and user_ob[column_name] == column_value: + return key + + return None + +def get_user_by_unique_column(column_name, column_value): + item_details = None + + user_list = Redis.hgetall("users") + if column_name != "user_id": + for key in user_list: + user_ob = json.loads(user_list[key]) + if column_name in user_ob and user_ob[column_name] == column_value: + item_details = user_ob + else: + item_details = json.loads(user_list[column_value]) + + return item_details + +def set_user_attribute(user_id, column_name, column_value): + user_info = json.loads(Redis.hget("users", user_id)) + user_info[column_name] = column_value + + Redis.hset("users", user_id, json.dumps(user_info)) + +def get_user_collections(user_id): + collections = None + collections = Redis.hget("collections", user_id) + + if collections: + return json.loads(collections) + else: + return [] + +def save_user(user, user_id): + Redis.hset("users", user_id, json.dumps(user)) + +def save_collections(user_id, collections_ob): + Redis.hset("collections", user_id, collections_ob) + +def save_verification_code(user_email, code): + Redis.hset("verification_codes", code, user_email) + +def check_verification_code(code): + email_address = None + user_details = None + email_address = Redis.hget("verification_codes", code) + return email_address + + if email_address: + user_details = get_user_by_unique_column('email_address', email_address) + return user_details + else: + return None + flash("Invalid code: Password reset code does not exist or might have expired!", "error") |