99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
from configparser import RawConfigParser
|
|
from os import path
|
|
from ordigi import constants
|
|
from geopy.geocoders import options as gopt
|
|
|
|
|
|
class Config:
|
|
"""Manage config file"""
|
|
|
|
def __init__(self, conf_path=constants.CONFIG_FILE, conf={}):
|
|
self.conf_path = conf_path
|
|
if conf == {}:
|
|
self.conf = self.load_config()
|
|
if self.conf == {}:
|
|
# Fallback to default config
|
|
self.conf_path = constants.CONFIG_FILE
|
|
self.conf = self.load_config()
|
|
else:
|
|
self.conf = conf
|
|
|
|
def write(self, conf):
|
|
with open(self.conf_path, 'w') as conf_file:
|
|
conf.write(conf_file)
|
|
return True
|
|
|
|
return False
|
|
|
|
def load_config(self):
|
|
if not self.conf_path.exists():
|
|
return {}
|
|
|
|
conf = RawConfigParser()
|
|
conf.read(self.conf_path)
|
|
return conf
|
|
|
|
def get_option(self, option, section):
|
|
|
|
if section in self.conf and option in self.conf[section]:
|
|
return self.conf[section][option]
|
|
|
|
return False
|
|
|
|
def get_path_definition(self):
|
|
"""Returns a list of folder definitions.
|
|
|
|
Each element in the list represents a folder.
|
|
Fallback folders are supported and are nested lists.
|
|
|
|
:returns: string
|
|
"""
|
|
|
|
if 'Path' in self.conf:
|
|
if 'format' in self.conf['Path']:
|
|
return self.conf['Path']['format']
|
|
elif 'dirs_path' and 'name' in self.conf['Path']:
|
|
return self.conf['Path']['dirs_path'] + '/' + self.conf['Path']['name']
|
|
|
|
return constants.DEFAULT_PATH_FORMAT
|
|
|
|
def get_options(self):
|
|
"""Get config options
|
|
:returns: dict
|
|
"""
|
|
|
|
options = {}
|
|
geocoder = self.get_option('geocoder', 'Geolocation')
|
|
if geocoder and geocoder in ('Nominatim',):
|
|
options['geocoder'] = geocoder
|
|
else:
|
|
options['geocoder'] = constants.DEFAULT_GEOCODER
|
|
|
|
prefer_english_names = self.get_option('prefer_english_names', 'Geolocation')
|
|
if prefer_english_names:
|
|
options['prefer_english_names'] = bool(prefer_english_names)
|
|
else:
|
|
options['prefer_english_names'] = False
|
|
|
|
timeout = self.get_option('timeout', 'Geolocation')
|
|
if timeout:
|
|
options['timeout'] = timeout
|
|
else:
|
|
options['timeout'] = gopt.default_timeout
|
|
|
|
options['path_format'] = self.get_path_definition()
|
|
|
|
options['day_begins'] = 0
|
|
options['max_deep'] = None
|
|
if 'Path' in self.conf:
|
|
if 'day_begins' in self.conf['Path']:
|
|
options['day_begins'] = int(self.conf['Path']['day_begins'])
|
|
if 'max_deep' in self.conf['Path']:
|
|
options['max_deep'] = int(self.conf['Path']['max_deep'])
|
|
|
|
options['exclude'] = []
|
|
if 'Exclusions' in self.conf:
|
|
options['exclude'] = [value for key, value in self.conf.items('Exclusions')]
|
|
|
|
return options
|