class TestRedis(TestCase):

    def setUp(self):
        self.redis = Redis({'host': '127.0.0.1',
                            'port': 6379,
                            'db': 0,
                            'password': ''
                            })

    def test_save_update_delete(self):
        self.redis.set_data("1", "哈哈")
        self.assertEqual(self.redis.get_data("1"), "哈哈")
        self.assertFalse(self.redis.get_data("1") == "哈哈2")
        self.redis.delete("1")
        self.assertFalse(self.redis.get_data("1"))
Exemple #2
0
class RedisService:
    def __init__(self, config):
        self._redis = Redis(redis_config=config)

    def get_convert_arpabet(self, word):
        if not word:
            return None
        key = get_rapabet_redis_key(word=word)
        return self._redis.get_data(key)

    def save_convert_arpabet(self, model):
        if not model:
            return None

        key = get_rapabet_redis_key(word=model.word)
        return self._redis.set_data(key=key, data=model.get_default_arpabet())

    # 爬回来的音标
    def get_crawl_alphabet(self, word):
        if not word:
            return None
        key = get_crawl_redis_key(word=word)
        return self._redis.get_data(key)

    # 爬回来的音标
    def save_crawl_alphabet(self, model):
        if not model:
            return None
        key = get_crawl_redis_key(word=model.get("word"))
        return self._redis.set_data(
            key=key, data=model.get("american_phonetic_alphabet"))

    def delete_crawl_alphabet(self, word):
        if not word:
            return None
        key = get_crawl_redis_key(word=word)
        return self._redis.delete(key=key)