Source code for livebridge.base.converters

# -*- coding: utf-8 -*-
#
# Copyright 2016 dpa-infocom GmbH
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import logging
import os.path
import uuid
import aiohttp


logger = logging.getLogger(__name__)


FILE_EXT = {
    "image/jpeg": ".jpg",
    "image/gif": ".gif",
    "image/png": ".png",
}


class BaseConverter(object):
    """Base class for converters.

    Concrete implementations have to implement public :func:`convert` method."""
    __module__ = "livebridge.base"

    source = ""
    target = ""

    def __init__(self):
        pass

    async def _download_image(self, data):
        basename = os.path.basename(data["media"])
        file_ext = os.path.splitext(basename)[1]
        filename = "{}-{}{}".format(
            str(uuid.uuid4())[:8],
            basename,
            FILE_EXT[data["mimetype"]] if not file_ext else ""
        )
        filepath = os.path.join("/tmp/", filename)
        async with aiohttp.ClientSession() as session:
            async with session.get(data["href"]) as resp:
                test = await resp.read()
                with open(filepath, "wb") as file:
                    file.write(test)
        return filepath

    async def remove_images(self, images):
        try:
            for filepath in images:
                os.remove(filepath)
                logger.info("Removed image {}".format(filepath))
        except Exception as exc:
            logger.error("Error when deleting image.")
            logger.exception(exc)

    async def convert(self, post):
        """Convert incoming content of the incoming raw source post to a string suitable to the \
           target as content.

           :param post: raw source post
           :type dictionary: dictionary
           :returns: :class:`livebridge.base.ConversionResult` - result of the conversion."""
        raise NotImplementedError("Not implemented in converter.")