fix: import issue

restadmin/__init__.py

@@ -1,3 +1,3 @@
 from .sites import site
 from .decorators import register
-from .restmodeladmin import RestModelAdmin
+from .restmodeladmin import RestModelAdmin

restadmin/restmodeladmin.py

@@ -9,5 +9,5 @@ class RestModelAdmin(ModelViewSet):
     and using vanilla ModelViewSet. This will allow implementing
     more features in the future
     """
-    
-    permission_classes = [IsAdminUser] # By default allow admin users only
+
+    permission_classes = [IsAdminUser]  # By default allow admin users only

setup.py

@@ -1,4 +1,5 @@
 import requests
+import setuptools
 from setuptools import setup
 
 
@@ -27,7 +28,7 @@
         'Intended Audience :: Developers',  # Define that your audience are developers
         'Topic :: Software Development :: Build Tools'
     ],
-    packages=[],
+    packages=setuptools.find_packages(),
     include_package_data=True,
     install_requires=["Django>=3.2.9", "djangorestframework", "coreapi"],
     package_dir={"": "restadmin"},

tests/tests.py

@@ -1,4 +1,4 @@
-from rest_framework.test import APITestCase, override_settings, APIRequestFactory, URLPatternsTestCase
+from rest_framework.test import APITestCase, override_settings
 from rest_framework.viewsets import ModelViewSet
 from rest_framework.permissions import IsAdminUser
 from rest_framework.settings import api_settings
@@ -13,9 +13,6 @@
 from django.core.exceptions import ImproperlyConfigured
 from django.urls import path, reverse
 from django.contrib.auth.models import User
-from django.template.response import TemplateResponse
-from django.test.client import RequestFactory
-import pdb
 
 
 # Create your tests here.
@@ -203,6 +200,8 @@ def test_restmodeladmin_list_objects(self):
         for i in [10, 20, 30, 40, 50, 60]:
             SecondTestModel.objects.create(age=i, name=str(i))
         url = reverse("restadmin:admin_SecondTestModel-list")
+        # This gets only the ages less than 30
+        # TODO: Refactor this. It's misleading(You think it gets all instances when It get just those less than 30)
         response = self.client.get(url)
         self.assertEqual(response.status_code, 200)
         self.assertEqual(len(response.json()), 2)