#!/usr/bin/env python  
# encoding: utf-8
""" 
@contact: [email protected] 
@software: PyCharm 
@file: sql_redispy.py 
@time: 2018/6/8 11:19 
"""
import redis


class TestRedis(object):
    def __init__(self):
        self.r = redis.StrictRedis(host='192.168.199.151',port=6379, db=1)

    # String Key
 


    def  test_set(self):
        """set -- 設定單個鍵值"""
        rest = self.r.set('user1', 'Amy')
        return rest # 返回True or Flase

    def test_get(self):
        """get -- 獲取單個鍵值"""
        rest = self.r.get('user1')
        return rest

    def test_mset(self):
        """mset -- 設定多個鍵值 ,傳入dict"""
 

        d = {
            'user2': 'James',
            'user3': 'Sakura',
            'user4': 'Gaara'
        }
        rest = self.r.mset(d)
        return rest # 返回True or Flase

    def test_mget(self):
        """mget -- 獲取多個鍵值,傳入list或者tuple"""
        t=('user2', 'user3', 'user4')
        rest = self.r.mget(t)
        return
 
 rest

    def test_del(self):
        """del -- 刪除單個鍵值"""
        rest = self.r.delete(('user1'))
        return rest # 返回1 or 0

    def test_incr(self):
        """incr -- 鍵值增加1"""
        self.r.set('age1',33)
        rest = self.r.incr('age1')
        return rest # 但會增加後鍵值結果

    # List
    def test_push(self):
        """lpush/rpush -- 從左/右插入資料 names1:['Amy', 'Jhon]"""
        # rest = self.r.lpush('names1', 'Amy', 'Jhon')
        t = ('Amy1', 'Jhon1')
        rest = self.r.lpush('names1', *t)
        return rest

    def test_lrange(self):
        """lrange -- 查詢List"""
        rest = self.r.lrange('names1', 0, -1)
        return rest

    def test_pop(self):
        """lpop/rpop -- 刪除最左/右邊的元素，並返回該元素"""
        rest = self.r.lpop('names1')
        return rest

    # Set
    def  test_sadd(self):
        """sadd -- 新增元素 zoo1:{'Dog', 'Cat'}"""
        # return self.r.sadd('zoo1', 'Cat', 'Dog', 'Cat1', 'Dog1', 'Cat2', 'Dog2')    # 返回新增元素數
        t = ('Cat', 'Dog', 'Cat1', 'Dog1', 'Cat2', 'Dog2')
        return self.r.sadd('zoo2', *t)


    def test_smembers(self):
        """smembers -- 查詢元素"""
        return self.r.smembers('zoo2')

    def test_srem(self):
        """ srem 刪除元素"""
        # return self.r.srem('zoo1', 'Dog2', 'Dog1')   # 返回刪除元素數
        list = ['Dog2', 'Dog1']
        return self.r.srem('zoo2', *list)

    def test_sinter(self):
        """ sinter集合交集"""
        return self.r.sinter('zoo1', 'zoo2')    # 返回集合交集


    def test_sunion(self):
        """sunion集合並集"""
        arg = ['zoo1', 'zoo2']
        return self.r.sunion(*arg)    # 返回集合並集

    def test_sdiff(self):
        """ sdiff 返回集合直接與其他集合的差異"""
        return self.r.sdiff('zoo1', 'zoo2') # 返回 zoo1中zoo2沒有的集合

    # Hash 雜湊
    def test_hset(self):
        """hset 設定雜湊"""
        return self.r.hset('news:id_001', 'title', 'The News!') # 返回0/1

    def test_hget(self):
        """hget 獲取雜湊結果,hexists 判斷是否有鍵值"""
        if self.r.hexists('news:id_001', 'title'):
            return self.r.hget('news:id_001', 'title')  # 返回結果

    def test_hmset(self):
        """hmset 設定雜湊"""
        m = {
            'title': 'The News1 !',
            'content': 'The News1 content',
            'aurth': 'Sakura'
        }
        return self.r.hmset('news:id_0002',m)   # 返回True or False

    def test_hmget(self):
        """hmget 獲取雜湊"""
        return self.r.hmget('news:id_0002', 'title', 'content') # 返回List

    def test_hvals(self):
        """hvals獲取雜湊"""
        if self.r.hkeys('news:id_0002'):    # hkeys：如果雜湊存在
            return self.r.hvals('news:id_0002') # 返回List


def main():
    obj = TestRedis()
    # String Key
    # print(obj.test_set())
    # print(obj.test_get())
    # print(obj.test_mset())
    # print(obj.test_mget())
    # print(obj.test_del())
    # print(obj.test_incr())

    # List
    # print(obj.test_push())
    # print(obj.test_lrange())
    # print(obj.test_pop())

    # Set
    # print(obj.test_sadd())
    # print(obj.test_smembers())
    # print(obj.test_srem())
    # print(obj.test_sinter())
    # print(obj.test_sunion())
    # print(obj.test_sdiff())

    # Hash
    # print(obj.test_hset())
    # print(obj.test_hget())
    # print(obj.test_hmset())
    # print(obj.test_hmget())
    # print(obj.test_hvals())


if __name__ == "__main__":
    main()