encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-04 20:40:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added support for custom primary key types of custom user models.

Browse Source 
The following change in migration allows you to use a different field type (other then ```AutoField```) for User pks.
This commit is contained in:
Omer Katz 2014-07-20 14:34:40 +03:00
parent 08772799d4
commit f2097f2dbe
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
rest_framework/authtoken/migrations/0001_initial.py
View File

@ -13,6 +13,15 @@ except ImportError: # django < 1.5
from django.contrib.auth.models import User
else:
User = get_user_model()
def get_user_pk_field_fully_qualified_name():
pk_field = User._meta.pk
module = pk_field.__class__.__module__
if module is None:
return pk_field.__class__.__name__
return module + '.' + pk_field.__class__.__name__
class Migration(SchemaMigration):
@ -48,6 +57,11 @@ class Migration(SchemaMigration):
},
"%s.%s" % (User._meta.app_label, User._meta.module_name): {
'Meta': {'object_name': User._meta.module_name},
User._meta.pk.attname: (
get_user_pk_field_fully_qualified_name(), [],
{'primary_key': 'True',
'db_column': "'%s'" % User._meta.pk.column}
),
},
'authtoken.token': {
'Meta': {'object_name': 'Token'},