claudecode/AvianVisitors
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

configparser: strip double quotes

Browse Source
This commit is contained in:
frederik
2024-01-30 10:37:55 +01:00
parent 3761d945a1
commit 53b738636e
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/utils/helpers.py
+13 -4
View File
@@ -1,4 +1,4 @@
import configparser from configparser import ConfigParser
import datetime import datetime
import glob import glob
import os import os
@@ -16,11 +16,20 @@ THISRUN = os.path.expanduser('~/BirdNET-Pi/scripts/thisrun.txt')
ANALYZING_NOW = os.path.expanduser('~/BirdNET-Pi/analyzing_now.txt') ANALYZING_NOW = os.path.expanduser('~/BirdNET-Pi/analyzing_now.txt')
class PHPConfigParser(ConfigParser):
def get(self, section, option, *, raw=False, vars=None, fallback=None):
value = super().get(section, option, raw=raw, vars=vars, fallback=fallback)
if raw:
return value
else:
return value.strip('"')
def _load_settings(settings_path='/etc/birdnet/birdnet.conf', force_reload=False): def _load_settings(settings_path='/etc/birdnet/birdnet.conf', force_reload=False):
global _settings global _settings
if _settings is None or force_reload: if _settings is None or force_reload:
with open(settings_path) as f: with open(settings_path) as f:
parser = configparser.ConfigParser() parser = PHPConfigParser()
# preserve case # preserve case
parser.optionxform = lambda option: option parser.optionxform = lambda option: option
lines = chain(("[top]",), f) lines = chain(("[top]",), f)
@@ -37,8 +46,8 @@ def get_settings(settings_path='/etc/birdnet/birdnet.conf', force_reload=False):
def write_settings(file_name=THISRUN): def write_settings(file_name=THISRUN):
settings = _load_settings() settings = _load_settings()
with open(file_name, 'w') as configfile: with open(file_name, 'w') as configfile:
for key, value in settings.items(): for key in settings.keys():
configfile.write(f'{key}={value}\n') configfile.write(f'{key}={settings.get(key, raw=True)}\n')
os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IWGRP) os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IWGRP)