#!/usr/bin/python3
# -*- coding: utf-8 -*-
from redis import StrictRedis
from config import settings
import json
import time

def get_redis():
    redis = StrictRedis(host=settings.REDIS_DB_URL['host'], port=settings.REDIS_DB_URL['port'], db=settings.REDIS_DB_URL['db'], password=settings.REDIS_DB_URL['password'])
    return redis

def redis_set(name: str, val: str):
    redis = get_redis()
    redis.set(name, val)

def redis_set_with_time(name: str, val: str, seconds: int):
    redis = get_redis()
    redis_set(name, val)
    redis.expire(name, seconds)

def redis_get(name: str):
    redis = get_redis()
    val = redis.get(name)
    if val is not None:
        val = bytes.decode(val)
    return val

def redis_lock(name: str, secs: int = 60):
    redis = get_redis()
    ret = redis.setnx(name, "1")
    if ret:
        redis.expire(name, secs)
    
    time.sleep(1)
    return ret

def redis_unlock(name: str):
    val = redis_get(name)
    if val is not None and val == "1":
        redis = get_redis()
        redis.delete(name)

def redis_set_json(name: str, val: dict, secs: int = 3600):
    json_str = json.dumps(val, ensure_ascii=False)
    redis_set_with_time(name, json_str, secs)

def redis_get_json(name: str):
    json_str = redis_get(name)
    if json_str is not None:
        return json.loads(json_str)