如何覆盖第三方安装的应用程序的模型默认方法,django?

时间:2017-06-20 13:54:36

标签: python django model override

我刚刚安装了this django-csvimport package。现在我想覆盖Admin区域表单中的默认值。我找到了code here,它定义了模型,并包含当前的默认文本:

class CSVImport(models.Model):
    """ Logging model for importing files """
    model_choice = []
    model_name = models.CharField(max_length=255, blank=False,
                                  default='csvimport.Item',
                                  help_text='Please specify the app_label.model_name',
                                  choices=get_models())
    field_list = models.TextField(blank=True,
                                  help_text='''Enter list of fields in order only if
                                     you dont have a header row with matching field names, eg.
                                     "column1=shared_code,column2=org(Organisation|name)"''')
    upload_file = models.FileField(upload_to='csv', storage=fs)
    file_name = models.CharField(max_length=255, blank=True)
    encoding = models.CharField(max_length=32, blank=True)
    upload_method = models.CharField(blank=False, max_length=50,
                                     default='manual', choices=CHOICES)
    error_log = models.TextField(help_text='Each line is an import error')
    import_date = models.DateField(auto_now=True)
    import_user = models.CharField(max_length=255, default='anonymous',
                                   help_text='User id as text', blank=True)

    def error_log_html(self):
        return re.sub('\n', '<br/>', self.error_log)
    error_log_html.allow_tags = True

    def __unicode__(self):
        return self.upload_file.name

例如,我想用其他内容覆盖model_name字段默认csvimport.Item。我有点不知道如何覆盖这个,因为我没有csvimport的app文件夹,因为它是第3部分安装。这将是我第一次覆盖第三方安装的应用程序模型。

现在我再研究一下,不确定我是否应该覆盖这个模型,或者更好的ModelAdmin文件的admin.py类?

谢谢!

2 个答案:

答案 0 :(得分:2)

"""Your admin.py"""

from csvimport.models import CSVImport
from csvimport.admin import CSVImportAdmin

class MyCSVImportAdmin(CSVImportAdmin):
    """Override some of the form's field properties:
       clean, creation_counter, default_error_messages, 
       default_validators, disabled, empty_value, empty_values .. etc
    """

    def get_form(self, request, obj=None, **kwargs):
        form = super(MyCSVImportAdmin, self).get_form(request, obj, **kwargs)
        form.base_fields["model_name"].initial = 'What you want'
        form.base_fields["model_name"].help_text = 'Please customize the fields however you like'
        return form

admin.site.unregister(CSVImport)
admin.site.register(CSVImport, MyCSVImportAdmin)

答案 1 :(得分:1)

我看到整个代码和django-csvimport包没有为您提供覆盖代码中任何内容的功能,因此如果不将应用程序复制到项目中,则无法覆盖它。下面是另一个应用django-oauth-toolkit的示例,它使用用户设置参数来提供修改功能。

USER_SETTINGS = getattr(settings, "OAUTH2_PROVIDER", None)

现在解决方案只是复制应用程序,然后修改应用程序以供您自己使用。