devicehub-teal/ereuse_devicehub/resources/versions/versions.py

import flask
import json
import requests
import teal.marshmallow
import pkg_resources

from typing import Callable, Iterable, Tuple
from urllib.parse import urlparse
from flask import make_response, g
from teal.resource import Resource, View

from ereuse_devicehub.resources.inventory.model import Inventory


def get_tag_version():
    """Get version of microservice ereuse-tag."""
    path = "/versions/version/"
    url = urlparse(Inventory.current.tag_provider.to_text())._replace(path=path)
    res = requests.get(url.geturl())
    if res.status_code == 200:
        return json.loads(res.content)
    else:
        return {}

class VersionView(View):
    def get(self, *args, **kwargs):
        """Get version of DeviceHub and ereuse-tag."""
        dh_version = pkg_resources.require('ereuse-devicehub')[0].version
        tag_version = get_tag_version()
        versions = {'devicehub': dh_version}
        versions.update(tag_version)
        return json.dumps(versions)


class VersionDef(Resource):
    __type__ = 'Version'
    SCHEMA = None
    VIEW = None  # We do not want to create default / documents endpoint
    AUTH = False

    def __init__(self, app,
                 import_name=__name__,
                 static_folder='static',
                 static_url_path=None,
                 template_folder='templates',
                 url_prefix=None,
                 subdomain=None,
                 url_defaults=None,
                 root_path=None,
                 cli_commands: Iterable[Tuple[Callable, str or None]] = tuple()):
        super().__init__(app, import_name, static_folder, static_url_path, template_folder,
                         url_prefix, subdomain, url_defaults, root_path, cli_commands)

        d = {'devicehub': '0.1.0a'}
        get = {'GET'}

        version_view = VersionView.as_view('VersionView', definition=self)
        self.add_url_rule('/', defaults=d, view_func=version_view, methods=get)