آپلود امن عکس؛ محدودیت حجم و اعتبارسنجی

این فایل را با همین نام و بدون حذف سطرها کنار بقیهٔ فایل‌ها ذخیره کنید.

{"maxBytes": 1000000, "maxPixels": 4000000, "maxWidth": 1000}

دستور اجرا، نتیجهٔ مورد انتظار و محدودیت این فایل در بخش اجرای مقاله توضیح داده شده است.

این فایل را با همین نام و بدون حذف سطرها کنار بقیهٔ فایل‌ها ذخیره کنید.

from io import BytesIO
from pathlib import Path
import uuid
from PIL import Image, ImageOps


def store_avatar(stream, directory, policy):
    data = stream.read(policy['maxBytes'] + 1)
    if len(data) > policy['maxBytes']:
        raise ValueError('Upload exceeds byte limit')
    with Image.open(BytesIO(data)) as original:
        if original.format not in {'JPEG', 'PNG'} or getattr(original, 'is_animated', False):
            raise ValueError('Only single JPEG or PNG images are accepted')
        if original.width * original.height > policy['maxPixels']:
            raise ValueError('Image exceeds pixel limit')
        original.load()
        image = ImageOps.exif_transpose(original).convert('RGB')
        image.thumbnail((policy['maxWidth'], policy['maxWidth']))
        root = Path(directory); root.mkdir(mode=0o700, parents=True, exist_ok=True)
        target = root / (uuid.uuid4().hex + '.webp')
        with target.open('xb') as output:
            image.save(output, format='WEBP', quality=85)
    return target.name

دستور اجرا، نتیجهٔ مورد انتظار و محدودیت این فایل در بخش اجرای مقاله توضیح داده شده است.

این فایل را با همین نام و بدون حذف سطرها کنار بقیهٔ فایل‌ها ذخیره کنید.

from io import BytesIO
import json
from pathlib import Path
from PIL import Image
from avatar_store import store_avatar

if __name__ == '__main__':
    policy = json.loads(Path(__file__).with_name('policy.json').read_text())
    incoming = BytesIO(); Image.new('RGB', (600, 300), 'green').save(incoming, format='PNG'); incoming.seek(0)
    print(store_avatar(incoming, 'private-avatars', policy))

دستور اجرا، نتیجهٔ مورد انتظار و محدودیت این فایل در بخش اجرای مقاله توضیح داده شده است.

این فایل را با همین نام و بدون حذف سطرها کنار بقیهٔ فایل‌ها ذخیره کنید.

from io import BytesIO
from pathlib import Path
import tempfile
import unittest
from PIL import Image, UnidentifiedImageError
from avatar_store import store_avatar


class UploadTests(unittest.TestCase):
    def test_reencode_and_generated_name(self):
        with tempfile.TemporaryDirectory() as directory:
            incoming = BytesIO(); Image.new('RGB', (60, 30)).save(incoming, format='PNG'); incoming.seek(0)
            name = store_avatar(incoming, directory, {'maxBytes': 10000, 'maxPixels': 10000, 'maxWidth': 40})
            self.assertRegex(name, r'^[a-f0-9]{32}\.webp$')
            with Image.open(Path(directory) / name) as image: self.assertEqual(image.size, (40, 20))

    def test_invalid_bytes_and_size(self):
        with tempfile.TemporaryDirectory() as directory:
            policy = {'maxBytes': 10, 'maxPixels': 10000, 'maxWidth': 40}
            with self.assertRaises(ValueError): store_avatar(BytesIO(b'x' * 11), directory, policy)
            with self.assertRaises(UnidentifiedImageError): store_avatar(BytesIO(b'fake.jpg'), directory, policy)


if __name__ == '__main__': unittest.main()

دستور اجرا، نتیجهٔ مورد انتظار و محدودیت این فایل در بخش اجرای مقاله توضیح داده شده است.