from django_roa.db.managers import ROAManager

from dockerhub.query import HubRemoteQuerySet as RemoteQuerySet


class HubROAManager(ROAManager):
    """
    Hub overrides for Manager which access remote resources.
    """
    def __init__(self, username=None, password=None, *args, **kwargs):
        self.username = username
        self.password = password
        super(HubROAManager, self).__init(*args, **kwargs)

    def get_query_set(self):
        """
        Returns a QuerySet which access remote resources.
        """
        return RemoteQuerySet(self.model, self.username, self.password)

    def get_queryset(self, username, password=None):
        return RemoteQuerySet(self.model, self.username, self.password)