python - 在 Django 模板中获取文件扩展名

Question

我有这样的模型:

class File(models.Model):
    name = models.CharField(max_length=45)
    description = models.CharField(max_length=100, blank=True)
    file = models.FileField(upload_to='files')

我在 View 中获取了所有 File 对象，并且根据文件类型，我想打印适当的类:

<a class="pdf">link</a>

或

<a class="word">link</a>

或

<a class="other">link<a/>

在我的模板中。

如何在 Django 模板中获取文件扩展名？

我想做这样的事情:

{% for file in files %}
    {% if file.get_extension == 'pdf' %}
           <a class="pdf">link</a>
    {% elif file.get_extension = '.doc' %}
           <a class="word">link</a>
    {% else %}
           <a class="other">link<a/>
    {% endif %}
{% endfor %}

当然，'file.get_extension' 不存在...

Answer 1

您的模型缺少 .get_extension 吗？这很简单，只需添加它 :-) 您可以在模型上使用各种方法。所以像这样:

class File(models.Model):
    name = models.CharField(max_length=45)
    description = models.CharField(max_length=100, blank=True)
    file = models.FileField(upload_to='files')

    def extension(self):
        name, extension = os.path.splitext(self.file.name)
        return extension

(名称 .extension() 比 .get_extension() 更符合 pythonic，顺便说一句)。

你可以走得更远。你的模板中的 if/else 结构是不是有点乏味？它在 Python 代码中不那么麻烦:

class File(models.Model):
    ...
    def css_class(self):
        name, extension = os.path.splitext(self.file.name)
        if extension == 'pdf':
            return 'pdf'
        if extension == 'doc':
            return 'word'
        return 'other'

这样模板更简单:

{% for file in files %}
  <a class="{{ file.css_class }}">link</a>
{% endfor %}