#!/usr/bin/env python
# coding=utf-8


try:
    import cPickle as pickle
except ImportError:
    import pickle

from redis import from_url
from hotqueue import HotQueue as OrigHotQueue


class HotQueue(OrigHotQueue):
    def __init__(self, name, redis_url=None, serializer=pickle, **kwargs):
        if redis_url is None:
            super().__init__(name=name, serializer=serializer, **kwargs)

        else:
            self.name = name
            self.serializer = serializer
            self.__redis = from_url(url=redis_url)