add custom user field

jet/dashboard/models.py

@@ -3,15 +3,17 @@
 from django.db import models
 from django.utils.encoding import python_2_unicode_compatible
 from django.utils.translation import ugettext_lazy as _
+
 from jet.utils import LazyDateTimeEncoder
+from jet.fields import  UserField
 
 
 @python_2_unicode_compatible
 class UserDashboardModule(models.Model):
     title = models.CharField(verbose_name=_('Title'), max_length=255)
     module = models.CharField(verbose_name=_('module'), max_length=255)
     app_label = models.CharField(verbose_name=_('application name'), max_length=255, null=True, blank=True)
-    user = models.PositiveIntegerField(verbose_name=_('user'))
+    user = UserField(verbose_name=_('user'))
     column = models.PositiveIntegerField(verbose_name=_('column'))
     order = models.IntegerField(verbose_name=_('order'))
     settings = models.TextField(verbose_name=_('settings'), default='', blank=True)

jet/fields.py

@@ -0,0 +1,15 @@
+from importlib import import_module
+
+from django.db.models import Field
+
+from jet import settings
+
+
+def get_user_field():
+    pkg, mod = settings.JET_USER_FIELD.rsplit('.', 1)
+    cls = getattr(import_module(pkg), mod)
+    assert issubclass(cls, Field), 'JET_USER_FIELD is not instance of Field class'
+    return cls
+
+
+UserField = get_user_field()

jet/models.py

@@ -3,12 +3,14 @@
 from django.utils.encoding import python_2_unicode_compatible
 from django.utils.translation import ugettext_lazy as _
 
+from jet.fields import UserField
+
 
 @python_2_unicode_compatible
 class Bookmark(models.Model):
     url = models.URLField(verbose_name=_('URL'))
     title = models.CharField(verbose_name=_('title'), max_length=255)
-    user = models.PositiveIntegerField(verbose_name=_('user'))
+    user = UserField(verbose_name=_('user'))
     date_add = models.DateTimeField(verbose_name=_('date created'), default=timezone.now)
 
     class Meta:
@@ -23,7 +25,7 @@ def __str__(self):
 @python_2_unicode_compatible
 class PinnedApplication(models.Model):
     app_label = models.CharField(verbose_name=_('application name'), max_length=255)
-    user = models.PositiveIntegerField(verbose_name=_('user'))
+    user = UserField(verbose_name=_('user'))
     date_add = models.DateTimeField(verbose_name=_('date created'), default=timezone.now)
 
     class Meta:

jet/settings.py

@@ -11,3 +11,6 @@
 
 # Improved usability
 JET_CHANGE_FORM_SIBLING_LINKS = getattr(settings, 'JET_CHANGE_FORM_SIBLING_LINKS', True)
+
+# Improved customability
+JET_USER_FIELD = getattr(settings, 'JET_USER_FIELD', 'django.db.models.PositiveIntegerField')

jet/tests/test_fields.py

@@ -0,0 +1,13 @@
+from django.db.models import PositiveIntegerField
+from django.test import TestCase
+
+from jet.fields import get_user_field
+
+
+class UserFieldTest(TestCase):
+
+    def test_default_user_field(self):
+        default_cls = PositiveIntegerField
+        cls = get_user_field()
+        self.assertIsNotNone(cls)
+        self.assertEqual(cls, default_cls)