Merge pull request #43 from claui/user-config

Add support for user-configurable settings

Author: claui

{{ cookiecutter.pypi_package_name }}/doc/sphinx/conf.py

@@ -50,6 +50,9 @@ def skip_module(app, what, name, obj, skip, options):
         {% if cookiecutter.include_executable == "y" -%}
         '{{ cookiecutter.python_package_name }}.__main__',
         '{{ cookiecutter.python_package_name }}.cli',
+        {% endif -%}
+        '{{ cookiecutter.python_package_name }}.config',
+        {% if cookiecutter.include_executable == "y" -%}
         '{{ cookiecutter.python_package_name }}.fire_workarounds',
         {% endif -%}
         '{{ cookiecutter.python_package_name }}.version',
@@ -68,6 +71,9 @@ def setup(sphinx):
     {% if cookiecutter.include_executable == "y" -%}
     '**/{{ cookiecutter.python_package_name }}/__main__/**',
     '**/{{ cookiecutter.python_package_name }}/cli/**',
+    {% endif -%}
+    '**/{{ cookiecutter.python_package_name }}/config/**',
+    {% if cookiecutter.include_executable == "y" -%}
     '**/{{ cookiecutter.python_package_name }}/fire_workarounds/**',
     {% endif -%}
     '**/{{ cookiecutter.python_package_name }}/version/**',

{{ cookiecutter.pypi_package_name }}/pyproject.toml

@@ -23,9 +23,7 @@ name = "{{ cookiecutter.python_package_name }}"
 version = "{{ cookiecutter.project_version }}"
 description = "{{ cookiecutter.project_description }}"
 readme = ["README.md", "USAGE.md"]
-authors = [
-    "{{ cookiecutter.author_full_name }} <{{ cookiecutter.author_email }}>",
-]
+authors = ["{{ cookiecutter.author_full_name }} <{{ cookiecutter.author_email }}>"]
 license = "{{ cookiecutter.project_license }}"
 # See https://pypi.org/pypi?%3Aaction=list_classifiers
 classifiers = [
@@ -43,6 +41,7 @@ python = ">={{ cookiecutter.python_version }}"
 colorama = "*"
 fire = "*"
 {%- endif %}
+pyxdg = "*"
 
 [tool.poetry.group.dev.dependencies]
 autopep8 = "*"

{{ cookiecutter.pypi_package_name }}/{{ cookiecutter.python_package_name }}/config.py

@@ -0,0 +1,94 @@
+"""User configuration management."""
+
+import json
+import os
+from typing import Any
+
+import xdg.BaseDirectory  # type: ignore
+
+from .logging import get_logger
+from .settings import PACKAGE_NAME
+
+logger = get_logger(__name__)
+
+
+def config_get(key: str, default: Any | None = None) -> Any | None:
+    """Reads a config from the first level of a JSON file by key.
+
+    :param key: the name of the top-level JSON property.
+
+    :param default:
+        a default value if the property is not present in the JSON
+        or if its value is `null`.
+        Defaults to `None`.
+
+    :return:
+        the value as a Python object.
+        Returns `None` if the JSON value is `null` or if `key` does
+        not exist as a property at the top-level.
+    """
+    path = os.path.join(config_dir(), f'{PACKAGE_NAME}.json')
+    if not os.path.exists(path):
+        logger.debug(
+            'File %s does not exist; interpreting %s == ', path, default
+        )
+        return default
+    with open(path, 'r', encoding='utf-8') as stream:
+        config_dict = json.load(stream)
+    return config_dict.get(key, default)
+
+
+def config_set(key: str, value: Any) -> None:
+    """Writes a key-value pair to the config JSON file.
+
+    Overwrites an existing property by the given key.
+
+    :param key: the name of the top-level JSON property to store.
+
+    :param value: the value as a Python object, or `None` for `null`.
+    """
+    path = os.path.join(config_dir(), f'{PACKAGE_NAME}.json')
+    if os.path.exists(path):
+        with open(path, 'r+', encoding='utf-8') as stream:
+            config_dict = json.load(stream) or {}
+            if value and value == config_dict.get(key, None):
+                logger.debug('Configuration unchanged for key %s', key)
+                return
+            config_dict[key] = value
+            logger.info('Writing new value for key %s: %s', key, value)
+            stream.seek(0)
+            stream.truncate(0)
+            json.dump(config_dict, stream)
+            os.fsync(stream)
+    else:
+        with open(path, 'x', encoding='utf-8') as stream:
+            logger.info(
+                'Creating configuration file %s with value %s: %s',
+                path,
+                key,
+                value,
+            )
+            json.dump({key: value}, stream)
+            os.fsync(stream)
+
+    logger.info('New value written')
+
+
+def config_dir(subdir: str | None = None) -> str:
+    """Returns the configuration directory for this package.
+
+    Respects the XDG settings. Also creates the directory if it does
+    not exist.
+
+    :param subdir: an optional subdirectory, relative to the
+                   configuration directory.
+
+    :return: the absolute path to the directory.
+    """
+    return str(
+        xdg.BaseDirectory.save_config_path(
+            os.path.join(PACKAGE_NAME, subdir)
+            if subdir
+            else PACKAGE_NAME
+        )
+    )

{{ cookiecutter.pypi_package_name }}/{{ cookiecutter.python_package_name }}/errors.py

@@ -2,5 +2,5 @@
 {% if cookiecutter.include_executable == "y" %}
 
 class CliError(Exception):
-    """An user-facing error message."""
+    """A user-facing error message."""
 {% endif -%}

{{ cookiecutter.pypi_package_name }}/{{ cookiecutter.python_package_name }}/settings.py

@@ -7,4 +7,6 @@
 PACKAGE_ROOT = Path(__file__).parent.absolute()
 PYPROJECT_TOML = PROJECT_ROOT / 'pyproject.toml'
 
+PACKAGE_NAME = '{{ cookiecutter.pypi_package_name }}'
+
 debugMode = bool(os.getenv('{{ cookiecutter.python_package_name | upper }}_DEBUG'))

{{ cookiecutter.pypi_package_name }}/{{ cookiecutter.python_package_name }}/{{ cookiecutter.first_module_name }}.py

@@ -12,7 +12,7 @@
 logger = get_logger(__name__)
 
 
-class {{ cookiecutter.first_module_name.capitalize() }}:
+class {{ cookiecutter.first_module_name.capitalize() }}:  # pylint: disable=too-few-public-methods
     """
     :param foobar:
         The Foobar to connect to.