libervia-backend: sat/tools/images.py annotate

annotate sat/tools/images.py @ 3200:5c3bf37f2202

tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: `context` has been removed as it is not used and doesn't seem that useful.

author	Goffi <goffi@goffi.org>
date	Sun, 01 Mar 2020 18:47:05 +0100
parents	8b4354b5c05f
children

rev	line source
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	1 #!/usr/bin/env python3
3137 559a625a236b fixed shebangs Goffi <goffi@goffi.org> parents: 3136 diff changeset	2
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	3
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	4 # SAT: a jabber client
3136 9d0df638c8b4 dates update Goffi <goffi@goffi.org> parents: 3066 diff changeset	5 # Copyright (C) 2009-2020 Jérôme Poisson (goffi@goffi.org)
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	6
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	7 # This program is free software: you can redistribute it and/or modify
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	8 # it under the terms of the GNU Affero General Public License as published by
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	9 # the Free Software Foundation, either version 3 of the License, or
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	10 # (at your option) any later version.
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	11
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	12 # This program is distributed in the hope that it will be useful,
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	15 # GNU Affero General Public License for more details.
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	16
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	17 # You should have received a copy of the GNU Affero General Public License
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	19
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	20 """Methods to manipulate images"""
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	21
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	22 import tempfile
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	23 from PIL import Image
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	24 from pathlib import Path
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	25 from twisted.internet import threads
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	26
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	27
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	28 def checkImage(host, path, max_size=None):
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	29 """Analyze image and return a report
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	30
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	31 report will indicate if image is too large, and the recommended new size if this is
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	32 the case
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	33 @param host: SàT instance
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	34 @param path(str, pathlib.Path): image to open
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	35 @param max_size(tuple[int, int]): maximum accepted size of image
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	36 None to use value set in config
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	37 @return dict: report on image, with following keys:
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	38 - too_large: true if image is oversized
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	39 - recommended_size: if too_large is True, recommended size to use
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	40 """
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	41 report = {}
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	42 image = Image.open(path)
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	43 if max_size is None:
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	44 max_size = tuple(host.memory.getConfig(None, "image_max", (1200, 720)))
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	45 if image.size > max_size:
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	46 report['too_large'] = True
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	47 if image.size[0] > max_size[0]:
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	48 factor = max_size[0] / image.size[0]
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	49 if image.size[1] * factor > max_size[1]:
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	50 factor = max_size[1] / image.size[1]
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	51 else:
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	52 factor = max_size[1] / image.size[1]
3157 8b4354b5c05f tools (images): fixed type for recommended_size. Goffi <goffi@goffi.org> parents: 3137 diff changeset	53 report['recommended_size'] = [int(image.widthfactor), int(image.heightfactor)]
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	54 else:
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	55 report['too_large'] = False
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	56
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	57 return report
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	58
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	59
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	60 def _resizeImageBlocking(image_path, new_size, dest=None):
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	61 im_path = Path(image_path)
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	62 im = Image.open(im_path)
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	63 resized = im.resize(new_size, Image.LANCZOS)
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	64
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	65 if dest is None:
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	66 dest = tempfile.NamedTemporaryFile(suffix=im_path.suffix, delete=False)
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	67 elif isinstance(dest, Path):
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	68 dest = Path.open('wb')
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	69
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	70 with dest as f:
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	71 resized.save(f, format=im.format)
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	72
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	73 return Path(f.name)
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	74
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	75
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	76 def resizeImage(image_path, new_size, dest=None):
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	77 """Resize an image to a new file, and return its path
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	78
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	79 @param image_path(str, Path): path of the original image
2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	80 @param new_size(tuple[int, int]): size to use for new image
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	81 @param dest(None, Path, file): where the resized image must be stored, can be:
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	82 - None: use a temporary file
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	83 - Path: path to the file to create/overwrite
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	84 - file: a file object which must be opened for writing in binary mode
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	85 @return (Path): path of the resized file.
5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	86 The image at this path should be deleted after use
3066 2cc2f65379f7 core: added imageCheck and imageResize methods: Goffi <goffi@goffi.org> parents: diff changeset	87 """
3200 5c3bf37f2202 tools (images): max_size can now be manually specified in checkImage and dest in resizeImage: Goffi <goffi@goffi.org> parents: 3157 diff changeset	88 return threads.deferToThread(_resizeImageBlocking, image_path, new_size, dest)

Mercurial > libervia-backend

annotate sat/tools/images.py @ 3200:5c3bf37f2202