mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-23 11:12:40 +01:00
103 lines
2.6 KiB
Python
103 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2015 Mike Fährmann
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 2 as
|
|
# published by the Free Software Foundation.
|
|
|
|
"""Global configuration module"""
|
|
|
|
import sys
|
|
import json
|
|
import os.path
|
|
import platform
|
|
|
|
# --------------------------------------------------------------------
|
|
# public interface
|
|
|
|
def load(*files, strict=False):
|
|
"""Load JSON configuration files"""
|
|
configfiles = files or _default_configs
|
|
for conf in configfiles:
|
|
try:
|
|
path = os.path.expanduser(conf)
|
|
with open(path) as file:
|
|
confdict = json.load(file)
|
|
_config.update(confdict)
|
|
except FileNotFoundError:
|
|
if strict:
|
|
raise
|
|
continue
|
|
except json.decoder.JSONDecodeError as exception:
|
|
print("Error while loading '", path, "':", sep="", file=sys.stderr)
|
|
print(exception, file=sys.stderr)
|
|
|
|
def clear():
|
|
"""Reset configuration to en empty state"""
|
|
globals()["_config"] = {}
|
|
|
|
def get(keys, default=None):
|
|
"""Get the value of property 'key' or a default-value if it doenst exist"""
|
|
conf = _config
|
|
try:
|
|
for k in keys:
|
|
conf = conf[k]
|
|
return conf
|
|
except (KeyError, AttributeError):
|
|
return default
|
|
|
|
def interpolate(keys, default=None):
|
|
"""Interpolate the value of 'key'"""
|
|
conf = _config
|
|
try:
|
|
for k in keys:
|
|
default = conf.get(keys[-1], default)
|
|
conf = conf[k]
|
|
return conf
|
|
except (KeyError, AttributeError):
|
|
return default
|
|
|
|
def set(keys, value):
|
|
"""Set the value of property 'key' for this session"""
|
|
conf = _config
|
|
for k in keys[:-1]:
|
|
try:
|
|
conf = conf[k]
|
|
except KeyError:
|
|
temp = {}
|
|
conf[k] = temp
|
|
conf = temp
|
|
conf[keys[-1]] = value
|
|
|
|
def setdefault(keys, value):
|
|
"""Set the value of property 'key' if it doesn't exist"""
|
|
conf = _config
|
|
for k in keys[:-1]:
|
|
try:
|
|
conf = conf[k]
|
|
except KeyError:
|
|
temp = {}
|
|
conf[k] = temp
|
|
conf = temp
|
|
return conf.setdefault(keys[-1], value)
|
|
|
|
|
|
# --------------------------------------------------------------------
|
|
# internals
|
|
|
|
_config = {}
|
|
|
|
if platform.system() == "Windows":
|
|
_default_configs = [
|
|
r"~\.config\gallery-dl.conf",
|
|
r"~\.gallery-dl.conf",
|
|
]
|
|
else:
|
|
_default_configs = [
|
|
"/etc/gallery-dl.conf",
|
|
"~/.config/gallery/config.json",
|
|
"~/.config/gallery-dl.conf",
|
|
"~/.gallery-dl.conf",
|
|
]
|