35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
import os
|
|
import configobj
|
|
import szurubooru.errors
|
|
|
|
class Config(object):
|
|
''' INI config parser and container. '''
|
|
def __init__(self):
|
|
self.config = configobj.ConfigObj('../config.ini.dist')
|
|
if os.path.exists('../config.ini'):
|
|
self.config.merge(configobj.ConfigObj('../config.ini'))
|
|
self._validate()
|
|
|
|
def __getitem__(self, key):
|
|
return self.config[key]
|
|
|
|
def _validate(self):
|
|
'''
|
|
Checks whether config.ini doesn't contain errors that might prove
|
|
lethal at runtime.
|
|
'''
|
|
all_ranks = self['service']['user_ranks']
|
|
for privilege, rank in self['privileges'].items():
|
|
if rank not in all_ranks:
|
|
raise szurubooru.errors.ConfigError(
|
|
'Rank %r for privilege %r is missing from user_ranks' % (
|
|
rank, privilege))
|
|
for rank in ['anonymous', 'admin', 'nobody']:
|
|
if rank not in all_ranks:
|
|
raise szurubooru.errors.ConfigError(
|
|
'Fixed rank %r is missing from user_ranks' % rank)
|
|
if self['service']['default_user_rank'] not in all_ranks:
|
|
raise szurubooru.errors.ConfigError(
|
|
'Default rank %r is missing from user_ranks' % (
|
|
self['service']['default_user_rank']))
|