86a0db050d
*Code wurde PEP-8 gerecht formatiert * Kleine Fehler die der PyCharm Inspector beanstandet wurden korrigiert
150 lines
4.2 KiB
Python
150 lines
4.2 KiB
Python
"""
|
|
Created on 08.05.2011
|
|
|
|
@author: christian
|
|
"""
|
|
from django.db import models
|
|
from django.db.models import ForeignKey, Model, SET_NULL # @UnusedImport
|
|
from django.db.models import SET_DEFAULT, ManyToManyField # @UnusedImport
|
|
from django.utils import six
|
|
|
|
from . import forms, widgets
|
|
|
|
|
|
class BooleanField(models.BooleanField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.BooleanField}
|
|
defaults.update(kwargs)
|
|
return super(BooleanField, self).formfield(**defaults)
|
|
|
|
|
|
class CharField(models.CharField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.CharField}
|
|
defaults.update(kwargs)
|
|
return super(CharField, self).formfield(**defaults)
|
|
|
|
|
|
class DateField(models.DateField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {
|
|
'form_class': forms.DateField}
|
|
defaults.update(kwargs)
|
|
return super(DateField, self).formfield(**defaults)
|
|
|
|
|
|
class DateTimeField(models.DateTimeField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.DateTimeField}
|
|
defaults.update(kwargs)
|
|
return super(DateTimeField, self).formfield(**defaults)
|
|
|
|
|
|
class EmailField(models.EmailField):
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.EmailField}
|
|
defaults.update(kwargs)
|
|
return super(EmailField, self).formfield(**defaults)
|
|
|
|
|
|
class FileField(models.FileField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.FileField}
|
|
defaults.update(kwargs)
|
|
return super(FileField, self).formfield(**defaults)
|
|
|
|
|
|
class ForeignKey(models.ForeignKey):
|
|
|
|
def formfield(self, **kwargs):
|
|
db = kwargs.pop('using', None)
|
|
if isinstance(self.rel.to, six.string_types):
|
|
raise ValueError("Cannot create form field for %r yet, because "
|
|
"its related model %r has not been loaded yet" %
|
|
(self.name, self.rel.to))
|
|
queryset = self.rel.to._default_manager.using(db)
|
|
queryset = queryset.complex_filter(self.rel.limit_choices_to)
|
|
defaults = {
|
|
'form_class': forms.ModelChoiceField,
|
|
'queryset': queryset,
|
|
'to_field_name': self.rel.field_name,
|
|
}
|
|
defaults.update(kwargs)
|
|
return super(ForeignKey, self).formfield(**defaults)
|
|
|
|
|
|
class ImageField(models.ImageField):
|
|
pass
|
|
|
|
|
|
class IntegerField(models.IntegerField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.IntegerField}
|
|
defaults.update(kwargs)
|
|
return super(IntegerField, self).formfield(**defaults)
|
|
|
|
|
|
class NullBooleanField(models.NullBooleanField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.BooleanField}
|
|
defaults.update(kwargs)
|
|
return super(NullBooleanField, self).formfield(**defaults)
|
|
|
|
|
|
class PositiveSmallIntegerField(models.PositiveSmallIntegerField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {
|
|
'form_class': forms.IntegerField,
|
|
'min_value': 0,
|
|
'max_value': 32767
|
|
}
|
|
defaults.update(kwargs)
|
|
return super(PositiveSmallIntegerField, self).formfield(**defaults)
|
|
|
|
|
|
class PositiveIntegerField(models.IntegerField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {
|
|
'form_class': forms.IntegerField,
|
|
'min_value': 0
|
|
}
|
|
defaults.update(kwargs)
|
|
return super(PositiveIntegerField, self).formfield(**defaults)
|
|
|
|
|
|
class SlugField(models.SlugField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.SlugField}
|
|
defaults.update(kwargs)
|
|
return super(SlugField, self).formfield(**defaults)
|
|
|
|
|
|
class TextField(models.TextField):
|
|
|
|
def formfield(self, **kwargs):
|
|
|
|
defaults = {
|
|
'form_class': forms.CharField,
|
|
'widget': widgets.Textarea
|
|
}
|
|
defaults.update(kwargs)
|
|
return super(TextField, self).formfield(**defaults)
|
|
|
|
|
|
class URLField(models.URLField):
|
|
|
|
def formfield(self, **kwargs):
|
|
defaults = {'form_class': forms.URLField}
|
|
defaults.update(kwargs)
|
|
return super(URLField, self).formfield(**defaults)
|