114.django复制模型对象和自定字段

1.模型对象复制

产生原因:项目增加某部分表数据的复制功能,但是很难受由于项目是初始阶段,表中字段一直变化,通过对象复制操作
每次增加一些字段我就需要改代码,特别烦人,所以产生了这个东西.

代码如下很简单:
from django.db.models import AutoField, DateTimeField
from PublicMethod.CustomDjangoModelFiled import AutoIncreField   # 这个字段是我自定义的一个自增类型字段(google可以找到)
def copy_object(obj):
    """
    拷贝模型对象对象,并排除exclude中的字段,返回一个新的对象
    :param obj 表示要copy的对象
    :param exclude 表示哪些字段排除之外
    :return 返回一个新的对象
    """
    # 关于其中排除字段的原因AutoField和AutoIncreField由于是自增字段所以该字段对应的值不应该复制过来,
DateTimeField字段也是auto_now=True所以排除掉
    initial = dict([(f.name, getattr(obj, f.name))for f in obj._meta.fields
                    if not (isinstance(f, AutoField) or isinstance(f, DateTimeField) or isinstance(f, AutoIncreField))
                    and not f in obj._meta.parents.values()])
    return obj.__class__(**initial)

2.自定义自增字段

产生原因:我们的项目我来公司之前的表没有设置id字段,使用Char类型的字段当做主键的,但是后期项目需要我给所有的表添加自增的id
字段,但是如果你在django里面使用AutoField那么primary_key=True是必须的,而现在我只需要一个自增的主键id,所以通过google和
看AutoField的源码拼出来了一个新字段.

from django.db.models.fields import Field, IntegerField
from django.core import checks, exceptions
from django.utils.translation import ugettext_lazy as _


class AutoIncreField(Field):
    def __init__(self, *args, **kwargs):
        kwargs["blank"] = True
        super(AutoIncreField, self).__init__(*args, **kwargs)
        
    empty_strings_allowed = False
    default_error_messages = {
        'invalid': _('“%(value)s” value must be an integer.'),
    }
    description = _("Integer")

    def check(self, **kwargs):
        return [
            *super().check(**kwargs),
            *self._check_key(),
        ]

    def _check_key(self):
        if not self.unique:
            return [
                checks.Error(
                    'AutoIncreFields must set key(unique=True).',
                    obj=self,
                    id='fields.E100',
                ),
            ]
        else:
            return []

    def validate(self, value, model_instance):
        pass

    def get_prep_value(self, value):
        value = super().get_prep_value(value)
        if value is None:
            return None
        try:
            return int(value)
        except (TypeError, ValueError) as e:
            raise e.__class__(
                "Field '%s' expected a number but got %r." % (self.name, value),
            ) from e

    def deconstruct(self):
        name, path, args, kwargs = super().deconstruct()
        del kwargs['blank']
        kwargs['unique'] = True
        return name, path, args, kwargs

    def get_internal_type(self):
        return "AutoIncreField"

    def to_python(self, value):
        if value is None:
            return value
        try:
            return int(value)
        except (TypeError, ValueError):
            raise exceptions.ValidationError(
                self.error_messages['invalid'],
                code='invalid',
                params={'value': value},
            )

    def rel_db_type(self, connection):
        return IntegerField().db_type(connection=connection)

    def get_db_prep_value(self, value, connection, prepared=False):
        if not prepared:
            value = self.get_prep_value(value)
            value = connection.ops.validate_autopk_value(value)
        return value

    def contribute_to_class(self, cls, name, **kwargs):
        assert not cls._meta.auto_field, (
            "Model %s can't have more than one auto-generated field."
            % cls._meta.label
        )
        super().contribute_to_class(cls, name, **kwargs)
        cls._meta.auto_field = self

    def formfield(self, **kwargs):
        return None
上一篇:7-函数


下一篇:13、函数的参数