Compare commits
4 Commits
dac01a224b
...
2173f2984f
| Author | SHA1 | Date | |
|---|---|---|---|
| 2173f2984f | |||
| 5ecfb2a37b | |||
| fbe9799df7 | |||
| 7212b8eee4 |
59
config.py
59
config.py
@@ -31,21 +31,66 @@ import ast
|
||||
|
||||
import pywikibot
|
||||
|
||||
__config = configparser.ConfigParser(interpolation=None)
|
||||
__config.read(pywikibot.config.get_base_dir() + "/jogobot.conf")
|
||||
|
||||
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( __config )
|
||||
config = dict( raw_config )
|
||||
|
||||
# Parse all sections
|
||||
for section in __config.sections():
|
||||
for section in raw_config.sections():
|
||||
# Convert to dict as configparser could contain only strings
|
||||
config[section] = dict( __config[section] )
|
||||
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 )
|
||||
|
||||
# Make jogobot entrys available in root level (without sections)
|
||||
for key, value in config["jogobot"].items():
|
||||
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 )
|
||||
|
||||
@@ -30,6 +30,14 @@ from subprocess import Popen, PIPE, TimeoutExpired
|
||||
|
||||
import pywikibot
|
||||
|
||||
# Location of logging related stuff in pywikibot was changed with
|
||||
# commit d7d7a14 on Mon Sep 7 14:41:43 2015
|
||||
# This is to be compatible with recent versions as well as 2.0-branch
|
||||
try:
|
||||
from pywikibot.logging import(
|
||||
DEBUG, INFO, WARNING, ERROR, CRITICAL, STDOUT, VERBOSE, logoutput )
|
||||
except( ImportError ):
|
||||
# Legacy support (pywikibot 2.0 branch)
|
||||
from pywikibot.bot import(
|
||||
DEBUG, INFO, WARNING, ERROR, CRITICAL, STDOUT, VERBOSE, logoutput )
|
||||
|
||||
|
||||
Reference in New Issue
Block a user