52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
class ConfigWriter:
|
|
def __init__(self):
|
|
self._config_file = 'C:\\Program Files\\EqualizerAPO\\config\\config.txt'
|
|
self._config_output = None
|
|
|
|
self.headphone_mode_enabled = False
|
|
self.active_headphone = None
|
|
self.inactive_headphones = list()
|
|
|
|
# Read current config.
|
|
with open(self._config_file, 'r') as c:
|
|
self._config_output = c.readlines()
|
|
c.close()
|
|
|
|
# Extract spec name from text output.
|
|
for spec in self._config_output:
|
|
if spec.startswith('# '):
|
|
_spec = spec.replace('# ', '').removesuffix('\n').removesuffix('.txt').removeprefix('Include:')
|
|
self.inactive_headphones.append(_spec.strip())
|
|
else:
|
|
_spec = spec.removesuffix('\n').removesuffix('.txt').removeprefix('Include:')
|
|
self.active_headphone = _spec.strip()
|
|
|
|
if self.active_headphone is not None:
|
|
self.headphone_mode_enabled = True
|
|
|
|
|
|
def write_specs(self, disabled_specs, enabled_spec):
|
|
formatted_disabled_specs = list()
|
|
formatted_enabled_spec = self._format_spec(enabled_spec, False)
|
|
|
|
if disabled_specs is not None:
|
|
for s in disabled_specs:
|
|
formatted_disabled_specs.append(self._format_spec(s, True))
|
|
|
|
with open(self._config_file, 'w+') as c:
|
|
for i in formatted_disabled_specs:
|
|
c.write(i + '\n')
|
|
|
|
if enabled_spec is not None:
|
|
c.write(formatted_enabled_spec)
|
|
|
|
c.close()
|
|
|
|
def _format_spec(self, entry: str, disable: bool):
|
|
_entry = None
|
|
if disable:
|
|
_entry = '# Include: {}.txt'.format(entry)
|
|
else:
|
|
_entry = 'Include: {}.txt'.format(entry)
|
|
|
|
return _entry |