diff --git a/lib/nntp/tiny/config.py b/lib/nntp/tiny/config.py index 06179a4..31de94b 100644 --- a/lib/nntp/tiny/config.py +++ b/lib/nntp/tiny/config.py @@ -61,6 +61,12 @@ class Config(configparser.ConfigParser): return config + def section(self, section: str): + if not self.has_section(section): + raise ConfigSectionException(section) + + return self[section] + def get(self, section: str, option: str, *args, **kwargs): if not self.has_section(section): raise ConfigSectionException(section) diff --git a/lib/nntp/tiny/server.py b/lib/nntp/tiny/server.py index 6eb5398..38669de 100644 --- a/lib/nntp/tiny/server.py +++ b/lib/nntp/tiny/server.py @@ -22,7 +22,7 @@ class Server(): self.newsgroups = dict() self.sslctx = None - if config['listen'].get('tls', 'no') == 'yes': + if config.section('listen').get('tls', 'no') == 'yes': self.sslctx = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER) self.sslctx.load_cert_chain(config.get('tls', 'cert'), config.get('tls', 'key'))