mirror of https://github.com/muk-it/muk_base
MuK IT GmbH
5 years ago
3 changed files with 390 additions and 390 deletions
-
2muk_thumbnails/__manifest__.py
-
356muk_thumbnails/doc/index.rst
-
422muk_thumbnails/service/thumbnail.py
@ -1,178 +1,178 @@ |
|||||
============== |
|
||||
MuK Thumbnails |
|
||||
============== |
|
||||
|
|
||||
Technical module to provide some utility features and libraries that can be used |
|
||||
in other applications. This module has no direct effect on the running system. |
|
||||
|
|
||||
Requirements |
|
||||
============= |
|
||||
|
|
||||
Unoconv |
|
||||
------------- |
|
||||
|
|
||||
Universal Office Converter (unoconv) is a command line tool to convert any |
|
||||
document format that LibreOffice can import to any document format that |
|
||||
LibreOffice can export. It makes use of the LibreOffice's UNO bindings for |
|
||||
non-interactive conversion of documents. |
|
||||
|
|
||||
To install unoconv please follow the instructions (`here <https://github.com/dagwieers/unoconv>`_). |
|
||||
|
|
||||
Make sure that unoconv can be executed from the console and the conversion |
|
||||
is done correctly. |
|
||||
|
|
||||
To set an individual path for the LibreOffice installation, the config |
|
||||
variable ``uno_path`` can be set accordingly in the Odoo config. |
|
||||
|
|
||||
Under Windows you should rename the ``unoconv`` file to ``unoconv.py`` and set |
|
||||
the corresponding path for the ``uno_path`` variable. Since it does not work |
|
||||
reliably with the environment variable ``UNO_PATH``. |
|
||||
|
|
||||
FFmpeg |
|
||||
------------- |
|
||||
|
|
||||
FFmpeg is a complete, cross-platform solution to record, convert and stream audio and video. |
|
||||
|
|
||||
To install FFmpeg please follow the instructions (`here <https://www.ffmpeg.org/download.html>`_). |
|
||||
|
|
||||
Ghostscript |
|
||||
------------- |
|
||||
|
|
||||
Ghostscript is a suite of software based on an interpreter for Adobe Systems PostScript and |
|
||||
Portable Document Format (PDF) page description languages. Its main purposes are the |
|
||||
rasterization or rendering of such page description language files, for the display or printing |
|
||||
of document pages, and the conversion between PostScript and PDF files. |
|
||||
|
|
||||
To install Ghostscript please follow the instructions (`here <https://www.ghostscript.com/download.html>`_). |
|
||||
|
|
||||
ImageMagick |
|
||||
------------- |
|
||||
|
|
||||
ImageMagick can be used to create, edit, compose, or convert bitmap images. It can read and write |
|
||||
images in a variety of formats (over 200) including PNG, JPEG, GIF, HEIC, TIFF, DPX, EXR, WebP, |
|
||||
Postscript, PDF, and SVG. Use ImageMagick to resize, flip, mirror, rotate, distort, shear and |
|
||||
transform images, adjust image colors, apply various special effects, or draw text, lines, polygons, |
|
||||
ellipses and Bezier curves. |
|
||||
|
|
||||
To install ImageMagick please follow the instructions (`here <https://www.imagemagick.org/script/download.php>`_). |
|
||||
|
|
||||
Wand |
|
||||
------------- |
|
||||
|
|
||||
Wand is a ctypes-based simple ImageMagick binding for Python. It is used to interact with ImageMagick via a Python interface. |
|
||||
|
|
||||
To install Wand please follow the instructions (`here <http://docs.wand-py.org/en/0.4.1/guide/install.html>`_). |
|
||||
|
|
||||
Imageio |
|
||||
------------- |
|
||||
|
|
||||
Imageio is a Python library that provides an easy interface to read and write a wide range of image data, including animated images, volumetric data, and scientific formats. |
|
||||
|
|
||||
To install Imageio please follow the instructions (`here <https://pypi.org/project/imageio/>`_). |
|
||||
|
|
||||
MoviePy |
|
||||
------------- |
|
||||
|
|
||||
MoviePy is a Python module for video editing, which can be used for basic operations (like cuts, concatenations, title insertions), video compositing (a.k.a. non-linear editing), video processing, or to create advanced effects. It can read and write the most common video formats, including GIF. |
|
||||
|
|
||||
To install MoviePy please follow the instructions (`here <https://zulko.github.io/moviepy/install.html>`_). |
|
||||
|
|
||||
Installation |
|
||||
============ |
|
||||
|
|
||||
To install this module, you need to: |
|
||||
|
|
||||
Download the module and add it to your Odoo addons folder. Afterward, log on to |
|
||||
your Odoo server and go to the Apps menu. Trigger the debug mode and update the |
|
||||
list by clicking on the "Update Apps List" link. Now install the module by |
|
||||
clicking on the install button. |
|
||||
|
|
||||
Another way to install this module is via the package management for Python |
|
||||
(`PyPI <https://pypi.org/project/pip/>`_). |
|
||||
|
|
||||
To install our modules using the package manager make sure |
|
||||
`odoo-autodiscover <https://pypi.org/project/odoo-autodiscover/>`_ is installed |
|
||||
correctly. Then open a console and install the module by entering the following |
|
||||
command: |
|
||||
|
|
||||
``pip install --extra-index-url https://nexus.mukit.at/repository/odoo/simple <module>`` |
|
||||
|
|
||||
The module name consists of the Odoo version and the module name, where |
|
||||
underscores are replaced by a dash. |
|
||||
|
|
||||
**Module:** |
|
||||
|
|
||||
``odoo<version>-addon-<module_name>`` |
|
||||
|
|
||||
**Example:** |
|
||||
|
|
||||
``sudo -H pip3 install --extra-index-url https://nexus.mukit.at/repository/odoo/simple odoo11-addon-muk-utils`` |
|
||||
|
|
||||
Once the installation has been successfully completed, the app is already in the |
|
||||
correct folder. Log on to your Odoo server and go to the Apps menu. Trigger the |
|
||||
debug mode and update the list by clicking on the "Update Apps List" link. Now |
|
||||
install the module by clicking on the install button. |
|
||||
|
|
||||
The biggest advantage of this variant is that you can now also update the app |
|
||||
using the "pip" command. To do this, enter the following command in your console: |
|
||||
|
|
||||
``pip install --upgrade --extra-index-url https://nexus.mukit.at/repository/odoo/simple <module>`` |
|
||||
|
|
||||
When the process is finished, restart your server and update the application in |
|
||||
Odoo. The steps are the same as for the installation only the button has changed |
|
||||
from "Install" to "Upgrade". |
|
||||
|
|
||||
You can also view available Apps directly in our `repository <https://nexus.mukit.at/#browse/browse:odoo>`_ |
|
||||
and find a more detailed installation guide on our `website <https://mukit.at/page/open-source>`_. |
|
||||
|
|
||||
For modules licensed under OPL-1, you will receive access data when you purchase |
|
||||
the module. If the modules were not purchased directly from |
|
||||
`MuK IT <https://www.mukit.at/>`_ please contact our support (support@mukit.at) |
|
||||
with a confirmation of purchase to receive the corresponding access data. |
|
||||
|
|
||||
Upgrade |
|
||||
============ |
|
||||
|
|
||||
To upgrade this module, you need to: |
|
||||
|
|
||||
Download the module and add it to your Odoo addons folder. Restart the server |
|
||||
and log on to your Odoo server. Select the Apps menu and upgrade the module by |
|
||||
clicking on the upgrade button. |
|
||||
|
|
||||
If you installed the module using the "pip" command, you can also update the |
|
||||
module in the same way. Just type the following command into the console: |
|
||||
|
|
||||
``pip install --upgrade --extra-index-url https://nexus.mukit.at/repository/odoo/simple <module>`` |
|
||||
|
|
||||
When the process is finished, restart your server and update the application in |
|
||||
Odoo, just like you would normally. |
|
||||
|
|
||||
Configuration |
|
||||
============= |
|
||||
|
|
||||
No additional configuration is needed to use this module. |
|
||||
|
|
||||
Usage |
|
||||
============= |
|
||||
|
|
||||
This module has no direct visible effect on the system. It provide utility features. |
|
||||
|
|
||||
Credits |
|
||||
======= |
|
||||
|
|
||||
Contributors |
|
||||
------------ |
|
||||
|
|
||||
* Mathias Markl <mathias.markl@mukit.at> |
|
||||
|
|
||||
Author & Maintainer |
|
||||
------------------- |
|
||||
|
|
||||
This module is maintained by the `MuK IT GmbH <https://www.mukit.at/>`_. |
|
||||
|
|
||||
MuK IT is an Austrian company specialized in customizing and extending Odoo. |
|
||||
We develop custom solutions for your individual needs to help you focus on |
|
||||
your strength and expertise to grow your business. |
|
||||
|
|
||||
If you want to get in touch please contact us via mail |
|
||||
(sale@mukit.at) or visit our website (https://mukit.at). |
|
||||
|
============== |
||||
|
MuK Thumbnails |
||||
|
============== |
||||
|
|
||||
|
Technical module to provide some utility features and libraries that can be used |
||||
|
in other applications. This module has no direct effect on the running system. |
||||
|
|
||||
|
Requirements |
||||
|
============= |
||||
|
|
||||
|
Unoconv |
||||
|
------------- |
||||
|
|
||||
|
Universal Office Converter (unoconv) is a command line tool to convert any |
||||
|
document format that LibreOffice can import to any document format that |
||||
|
LibreOffice can export. It makes use of the LibreOffice's UNO bindings for |
||||
|
non-interactive conversion of documents. |
||||
|
|
||||
|
To install unoconv please follow the instructions (`here <https://github.com/dagwieers/unoconv>`_). |
||||
|
|
||||
|
Make sure that unoconv can be executed from the console and the conversion |
||||
|
is done correctly. |
||||
|
|
||||
|
To set an individual path for the LibreOffice installation, the config |
||||
|
variable ``uno_path`` can be set accordingly in the Odoo config. |
||||
|
|
||||
|
Under Windows you should rename the ``unoconv`` file to ``unoconv.py`` and set |
||||
|
the corresponding path for the ``uno_path`` variable. Since it does not work |
||||
|
reliably with the environment variable ``UNO_PATH``. |
||||
|
|
||||
|
FFmpeg |
||||
|
------------- |
||||
|
|
||||
|
FFmpeg is a complete, cross-platform solution to record, convert and stream audio and video. |
||||
|
|
||||
|
To install FFmpeg please follow the instructions (`here <https://www.ffmpeg.org/download.html>`_). |
||||
|
|
||||
|
Ghostscript |
||||
|
------------- |
||||
|
|
||||
|
Ghostscript is a suite of software based on an interpreter for Adobe Systems PostScript and |
||||
|
Portable Document Format (PDF) page description languages. Its main purposes are the |
||||
|
rasterization or rendering of such page description language files, for the display or printing |
||||
|
of document pages, and the conversion between PostScript and PDF files. |
||||
|
|
||||
|
To install Ghostscript please follow the instructions (`here <https://www.ghostscript.com/download.html>`_). |
||||
|
|
||||
|
ImageMagick |
||||
|
------------- |
||||
|
|
||||
|
ImageMagick can be used to create, edit, compose, or convert bitmap images. It can read and write |
||||
|
images in a variety of formats (over 200) including PNG, JPEG, GIF, HEIC, TIFF, DPX, EXR, WebP, |
||||
|
Postscript, PDF, and SVG. Use ImageMagick to resize, flip, mirror, rotate, distort, shear and |
||||
|
transform images, adjust image colors, apply various special effects, or draw text, lines, polygons, |
||||
|
ellipses and Bezier curves. |
||||
|
|
||||
|
To install ImageMagick please follow the instructions (`here <https://www.imagemagick.org/script/download.php>`_). |
||||
|
|
||||
|
Wand |
||||
|
------------- |
||||
|
|
||||
|
Wand is a ctypes-based simple ImageMagick binding for Python. It is used to interact with ImageMagick via a Python interface. |
||||
|
|
||||
|
To install Wand please follow the instructions (`here <http://docs.wand-py.org/en/0.4.1/guide/install.html>`_). |
||||
|
|
||||
|
Imageio |
||||
|
------------- |
||||
|
|
||||
|
Imageio is a Python library that provides an easy interface to read and write a wide range of image data, including animated images, volumetric data, and scientific formats. |
||||
|
|
||||
|
To install Imageio please follow the instructions (`here <https://pypi.org/project/imageio/>`_). |
||||
|
|
||||
|
MoviePy |
||||
|
------------- |
||||
|
|
||||
|
MoviePy is a Python module for video editing, which can be used for basic operations (like cuts, concatenations, title insertions), video compositing (a.k.a. non-linear editing), video processing, or to create advanced effects. It can read and write the most common video formats, including GIF. |
||||
|
|
||||
|
To install MoviePy please follow the instructions (`here <https://zulko.github.io/moviepy/install.html>`_). |
||||
|
|
||||
|
Installation |
||||
|
============ |
||||
|
|
||||
|
To install this module, you need to: |
||||
|
|
||||
|
Download the module and add it to your Odoo addons folder. Afterward, log on to |
||||
|
your Odoo server and go to the Apps menu. Trigger the debug mode and update the |
||||
|
list by clicking on the "Update Apps List" link. Now install the module by |
||||
|
clicking on the install button. |
||||
|
|
||||
|
Another way to install this module is via the package management for Python |
||||
|
(`PyPI <https://pypi.org/project/pip/>`_). |
||||
|
|
||||
|
To install our modules using the package manager make sure |
||||
|
`odoo-autodiscover <https://pypi.org/project/odoo-autodiscover/>`_ is installed |
||||
|
correctly. Then open a console and install the module by entering the following |
||||
|
command: |
||||
|
|
||||
|
``pip install --extra-index-url https://nexus.mukit.at/repository/odoo/simple <module>`` |
||||
|
|
||||
|
The module name consists of the Odoo version and the module name, where |
||||
|
underscores are replaced by a dash. |
||||
|
|
||||
|
**Module:** |
||||
|
|
||||
|
``odoo<version>-addon-<module_name>`` |
||||
|
|
||||
|
**Example:** |
||||
|
|
||||
|
``sudo -H pip3 install --extra-index-url https://nexus.mukit.at/repository/odoo/simple odoo11-addon-muk-utils`` |
||||
|
|
||||
|
Once the installation has been successfully completed, the app is already in the |
||||
|
correct folder. Log on to your Odoo server and go to the Apps menu. Trigger the |
||||
|
debug mode and update the list by clicking on the "Update Apps List" link. Now |
||||
|
install the module by clicking on the install button. |
||||
|
|
||||
|
The biggest advantage of this variant is that you can now also update the app |
||||
|
using the "pip" command. To do this, enter the following command in your console: |
||||
|
|
||||
|
``pip install --upgrade --extra-index-url https://nexus.mukit.at/repository/odoo/simple <module>`` |
||||
|
|
||||
|
When the process is finished, restart your server and update the application in |
||||
|
Odoo. The steps are the same as for the installation only the button has changed |
||||
|
from "Install" to "Upgrade". |
||||
|
|
||||
|
You can also view available Apps directly in our `repository <https://nexus.mukit.at/#browse/browse:odoo>`_ |
||||
|
and find a more detailed installation guide on our `website <https://mukit.at/page/open-source>`_. |
||||
|
|
||||
|
For modules licensed under OPL-1, you will receive access data when you purchase |
||||
|
the module. If the modules were not purchased directly from |
||||
|
`MuK IT <https://www.mukit.at/>`_ please contact our support (support@mukit.at) |
||||
|
with a confirmation of purchase to receive the corresponding access data. |
||||
|
|
||||
|
Upgrade |
||||
|
============ |
||||
|
|
||||
|
To upgrade this module, you need to: |
||||
|
|
||||
|
Download the module and add it to your Odoo addons folder. Restart the server |
||||
|
and log on to your Odoo server. Select the Apps menu and upgrade the module by |
||||
|
clicking on the upgrade button. |
||||
|
|
||||
|
If you installed the module using the "pip" command, you can also update the |
||||
|
module in the same way. Just type the following command into the console: |
||||
|
|
||||
|
``pip install --upgrade --extra-index-url https://nexus.mukit.at/repository/odoo/simple <module>`` |
||||
|
|
||||
|
When the process is finished, restart your server and update the application in |
||||
|
Odoo, just like you would normally. |
||||
|
|
||||
|
Configuration |
||||
|
============= |
||||
|
|
||||
|
No additional configuration is needed to use this module. |
||||
|
|
||||
|
Usage |
||||
|
============= |
||||
|
|
||||
|
This module has no direct visible effect on the system. It provide utility features. |
||||
|
|
||||
|
Credits |
||||
|
======= |
||||
|
|
||||
|
Contributors |
||||
|
------------ |
||||
|
|
||||
|
* Mathias Markl <mathias.markl@mukit.at> |
||||
|
|
||||
|
Author & Maintainer |
||||
|
------------------- |
||||
|
|
||||
|
This module is maintained by the `MuK IT GmbH <https://www.mukit.at/>`_. |
||||
|
|
||||
|
MuK IT is an Austrian company specialized in customizing and extending Odoo. |
||||
|
We develop custom solutions for your individual needs to help you focus on |
||||
|
your strength and expertise to grow your business. |
||||
|
|
||||
|
If you want to get in touch please contact us via mail |
||||
|
(sale@mukit.at) or visit our website (https://mukit.at). |
||||
@ -1,211 +1,211 @@ |
|||||
################################################################################### |
|
||||
# |
|
||||
# Copyright (C) 2018 MuK IT GmbH |
|
||||
# |
|
||||
# This program is free software: you can redistribute it and/or modify |
|
||||
# it under the terms of the GNU Affero General Public License as |
|
||||
# published by the Free Software Foundation, either version 3 of the |
|
||||
# License, or (at your option) any later version. |
|
||||
# |
|
||||
# This program is distributed in the hope that it will be useful, |
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
||||
# GNU Affero General Public License for more details. |
|
||||
# |
|
||||
# You should have received a copy of the GNU Affero General Public License |
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
||||
# |
|
||||
################################################################################### |
|
||||
|
|
||||
import os |
|
||||
import io |
|
||||
import sys |
|
||||
import PyPDF2 |
|
||||
import base64 |
|
||||
import shutil |
|
||||
import urllib |
|
||||
import logging |
|
||||
import tempfile |
|
||||
import mimetypes |
|
||||
|
|
||||
from contextlib import closing |
|
||||
|
|
||||
from odoo.tools import config |
|
||||
from odoo.tools.mimetypes import guess_mimetype |
|
||||
|
|
||||
from odoo.addons.muk_utils.tools import utils_os |
|
||||
from odoo.addons.muk_converter.service import unoconv |
|
||||
|
|
||||
_logger = logging.getLogger(__name__) |
|
||||
|
|
||||
try: |
|
||||
from wand.image import Image |
|
||||
from wand.color import Color |
|
||||
except ImportError: |
|
||||
Image = False |
|
||||
Color = False |
|
||||
_logger.warn('Cannot `import wand`.') |
|
||||
|
|
||||
try: |
|
||||
import imageio |
|
||||
except ImportError: |
|
||||
imageio = False |
|
||||
_logger.warn('Cannot `import imageio`.') |
|
||||
|
|
||||
try: |
|
||||
from moviepy.editor import VideoFileClip |
|
||||
except ImportError: |
|
||||
VideoFileClip = False |
|
||||
_logger.warn('Cannot `import moviepy`.') |
|
||||
|
|
||||
FORMATS = [ |
|
||||
"png", "jpg", "jepg" |
|
||||
] |
|
||||
|
|
||||
VIDEO_IMPORTS = [ |
|
||||
"mp4", "mov", "wav", "avi", "mpg", "flv", "wmv", "webm" |
|
||||
] |
|
||||
|
|
||||
PDF_IMPORTS = [ |
|
||||
"pdf" |
|
||||
] |
|
||||
|
|
||||
WAND_IMPORTS = [ |
|
||||
"aai", "art", "arw", "avi", "avs", "bpg", "brf", "cals", "cgm", "cin", "cip", "cmyk", "cmyka", "svg", |
|
||||
"cr2", "crw", "cur", "cut", "dcm", "dcr", "dcx", "dds", "dib", "djvu", "dng", "dot", "dpx", "tim", |
|
||||
"emf", "epdf", "epi", "eps", "eps2", "eps3", "epsf", "epsi", "ept", "exr", "fax", "fig", "fits", |
|
||||
"fpx", "gplt", "gray", "graya", "hdr", "hdr", "heic", "hpgl", "hrz", "html", "ico", "info", "ttf", |
|
||||
"inline", "isobrl", "isobrl6", "jbig", "jng", "jp2", "jpt", "j2c", "j2k", "jxr", "json", "man", "bmp", |
|
||||
"mat", "miff", "mono", "mng", "m2v", "mpeg", "mpc", "mpr", "mrw", "msl", "mtv", "mvg", "nef", "yuv", |
|
||||
"orf", "otb", "p7", "palm", "pam", "clipboard", "pbm", "pcd", "pcds", "pcl", "pcx", "pdb", "jpe", |
|
||||
"pef", "pes", "pfa", "pfb", "pfm", "pgm", "picon", "pict", "pix", "png8", "png00", "png24", "tiff", |
|
||||
"png32", "png48", "png64", "pnm", "ppm", "ps", "ps2", "ps3", "psb", "psd", "ptif", "pwp", "rad", |
|
||||
"raf", "rgb", "rgba", "rgf", "rla", "rle", "sct", "sfw", "sgi", "shtml", "sid", " mrsid", "jpeg", |
|
||||
"sparse-color", "sun", "tga", "ubrl", "ubrl6", "uyvy", "vicar", "viff", "wbmp", "jpg", "png", "uil", |
|
||||
"wdp", "webp", "wmf", "wpg", "x", "xbm", "xcf", "xpm", "xwd", "x3f", "ycbcr", "ycbcra", "bmp3", "bmp2", |
|
||||
] |
|
||||
|
|
||||
def formats(): |
|
||||
return FORMATS |
|
||||
|
|
||||
def imports(): |
|
||||
return VIDEO_IMPORTS + PDF_IMPORTS + WAND_IMPORTS + unoconv.IMPORTS |
|
||||
|
|
||||
def create_thumbnail(binary, mimetype=None, filename=None, export="binary", format="png", page=0, frame=0, |
|
||||
animation=False, video_resize={'width': 256}, image_resize='256x256>', image_crop=None): |
|
||||
""" |
|
||||
Converts a thumbnail for a given file. |
|
||||
|
|
||||
:param binary: The binary value. |
|
||||
:param mimetype: The mimetype of the binary value. |
|
||||
:param filename: The filename of the binary value. |
|
||||
:param export: The output format (binary, file, base64). |
|
||||
:param format: Specify the output format for the document. |
|
||||
:param page: Specifies the page if the file has several pages, e.g. if it is a PDF file. |
|
||||
:param frame: Specifies the frame if the file has several frames, e.g. if it is a video file. |
|
||||
:param animation: In this case, the parameter frame specifies the number of frames. |
|
||||
:param video_resize: Specify to resize the output image. |
|
||||
:param image_resize: Specify to resize the output image. |
|
||||
:param image_crop: Specify to crop the output image. |
|
||||
:return: Returns the output depending on the given format. |
|
||||
:raises ValueError: The file extension could not be determined or the format is invalid. |
|
||||
""" |
|
||||
extension = utils_os.get_extension(binary, filename, mimetype) |
|
||||
if not extension: |
|
||||
raise ValueError("The file extension could not be determined.") |
|
||||
if format not in FORMATS: |
|
||||
raise ValueError("Invalid export format.") |
|
||||
if extension not in (VIDEO_IMPORTS + PDF_IMPORTS + WAND_IMPORTS + unoconv.IMPORTS): |
|
||||
raise ValueError("Invalid import format.") |
|
||||
if not imageio or not Image or not VideoFileClip: |
|
||||
raise ValueError("Some libraries couldn't be imported.") |
|
||||
image_data = None |
|
||||
image_extension = extension |
|
||||
if extension in WAND_IMPORTS: |
|
||||
image_data = binary |
|
||||
elif not image_data and (extension in PDF_IMPORTS or extension in unoconv.IMPORTS): |
|
||||
pdf_data = binary if extension in PDF_IMPORTS else None |
|
||||
if not pdf_data: |
|
||||
image_extension = "pdf" |
|
||||
pdf_data = unoconv.convert_binary(binary, mimetype, filename) |
|
||||
reader = PyPDF2.PdfFileReader(io.BytesIO(pdf_data)) |
|
||||
writer = PyPDF2.PdfFileWriter() |
|
||||
if reader.getNumPages() >= page: |
|
||||
writer.addPage(reader.getPage(page)) |
|
||||
else: |
|
||||
writer.addPage(reader.getPage(0)) |
|
||||
pdf_bytes = io.BytesIO() |
|
||||
writer.write(pdf_bytes) |
|
||||
image_data = pdf_bytes.getvalue() |
|
||||
if image_data: |
|
||||
with Image(blob=image_data, format=image_extension) as thumbnail: |
|
||||
thumbnail.format = format |
|
||||
if image_extension == "pdf": |
|
||||
thumbnail.background_color = Color('white') |
|
||||
thumbnail.alpha_channel = 'remove' |
|
||||
if image_resize: |
|
||||
thumbnail.transform(resize=image_resize) |
|
||||
if image_crop: |
|
||||
thumbnail.transform(crop=image_crop) |
|
||||
if export == 'file': |
|
||||
return io.BytesIO(thumbnail.make_blob()) |
|
||||
elif export == 'base64': |
|
||||
return base64.b64encode(thumbnail.make_blob()) |
|
||||
else: |
|
||||
return thumbnail.make_blob() |
|
||||
elif extension in VIDEO_IMPORTS: |
|
||||
tmp_dir = tempfile.mkdtemp() |
|
||||
try: |
|
||||
tmp_wpath = os.path.join(tmp_dir, "tmpfile.%s" % extension) |
|
||||
if os.name == 'nt': |
|
||||
tmp_wpath = tmp_wpath.replace("\\", "/") |
|
||||
with closing(open(tmp_wpath, 'wb')) as file: |
|
||||
file.write(binary) |
|
||||
clip = VideoFileClip(tmp_wpath) |
|
||||
try: |
|
||||
tmp_opath = os.path.join(tmp_dir, "output.%s" % format) |
|
||||
clip.resize(**video_resize) |
|
||||
if animation: |
|
||||
files = [] |
|
||||
current_frame = 0 |
|
||||
while clip.duration > current_frame and current_frame < frame: |
|
||||
filename = os.path.join(tmp_dir, "output_%s.png" % frame) |
|
||||
clip.save_frame(filename, t=frame) |
|
||||
files.append(filename) |
|
||||
frame += 0.25 |
|
||||
tmp_opath = os.path.join(tmp_dir, "output.gif") |
|
||||
with imageio.get_writer(tmp_opath, fps=5, mode='I') as writer: |
|
||||
for filename in files: |
|
||||
image = imageio.imread(filename) |
|
||||
writer.append_data(image) |
|
||||
elif clip.duration > int(frame): |
|
||||
clip.save_frame(tmp_opath, t=int(frame)) |
|
||||
else: |
|
||||
clip.save_frame(tmp_opath, t=int(0)) |
|
||||
if os.path.isfile(tmp_opath): |
|
||||
with open(tmp_opath, 'rb') as file: |
|
||||
if export == 'file': |
|
||||
return io.BytesIO(file.read()) |
|
||||
elif export == 'base64': |
|
||||
return base64.b64encode(file.read()) |
|
||||
else: |
|
||||
return file.read() |
|
||||
else: |
|
||||
raise ValueError("No output could be created from the video.") |
|
||||
finally: |
|
||||
try: |
|
||||
clip.reader.close() |
|
||||
del clip.reader |
|
||||
if clip.audio != None: |
|
||||
clip.audio.reader.close_proc() |
|
||||
del clip.audio |
|
||||
del clip |
|
||||
except Exception as e: |
|
||||
sys.exc_clear() |
|
||||
finally: |
|
||||
try: |
|
||||
shutil.rmtree(tmp_dir) |
|
||||
except PermissionError: |
|
||||
_logger.warn("Temporary directory could not be deleted.") |
|
||||
else: |
|
||||
raise ValueError("No output could be generated.") |
|
||||
|
################################################################################### |
||||
|
# |
||||
|
# Copyright (C) 2018 MuK IT GmbH |
||||
|
# |
||||
|
# This program is free software: you can redistribute it and/or modify |
||||
|
# it under the terms of the GNU Affero General Public License as |
||||
|
# published by the Free Software Foundation, either version 3 of the |
||||
|
# License, or (at your option) any later version. |
||||
|
# |
||||
|
# This program is distributed in the hope that it will be useful, |
||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
|
# GNU Affero General Public License for more details. |
||||
|
# |
||||
|
# You should have received a copy of the GNU Affero General Public License |
||||
|
# along with this program. If not, see <http://www.gnu.org/licenses/>. |
||||
|
# |
||||
|
################################################################################### |
||||
|
|
||||
|
import os |
||||
|
import io |
||||
|
import sys |
||||
|
import PyPDF2 |
||||
|
import base64 |
||||
|
import shutil |
||||
|
import urllib |
||||
|
import logging |
||||
|
import tempfile |
||||
|
import mimetypes |
||||
|
|
||||
|
from contextlib import closing |
||||
|
|
||||
|
from odoo.tools import config |
||||
|
from odoo.tools.mimetypes import guess_mimetype |
||||
|
|
||||
|
from odoo.addons.muk_utils.tools import utils_os |
||||
|
from odoo.addons.muk_converter.service import unoconv |
||||
|
|
||||
|
_logger = logging.getLogger(__name__) |
||||
|
|
||||
|
try: |
||||
|
from wand.image import Image |
||||
|
from wand.color import Color |
||||
|
except ImportError: |
||||
|
Image = False |
||||
|
Color = False |
||||
|
_logger.warn('Cannot `import wand`.') |
||||
|
|
||||
|
try: |
||||
|
import imageio |
||||
|
except ImportError: |
||||
|
imageio = False |
||||
|
_logger.warn('Cannot `import imageio`.') |
||||
|
|
||||
|
try: |
||||
|
from moviepy.editor import VideoFileClip |
||||
|
except ImportError: |
||||
|
VideoFileClip = False |
||||
|
_logger.warn('Cannot `import moviepy`.') |
||||
|
|
||||
|
FORMATS = [ |
||||
|
"png", "jpg", "jepg" |
||||
|
] |
||||
|
|
||||
|
VIDEO_IMPORTS = [ |
||||
|
"mp4", "mov", "wav", "avi", "mpg", "flv", "wmv", "webm" |
||||
|
] |
||||
|
|
||||
|
PDF_IMPORTS = [ |
||||
|
"pdf" |
||||
|
] |
||||
|
|
||||
|
WAND_IMPORTS = [ |
||||
|
"aai", "art", "arw", "avi", "avs", "bpg", "brf", "cals", "cgm", "cin", "cip", "cmyk", "cmyka", "svg", |
||||
|
"cr2", "crw", "cur", "cut", "dcm", "dcr", "dcx", "dds", "dib", "djvu", "dng", "dot", "dpx", "tim", |
||||
|
"emf", "epdf", "epi", "eps", "eps2", "eps3", "epsf", "epsi", "ept", "exr", "fax", "fig", "fits", |
||||
|
"fpx", "gplt", "gray", "graya", "hdr", "hdr", "heic", "hpgl", "hrz", "html", "ico", "info", "ttf", |
||||
|
"inline", "isobrl", "isobrl6", "jbig", "jng", "jp2", "jpt", "j2c", "j2k", "jxr", "json", "man", "bmp", |
||||
|
"mat", "miff", "mono", "mng", "m2v", "mpeg", "mpc", "mpr", "mrw", "msl", "mtv", "mvg", "nef", "yuv", |
||||
|
"orf", "otb", "p7", "palm", "pam", "clipboard", "pbm", "pcd", "pcds", "pcl", "pcx", "pdb", "jpe", |
||||
|
"pef", "pes", "pfa", "pfb", "pfm", "pgm", "picon", "pict", "pix", "png8", "png00", "png24", "tiff", |
||||
|
"png32", "png48", "png64", "pnm", "ppm", "ps", "ps2", "ps3", "psb", "psd", "ptif", "pwp", "rad", |
||||
|
"raf", "rgb", "rgba", "rgf", "rla", "rle", "sct", "sfw", "sgi", "shtml", "sid", " mrsid", "jpeg", |
||||
|
"sparse-color", "sun", "tga", "ubrl", "ubrl6", "uyvy", "vicar", "viff", "wbmp", "jpg", "png", "uil", |
||||
|
"wdp", "webp", "wmf", "wpg", "x", "xbm", "xcf", "xpm", "xwd", "x3f", "ycbcr", "ycbcra", "bmp3", "bmp2", |
||||
|
] |
||||
|
|
||||
|
def formats(): |
||||
|
return FORMATS |
||||
|
|
||||
|
def imports(): |
||||
|
return VIDEO_IMPORTS + PDF_IMPORTS + WAND_IMPORTS + unoconv.UNOCONV_IMPORTS |
||||
|
|
||||
|
def create_thumbnail(binary, mimetype=None, filename=None, export="binary", format="png", page=0, frame=0, |
||||
|
animation=False, video_resize={'width': 256}, image_resize='256x256>', image_crop=None): |
||||
|
""" |
||||
|
Converts a thumbnail for a given file. |
||||
|
|
||||
|
:param binary: The binary value. |
||||
|
:param mimetype: The mimetype of the binary value. |
||||
|
:param filename: The filename of the binary value. |
||||
|
:param export: The output format (binary, file, base64). |
||||
|
:param format: Specify the output format for the document. |
||||
|
:param page: Specifies the page if the file has several pages, e.g. if it is a PDF file. |
||||
|
:param frame: Specifies the frame if the file has several frames, e.g. if it is a video file. |
||||
|
:param animation: In this case, the parameter frame specifies the number of frames. |
||||
|
:param video_resize: Specify to resize the output image. |
||||
|
:param image_resize: Specify to resize the output image. |
||||
|
:param image_crop: Specify to crop the output image. |
||||
|
:return: Returns the output depending on the given format. |
||||
|
:raises ValueError: The file extension could not be determined or the format is invalid. |
||||
|
""" |
||||
|
extension = utils_os.get_extension(binary, filename, mimetype) |
||||
|
if not extension: |
||||
|
raise ValueError("The file extension could not be determined.") |
||||
|
if format not in FORMATS: |
||||
|
raise ValueError("Invalid export format.") |
||||
|
if extension not in (VIDEO_IMPORTS + PDF_IMPORTS + WAND_IMPORTS + unoconv.UNOCONV_IMPORTS): |
||||
|
raise ValueError("Invalid import format.") |
||||
|
if not imageio or not Image or not VideoFileClip: |
||||
|
raise ValueError("Some libraries couldn't be imported.") |
||||
|
image_data = None |
||||
|
image_extension = extension |
||||
|
if extension in WAND_IMPORTS: |
||||
|
image_data = binary |
||||
|
elif not image_data and (extension in PDF_IMPORTS or extension in unoconv.UNOCONV_IMPORTS): |
||||
|
pdf_data = binary if extension in PDF_IMPORTS else None |
||||
|
if not pdf_data: |
||||
|
image_extension = "pdf" |
||||
|
pdf_data = unoconv.unoconv.convert(binary, mimetype, filename) |
||||
|
reader = PyPDF2.PdfFileReader(io.BytesIO(pdf_data)) |
||||
|
writer = PyPDF2.PdfFileWriter() |
||||
|
if reader.getNumPages() >= page: |
||||
|
writer.addPage(reader.getPage(page)) |
||||
|
else: |
||||
|
writer.addPage(reader.getPage(0)) |
||||
|
pdf_bytes = io.BytesIO() |
||||
|
writer.write(pdf_bytes) |
||||
|
image_data = pdf_bytes.getvalue() |
||||
|
if image_data: |
||||
|
with Image(blob=image_data, format=image_extension) as thumbnail: |
||||
|
thumbnail.format = format |
||||
|
if image_extension == "pdf": |
||||
|
thumbnail.background_color = Color('white') |
||||
|
thumbnail.alpha_channel = 'remove' |
||||
|
if image_resize: |
||||
|
thumbnail.transform(resize=image_resize) |
||||
|
if image_crop: |
||||
|
thumbnail.transform(crop=image_crop) |
||||
|
if export == 'file': |
||||
|
return io.BytesIO(thumbnail.make_blob()) |
||||
|
elif export == 'base64': |
||||
|
return base64.b64encode(thumbnail.make_blob()) |
||||
|
else: |
||||
|
return thumbnail.make_blob() |
||||
|
elif extension in VIDEO_IMPORTS: |
||||
|
tmp_dir = tempfile.mkdtemp() |
||||
|
try: |
||||
|
tmp_wpath = os.path.join(tmp_dir, "tmpfile.%s" % extension) |
||||
|
if os.name == 'nt': |
||||
|
tmp_wpath = tmp_wpath.replace("\\", "/") |
||||
|
with closing(open(tmp_wpath, 'wb')) as file: |
||||
|
file.write(binary) |
||||
|
clip = VideoFileClip(tmp_wpath) |
||||
|
try: |
||||
|
tmp_opath = os.path.join(tmp_dir, "output.%s" % format) |
||||
|
clip.resize(**video_resize) |
||||
|
if animation: |
||||
|
files = [] |
||||
|
current_frame = 0 |
||||
|
while clip.duration > current_frame and current_frame < frame: |
||||
|
filename = os.path.join(tmp_dir, "output_%s.png" % frame) |
||||
|
clip.save_frame(filename, t=frame) |
||||
|
files.append(filename) |
||||
|
frame += 0.25 |
||||
|
tmp_opath = os.path.join(tmp_dir, "output.gif") |
||||
|
with imageio.get_writer(tmp_opath, fps=5, mode='I') as writer: |
||||
|
for filename in files: |
||||
|
image = imageio.imread(filename) |
||||
|
writer.append_data(image) |
||||
|
elif clip.duration > int(frame): |
||||
|
clip.save_frame(tmp_opath, t=int(frame)) |
||||
|
else: |
||||
|
clip.save_frame(tmp_opath, t=int(0)) |
||||
|
if os.path.isfile(tmp_opath): |
||||
|
with open(tmp_opath, 'rb') as file: |
||||
|
if export == 'file': |
||||
|
return io.BytesIO(file.read()) |
||||
|
elif export == 'base64': |
||||
|
return base64.b64encode(file.read()) |
||||
|
else: |
||||
|
return file.read() |
||||
|
else: |
||||
|
raise ValueError("No output could be created from the video.") |
||||
|
finally: |
||||
|
try: |
||||
|
clip.reader.close() |
||||
|
del clip.reader |
||||
|
if clip.audio != None: |
||||
|
clip.audio.reader.close_proc() |
||||
|
del clip.audio |
||||
|
del clip |
||||
|
except Exception as e: |
||||
|
sys.exc_clear() |
||||
|
finally: |
||||
|
try: |
||||
|
shutil.rmtree(tmp_dir) |
||||
|
except PermissionError: |
||||
|
_logger.warn("Temporary directory could not be deleted.") |
||||
|
else: |
||||
|
raise ValueError("No output could be generated.") |
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue