blob: d32c2a1597df7f74697955560278819fc7c7f5ee [file] [log] [blame]
#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you 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
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
''' '''
class Config(object):
Responsible for reading the yaml config files and
exposing state locations through python methods.
A state location is represented by following dictionary:
type: zookeeper (or file),
name: unique_name_of_the_location,
hostport: if_hosted_on_remote_host:port,
rootpath: /root/path/for/states
def __init__(self):
self.location = None
# pylint: disable=attribute-defined-outside-init
def set_state_locations(self, state_locations):
""" set state locations """
self.locations = state_locations
def validate_state_locations(self):
Names of all state locations must be unique.
names = [loc["name"] for loc in self.locations]
assert len(names) == len(set(names)), "Names of state locations must be unique"
def get_all_state_locations(self):
Returns all the state locations.
return self.locations
def get_state_locations_of_type(self, location_type):
Return a list of state locations of a particular type.
return [location for location in self.locations if location['type'] == location_type]