Conatum
20697c4823
Add `meta.args` for command line argument access. Add command line argument support for shard number. Add shard count to config file. Add `meta.sharding` exposing shard properties. Add shard number to logging methods. Add shard number to data appid.
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import configparser as cfgp
|
|
|
|
from .args import args
|
|
|
|
|
|
class Conf:
|
|
def __init__(self, configfile, section_name="DEFAULT"):
|
|
self.configfile = configfile
|
|
|
|
self.config = cfgp.ConfigParser(
|
|
converters={
|
|
"intlist": self._getintlist,
|
|
"list": self._getlist,
|
|
}
|
|
)
|
|
self.config.read(configfile)
|
|
|
|
self.section_name = section_name if section_name in self.config else 'DEFAULT'
|
|
|
|
self.default = self.config["DEFAULT"]
|
|
self.section = self.config[self.section_name]
|
|
self.bot = self.section
|
|
|
|
# Config file recursion, read in configuration files specified in every "ALSO_READ" key.
|
|
more_to_read = self.section.getlist("ALSO_READ", [])
|
|
read = set()
|
|
while more_to_read:
|
|
to_read = more_to_read.pop(0)
|
|
read.add(to_read)
|
|
self.config.read(to_read)
|
|
new_paths = [path for path in self.section.getlist("ALSO_READ", [])
|
|
if path not in read and path not in more_to_read]
|
|
more_to_read.extend(new_paths)
|
|
|
|
global conf
|
|
conf = self
|
|
|
|
def __getitem__(self, key):
|
|
return self.section[key].strip()
|
|
|
|
def __getattr__(self, section):
|
|
return self.config[section]
|
|
|
|
def get(self, name, fallback=None):
|
|
result = self.section.get(name, fallback)
|
|
return result.strip() if result else result
|
|
|
|
def _getintlist(self, value):
|
|
return [int(item.strip()) for item in value.split(',')]
|
|
|
|
def _getlist(self, value):
|
|
return [item.strip() for item in value.split(',')]
|
|
|
|
def write(self):
|
|
with open(self.configfile, 'w') as conffile:
|
|
self.config.write(conffile)
|
|
|
|
|
|
conf = Conf(args.config)
|