url的db

# 使用redis实现保存url的db
import redis
import traceback


class UrlDB(object):
    status_failure = 0
    status_success = 1

    def __init__(self, host, port, db=0, username=None, password=None):
        pool = redis.ConnectionPool(host=host, port=port, decode_responses=True, db=db, username=username,
                                    password=password)
        self.r = redis.Redis(connection_pool=pool)

    def has(self, url):
        if self.r.get(url):
            return True
        else:
            return False

    def set_success(self, url):
        try:
            self.r.set(url, self.status_success)
            return True
        except:
            traceback.print_exc()
            return False

    def set_failure(self, url):
        try:
            self.r.set(url, self.status_failure)
            return True
        except:
            traceback.print_exc()
            return False

    def __del__(self):
        self.r.close()


if __name__ == '__main__':
    urldb = UrlDB(host='127.0.0.1', port=6379,username='root',password='xxx')
    result = urldb.set_success('https://1www.baidu.com')