config.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from json import loads


class Config:
    def __init__(self, config, database_config):
        self.config = config
        self.database_config = database_config
        self.valid_relations = ['contains', 'contains_properly', 'covered_by', 'covers', 'crosses', 'disjoint', 'distance',
                                'distance_within', 'equals', 'hausdorff_distance', 'intersects', 'overlaps', 'touches', 'within']
        self.check_config()

    def check_config(self):
        if 'source' not in self.config:
            raise ConfigNotValidError("Config is missing source")
        else:
            if 'endpoint' not in self.config['source']:
                raise ConfigNotValidError("Config is missing source endpoint")

            if 'var' not in self.config['source']:
                raise ConfigNotValidError("Config is missing source var")
            else:
                if 'uri' not in self.config['source']['var']:
                    raise ConfigNotValidError("Variable for uri not specified in source")

                if 'shape' not in self.config['source']['var']:
                    raise ConfigNotValidError("Variable for shape not specified in source")

            if 'rawquery' not in self.config['source']:
                if 'graph' not in self.config['source']:
                    raise ConfigNotValidError("Config is missing source graph")

                if 'property' not in self.config['source']:
                    raise ConfigNotValidError("Config is missing source property")

            if 'offset' in self.config['source']:
                if not isinstance(self.config['source']['offset'], int):
                    raise ConfigNotValidError("Source offset is not an integer")

            if 'limit' in self.config['source']:
                if not isinstance(self.config['source']['limit'], int):
                    raise ConfigNotValidError("Source limit is not an integer")

            if 'chunksize' in self.config['source']:
                if not isinstance(self.config['source']['chunksize'], int):
                    raise ConfigNotValidError("Source chunksize is not an integer")

        if 'target' not in self.config:
            raise ConfigNotValidError("Config is missing target")
        else:
            if 'endpoint' not in self.config['target']:
                raise ConfigNotValidError("Config is missing target endpoint")

            if 'var' not in self.config['target']:
                raise ConfigNotValidError("Config is missing target var")
            else:
                if 'uri' not in self.config['target']['var']:
                    raise ConfigNotValidError("Variable for uri not specified in target")

                if 'shape' not in self.config['target']['var']:
                    raise ConfigNotValidError("Variable for shape not specified in target")

            if 'rawquery' not in self.config['target']:
                if 'graph' not in self.config['target']:
                    raise ConfigNotValidError("Config is missing target graph")

                if 'property' not in self.config['target']:
                    raise ConfigNotValidError("Config is missing target property")

            if 'offset' in self.config['target']:
                if not isinstance(self.config['target']['offset'], int):
                    raise ConfigNotValidError("Target offset is not an integer")

            if 'limit' in self.config['target']:
                if not isinstance(self.config['target']['limit'], int):
                    raise ConfigNotValidError("Target limit is not an integer")

            if 'chunksize' in self.config['target']:
                if not isinstance(self.config['target']['chunksize'], int):
                    raise ConfigNotValidError("Target chunksize is not an integer")

        if 'measure' not in self.config:
            raise ConfigNotValidError("Measure not specified")
        else:
            if 'relation' not in self.config['measure']:
                raise ConfigNotValidError("Relation not specified")
            else:
                if self.config['measure']['relation'] not in self.valid_relations:
                    raise ConfigNotValidError("Relation not valid. Only the following relations are valid: {}".format(self.valid_relations))
                elif self.config['measure']['relation'] == 'distance' or self.config['measure']['relation'] == 'distance_within' or self.config['measure']['relation'] == 'hausdorff_distance':
                    if 'threshold' not in self.config['measure']:
                        raise ConfigNotValidError("Config is missing measure threshold")

        # Check database config
        if 'database_name' not in self.database_config:
            raise ConfigNotValidError("Database name not specified")

        if 'database_user' not in self.database_config:
            raise ConfigNotValidError("Database user not specified")

        if 'database_password' not in self.database_config:
            raise ConfigNotValidError("Database password not specified")

    def get_chunksize(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        if 'chunksize' in self.config[type]:
            return self.config[type]['chunksize']
        else:
            return 1000

    def get_database_name(self):
        return self.database_config['database_name']

    def get_database_user(self):
        return self.database_config['database_user']

    def get_database_password(self):
        return self.database_config['database_password']

    def get_database_host(self):
        if 'database_host' in self.database_config:
            return self.database_config['database_host']
        else:
            return 'localhost'

    def get_database_port(self):
        if 'database_port' in self.database_config:
            return str(self.database_config['database_port'])
        else:
            return '5432'

    def get_database_string(self):
        return "host='{}' dbname='{}' user='{}' password='{}' port={}".format(self.get_database_host(), self.get_database_name(),
                                                                              self.get_database_user(), self.get_database_password(),
                                                                              self.get_database_port())

    def get_endpoint(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        return self.config[type]['endpoint']

    def get_geometry(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        return self.config[type]['geometry']

    def get_graph(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        return self.config[type]['graph']

    def get_limit(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        if 'limit' in self.config[type]:
            return self.config[type]['limit']
        else:
            return -1

    def get_relation(self):
        return self.config['measure']['relation']

    def get_offset(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        if 'offset' in self.config[type]:
            return self.config[type]['offset']
        else:
            return 0

    def get_output_format(self):
        if 'output_format' in self.config:
            return self.config['output_format']
        else:
            return None

    def get_prefixes(self):
        if 'prefixes' in self.config:
            return self.config['prefixes']
        else:
            return None

    def get_property(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        return self.config[type]['property']

    def get_rawquery(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        if 'rawquery' in self.config[type]:
            return self.config[type]['rawquery']
        else:
            return None

    def get_restriction(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        if 'restriction' in self.config[type]:
            return self.config[type]['restriction']
        else:
            return None

    def get_threshold(self):
        return self.config['measure']['threshold']

    def get_var_uri(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        return self.config[type]['var']['uri']

    def get_var_shape(self, type):
        if type != 'source' and type != 'target':
            raise Exception("Wrong type (not source or target) specified")

        return self.config[type]['var']['shape']


class ConfigNotValidError(Exception):
    def __init__(self, error):
        self.error = error

    def __str__(self):
        return(self.error)


def load_config(config_file_path):
    with open(config_file_path, 'r') as config_file:
        return loads(config_file.read())