conf.py

#!/usr/bin/env python3

# This file is part of Openplotter.
# Copyright (C) 2022 by sailoog <https://github.com/openplotter/openplotter-settings>
#                     e-sailing <https://github.com/e-sailing/openplotter>
# Openplotter is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# any later version.
# Openplotter is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Openplotter. If not, see <http://www.gnu.org/licenses/>.

import os, configparser
from .version import *

class Conf:
	def __init__(self):
		self.user = os.environ.get('USER')
		if self.user == 'root': 
			try:
				self.user = os.path.expanduser(os.environ["SUDO_USER"])
			except:
				import pwd
				self.user = pwd.getpwuid(int(os.environ["PKEXEC_UID"])).pw_name
		self.home = os.path.expanduser('~'+self.user)

		self.data_conf = configparser.ConfigParser()


		self.conf_folder = self.home+'/.openplotter'
		if not os.path.exists(self.conf_folder):
			print('creating configuration directory', self.conf_folder)
			os.mkdir(self.conf_folder)

		self.conf_file = self.conf_folder+'/openplotter.conf'
		if not os.path.exists(self.conf_file):
			print('openplotter.conf not found, creating', self.conf_file)
			fo = open(self.conf_file, "w")
			fo.write( '[GENERAL]\nlang = en\nplay = /usr/share/sounds/openplotter/Store_Door_Chime.mp3\n')
			fo.close()
		self.read()

	def read(self):
		self.data_conf.read(self.conf_file)

	def write(self):
		with open(self.conf_file, 'w') as configfile:
			self.data_conf.write(configfile)

	def get(self, section, item):
		self.read()
		write = False
		if not self.data_conf.has_section(section):
			self.data_conf.add_section(section)
			write = True
		if not self.data_conf.has_option(section,item):
			self.data_conf.set(section, item, '')
			write = True
		if write: self.write()
		return self.data_conf.get(section, item)

	def set(self, section, item, value):
		self.read()
		if not self.data_conf.has_section(section):
			self.data_conf.add_section(section)
		self.data_conf.set(section, item, value)
		self.write()