#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2021/8/20 下午5:43
# @Author : 付孟奇

import redis
import json
from util.log import logger
from util.HandleJenkins import config
from config.vendorList import db_40, db_45


class HandleRedis:
    def __init__(self):
        if config['pro'] in db_40:
            self.db = 40
        elif config['pro'] in db_45:
            self.db = 45
        try:
            self.r = redis.Redis(host='r-2ze79mcysjugilpdskpd.redis.rds.aliyuncs.com', port=6379, db=self.db,
                                 password='$ecur1ty', decode_responses=True)
        except:
            logger.info('redis连接失败！')

    # 存储为字符串
    def set_string(self, k, v):
        self.r.set(k, json.dumps(v, ensure_ascii=False), ex=8640000)

    def get_string_value(self, k):
        redis_v = self.r.get(k)
        return redis_v

    def get_value(self, k):
        redis_v = json.loads(self.r.get(k))
        return redis_v


handle_redis = HandleRedis()

if __name__ == '__main__':
    HandleRedis