如何使用 Django / Python 选择特定大小的随机图像？

Question

我一直在使用这个小 sn-p 来选择随机图像。但是我想将其更改为仅选择特定尺寸的图像。我在检查图像大小时遇到​​了麻烦。如果我使用 get_image_dimensions() 我需要使用条件语句，这要求我允许异常。所以，我想我需要一些关于仅限制图像尺寸的指示。谢谢。

import os
import random
import posixpath
from django import template
from django.conf import settings

register = template.Library()

def is_image_file(filename):
    """Does `filename` appear to be an image file?"""
    img_types = [".jpg", ".jpeg", ".png", ".gif"]
    ext = os.path.splitext(filename)[1]
    return ext in img_types

@register.simple_tag
def random_image(path):
    """
    Select a random image file from the provided directory
    and return its href. `path` should be relative to MEDIA_ROOT.

    Usage:  <img src='{% random_image "images/whatever/" %}'>
    """
    fullpath = os.path.join(settings.MEDIA_ROOT, path)
    filenames = [f for f in os.listdir(fullpath) if is_image_file(f)]
    pick = random.choice(filenames)
    return posixpath.join(settings.MEDIA_URL, path, pick)

Answer 1

嗯，获取图像尺寸的一种更直接的方法是使用 Python 图像库（无论如何，这是 Django 在后端用于 get_image_dimensions 的方法）。

所以，你可以像这样使用它：

>> import Image
>> img = Image.open("foo.png")
>> img.size
(1729,828)

你最简单的解决方案是这样的：

img_dimensions = lambda f: Image.open(f).size
filenames = filter(lambda f: is_image_file(f) and img_dimensions(f)==my_dimensions, os.listdir(fullpath))

my_dimensions 在哪里——你的尺寸。

问题是，这个（以及任何其他检查文件尺寸的方法）实际上每次都必须打开和读取图像，重复这样做是不明智的。因此，根据应用程序的负载，您可能希望将 img_dimensions() 放入一个真实的、单独的函数中并对其进行记忆 (http://code.activestate.com/recipes/52201-memoizing-cacheing-function-return-values/)。或者，如果这将成为一项常见任务，只需将图像放入按尺寸预先排序的文件夹中。