57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
"""
|
|
Custom fields
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
from django.db.models.fields import AutoField
|
|
from django.db.models.fields.related import OneToOneField
|
|
|
|
|
|
class UnsignedBigIntAutoField(AutoField):
|
|
"""
|
|
An unsigned 8-byte integer for auto-incrementing primary keys.
|
|
"""
|
|
def db_type(self, connection):
|
|
if connection.settings_dict['ENGINE'] == 'django.db.backends.mysql':
|
|
return "bigint UNSIGNED AUTO_INCREMENT"
|
|
elif connection.settings_dict['ENGINE'] == 'django.db.backends.sqlite3':
|
|
# Sqlite will only auto-increment the ROWID column. Any INTEGER PRIMARY KEY column
|
|
# is an alias for that (https://www.sqlite.org/autoinc.html). An unsigned integer
|
|
# isn't an alias for ROWID, so we have to give up on the unsigned part.
|
|
return "integer"
|
|
elif connection.settings_dict['ENGINE'] == 'django.db.backends.postgresql_psycopg2':
|
|
# Pg's bigserial is implicitly unsigned (doesn't allow negative numbers) and
|
|
# goes 1-9.2x10^18
|
|
return "BIGSERIAL"
|
|
else:
|
|
return None
|
|
|
|
# rel_db_type was added in Django 1.10. For versions before, use UnsignedBigIntOneToOneField.
|
|
def rel_db_type(self, connection):
|
|
if connection.settings_dict['ENGINE'] == 'django.db.backends.mysql':
|
|
return "bigint UNSIGNED"
|
|
elif connection.settings_dict['ENGINE'] == 'django.db.backends.sqlite3':
|
|
return "integer"
|
|
elif connection.settings_dict['ENGINE'] == 'django.db.backends.postgresql_psycopg2':
|
|
return "BIGSERIAL"
|
|
else:
|
|
return None
|
|
|
|
|
|
class UnsignedBigIntOneToOneField(OneToOneField):
|
|
"""
|
|
An unsigned 8-byte integer one-to-one foreign key to a unsigned 8-byte integer id field.
|
|
|
|
Should only be necessary for versions of Django < 1.10.
|
|
"""
|
|
def db_type(self, connection):
|
|
if connection.settings_dict['ENGINE'] == 'django.db.backends.mysql':
|
|
return "bigint UNSIGNED"
|
|
elif connection.settings_dict['ENGINE'] == 'django.db.backends.sqlite3':
|
|
return "integer"
|
|
elif connection.settings_dict['ENGINE'] == 'django.db.backends.postgresql_psycopg2':
|
|
return "BIGSERIAL"
|
|
else:
|
|
return None
|