diff --git a/adminsortable/utils.py b/adminsortable/utils.py index e2948cf..c858f88 100644 --- a/adminsortable/utils.py +++ b/adminsortable/utils.py @@ -1,5 +1,17 @@ +from .models import Sortable + + def get_is_sortable(objects): if objects: - if objects.count() > 1: - return True + if issubclass(type(objects[0]), Sortable): + if objects.count() > 1: + return True + return False + + +def check_model_is_sortable(cls): + if cls: + if issubclass(type(cls), Sortable): + if cls.objects.count() > 1: + return True return False