#! -*- encoding: utf-8 -*- """ Config file utilities. This module has an :attr:`config_filename`, which can be used to set the filename outside of function calls: >>> from leap.mx.util import config >>> config.config_filename = "blahblah.yaml" If not set anywhere, it will default to using the top level repository directory, i.e. "/.../leap_mx/leap_mx.conf", and will create that file with the default settings if it does not exist. The config file can be loaded/created with :func:`config.loadConfig`: >>> config.loadConfig() Once the config file is loaded, this module presents a highly object-oriented interface, so that sections taken from the config file become attribute of this module, and the name of their respective settings become attributes of the section names. Like this: >>> print config.basic.postfix_port 465 @authors: Isis Lovecruft, 0x2cdb8b35 @version: 0.0.1 @license: see included LICENSE file """ from os import path as ospath import sys import yaml from leap.mx.util import version, storage from leap.mx.exceptions import MissingConfig, UnsupportedOS filename = None basic = storage.Storage() advanced = storage.Storage() PLATFORMS = {'LINUX': sys.platform.startswith("linux"), 'OPENBSD': sys.platform.startswith("openbsd"), 'FREEBSD': sys.platform.startswith("freebsd"), 'NETBSD': sys.platform.startswith("netbsd"), 'DARWIN': sys.platform.startswith("darwin"), 'SOLARIS': sys.platform.startswith("sunos"), 'WINDOWS': sys.platform.startswith("win32")} def getClientPlatform(platform_name=None): """ Determine the client's operating system platform. Optionally, if :param:`platform_name` is given, check that this is indeed the platform we're operating on. @param platform_name: A string, upper-, lower-, or mixed case, of one of the keys in the :attr:`leap.util.version.PLATFORMS` dictionary. E.g. 'Linux' or 'OPENBSD', etc. @returns: A string specifying the platform name, and the boolean test used to determine it. """ for name, test in PLATFORMS.items(): if not platform_name or platform_name.upper() == name: if test: return name, test def _create_config_file(conffile): """ xxx fill me in """ with open(conffile, 'w+') as conf: conf.write(""" # # mx.conf # ======= # Configurable options for the leap_mx encrypting mail exchange. # # This file follows YAML markup format: http://yaml.org/spec/1.2/spec.html # Keep in mind that indentation matters. # basic: # Where is the spoolfile of messages to encrypt?: spoolfile: /var/mail/encrypt_me advanced: # Which port on localhost should postfix send check_recipient queries to?: check_recipient_access_port: 1347 # Which port on localhost should postfix ask for UUIDs?: virtual_alias_map_port: 1348 # Enable debugging output in the logger: debug: true # Print enough things really fast to make you look super 1337: noisy: false """) conf.flush() def _get_config_filename(filename=None, use_dot_config_directory=False): """ Get the full path and filename of the config file. """ platform = version.getClientPlatform()[0] resource = version.name ## Oh hell, it could be said only to beguile: ## That windoze users are capable of editing a .conf file. ## Also, what maddened wingnut would be so fool ## To run a mail exchange on a windoze nodule? ## I'm ignoring these loons for now. And pardon if I seem jaded, ## But srsly, this and that solaris sh*t should be deprecated. if not platform.endswith('LINUX') and not platform.endswith('BSD'): raise UnsupportedOS("Sorry, your operating system isn't supported.") ## If not given, default to the application's name + '.conf' if not filename: filename = resource + ".conf" where = None if not use_dot_config_directory: repo_dir = version.getRepoDir() where = os.path.abspath(repo_dir) ## Use ~/.config/ instead: else: dot_config_dirs = BaseDirectory.xdg_config_dirs for dir in dot_config_dirs: our_dir = os.path.join(dir, resource) if os.path.isdir(our_dir): if filename in os.listdir(our_dir): where = os.path.abspath(our_dir) if not where: where = BaseDirectory.save_config_path(resource) conffile = os.path.join(where, filename) try: with open(conffile) as cf: pass except IOError: conffile = _create_config_file(conffile) finally: return conffile def loadConfig(filename=config_filename): """ Some of this is taken from OONI config code for now, and so this should be refacotored, along with the leap_client config code, so that we have similarly structured config files. It is perhaps desirable to also use soledad as a backend for remote setup and maintainance, and thus this code will need to hook into u1db (and potentially "pysqlcipher"). Excuse the yaml for now, I just wanted something that works. @param filename: (optional) If provided, use this filename. """ if not filename: filename = _get_config_filename() if os.path.isfile(filename): with open(filename, 'a+') as conf: config_contents = '\n'.join(conf.readlines()) configuration = yaml.safe_load(config_contents) ## These become objects with their keys loaded as attributes: ## ## from leap.util import config ## config.basic.foo = bar ## try: for k, v in configuration['basic'].items(): basic[k] = v except AttributeError: pass try: for k, v in configuration['advanced'].items(): advanced[k] = v except AttributeError: pass return basic, advanced else: raise MissingConfig("Could not load config file.") ## This is the name of the config file to use: ## If not set, it defaults to 'leap_mx/leap_mx.conf' if not filename: filename = _get_config_location() else: filename = _get_config_location(config_filename=filename)