gfw-api/gfw-analysis-gee

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import datetime

from flask import jsonify, request

from gfwanalysis.utils.landcover_lookup import get_landcover_types


# GENERIC Error


def error(status=500, detail='Generic Error'):
    error = {
        'status': status,
        'detail': detail
    }
    return jsonify(errors=[error]), status


def set_params():
    """-"""
    threshold = int(request.args.get('thresh', 30))
    begin = request.args.get('begin', '2000-01-01')
    end = request.args.get('end', '2019-12-31')
    period = request.args.get('period', None)
    if request.args.get('table', 'false').lower() == 'false':
        table = False
    else:
        table = True

    if period and len(period.split(',')) > 1:
        first = period.split(',')[0]
        second = period.split(',')[1]
        try:
            if len(first.split('-')) > 2 and len(second.split('-')) > 2:
                datetime.datetime(year=int(first.split('-')[0]), month=int(first.split('-')[1]),
                                  day=int(first.split('-')[2]))
                datetime.datetime(year=int(second.split('-')[0]), month=int(second.split('-')[1]),
                                  day=int(second.split('-')[2]))
                begin = first
                end = second
            else:
                pass
        except Exception:
            pass
    else:
        pass
    return threshold, begin, end, table


def get_layer():
    valid_layers = get_landcover_types()

    layer = request.args.get('layer', 'globcover').lower()

    if layer in valid_layers:
        return layer
    else:
        return None


def return_pixel_count():
    if request.method == 'GET':
        pixel_count = request.args.get('pixel_count', None)
    else:
        pixel_count = request.get_json().get('pixel_count', None) if request.get_json() else None

    if pixel_count and pixel_count.lower() == 'true':
        pixel_count = True
    else:
        pixel_count = False

    return pixel_count