sudacode/owocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More refactoring

Browse Source
This commit is contained in:
AuroraWright
2024-01-28 10:31:34 +01:00
parent 7c8cf92745
commit b7b87beaa0
3 changed files with 70 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
owocr/config.py
View File

@@ -3,10 +3,27 @@ import configparser
class Config:
has_config = False
general_config = {}
engine_config = {}
__general_config = {}
__engine_config = {}
__default_config = {
'read_from': 'clipboard',
'write_to': 'clipboard',
'engine': '',
'pause_at_startup': False,
'ignore_flag': False,
'delete_images': False,
'engines': [],
'logger_format': '<green>{time:HH:mm:ss.SSS}</green> | <level>{message}</level>',
'engine_color': 'cyan',
'delay_secs': 0.5,
'websocket_port': 7331,
'notifications': False,
'screen_capture_monitor': 1,
'screen_capture_coords': '',
'screen_capture_delay_secs': 3
}
def _parse(self, value):
def __parse(self, value):
value = value.strip()
if value.lower() == 'false':
return False
@@ -34,20 +51,23 @@ class Config:
for key in config:
if key == 'general':
for sub_key in config[key]:
self.general_config[sub_key.lower()] = self._parse(config[key][sub_key])
self.__general_config[sub_key.lower()] = self.__parse(config[key][sub_key])
elif key != 'DEFAULT':
self.engine_config[key.lower()] = {}
self.__engine_config[key.lower()] = {}
for sub_key in config[key]:
self.engine_config[key.lower()][sub_key.lower()] = self._parse(config[key][sub_key])
self.__engine_config[key.lower()][sub_key.lower()] = self.__parse(config[key][sub_key])
def get_general(self, value):
try:
return self.general_config[value]
return self.__general_config[value]
except KeyError:
return None
if value in self.__default_config:
return self.__default_config[value]
else:
return None
def get_engine(self, value):
try:
return self.engine_config[value]
return self.__engine_config[value]
except KeyError:
return None