# -*- coding: utf-8 -*-
import redis
from constant import redis_ip, redis_db, redis_pw, redis_zset_name
pool = redis.ConnectionPool(host=redis_ip, db=redis_db, password=redis_pw)
# pool = redis.ConnectionPool(db=6, password=redis_pw)
rr = redis.Redis(connection_pool=pool) # # 客戶端StrictRedis=Redis 合並了
pipe = rr.pipeline(transaction=True) # 批量操作, 遇到異步容易出錯, 可以不使用, 直接用 rr 連接對象,一次一個添加
def distinct_md5(date, eml_md5, eml_path, redis_fail_path):
"""
用md5標記去重, Redis有序集合去重, Redis版本不一樣, 用法有區別
:param date: 排序分數: int
:param eml_md5: 文件或者其他md5
:return: 1 添加成功, 沒有去重 0 失敗, 去重
"""
# pipe.zadd(redis_zset_name, eml_md5, date)
# pipe.zadd(redis_zset_name, {date: eml_md5}, nx=True) # nx 只插入,存在的值不會修改其分數
# pipe.zadd(redis_zset_name, {date: eml_md5})
pipe.zadd(redis_zset_name, {eml_md5: date})
return pipe.execute()[0] # 1 添加, 0 沒添加
# pipe.zcard('eml_md5_zset') # 查看總個數
def del_redis_md5(n, m): # 按照分數值進行刪除, n, m 為分數, 前后閉區間
pipe.zremrangebyscore(redis_zset_name, n, m)
pipe.execute()