# -*- coding: utf-8 -*-
from cmsplugin_filer_file import __version__  # NOQA

from django.db import models
from treebeard.mp_tree import MP_NodeManager


class FilerPluginManager(MP_NodeManager):
    def __init__(self, select_related=None):
        self._select_related = select_related
        super(FilerPluginManager, self).__init__()

    def get_queryset(self):
        # TODO: Remove once support for Django 1.4/1.5 are dropped
        try:
            qs = super(FilerPluginManager, self).get_queryset()
        except AttributeError:
            qs = super(FilerPluginManager, self).get_query_set()
        if self._select_related:
            qs = qs.prefetch_related(*self._select_related)
        return qs

    get_query_set = get_queryset