You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
2.6 KiB
96 lines
2.6 KiB
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# config.py
|
|
#
|
|
# Copyright 2017 Jonathan Golder <jonathan@golderweb.de>
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
# MA 02110-1301, USA.
|
|
#
|
|
#
|
|
"""
|
|
This module will read config for jogobot from config file (ini-style) using
|
|
python standard library module configparser
|
|
"""
|
|
|
|
import configparser
|
|
import ast
|
|
|
|
import pywikibot
|
|
|
|
|
|
def load_config():
|
|
"""
|
|
Reads raw configuration from file and provides it as attribute
|
|
|
|
@param Config-Object
|
|
@r-type configparser.ConfigParser
|
|
"""
|
|
# Load configparser
|
|
config = configparser.ConfigParser(interpolation=None)
|
|
# Load config file
|
|
config.read(pywikibot.config.get_base_dir() + "/jogobot.conf")
|
|
|
|
return config
|
|
|
|
|
|
def parse_config( raw_config ):
|
|
"""
|
|
Converts config to normal dictionary
|
|
Parses each entry with ast.literal_eval
|
|
|
|
@param config Config-Object to parse
|
|
@type config configparser.ConfigParser
|
|
|
|
@return Parsed configuration
|
|
@rtype dict
|
|
"""
|
|
# Convert to dict as configparser could contain only strings
|
|
config = dict( raw_config )
|
|
|
|
# Parse all sections
|
|
for section in raw_config.sections():
|
|
# Convert to dict as configparser could contain only strings
|
|
config[section] = dict( raw_config[section] )
|
|
|
|
# Parse config with ast.literal_eval to get python datatypes
|
|
for key, value in config[section].items():
|
|
config[section][key] = ast.literal_eval( value )
|
|
|
|
return config
|
|
|
|
|
|
def root_section( config, section="jogobot" ):
|
|
"""
|
|
Make 'section' entrys available in root level (without sections)
|
|
|
|
@param config Config-Object to parse
|
|
@type config dict
|
|
|
|
@return Parsed configuration
|
|
@rtype dict
|
|
"""
|
|
for key, value in config[section].items():
|
|
config[key] = value
|
|
|
|
return config
|
|
|
|
# Load config
|
|
config = load_config()
|
|
# Parse config to get python datatypes
|
|
config = parse_config( config )
|
|
# Make jogobot section available as root
|
|
config = root_section( config )
|
|
|