promtext-cli 0.1.2.dev96__py3-none-any.whl → 0.1.2.dev98__py3-none-any.whl

promtext_cli/main.py

@@ -1,152 +1,12 @@
-"""promtext_cli is providing a CLI to cleanly update prometheus textfiles from scripts"""
+"""This module takes care of entrypoints"""
 
-# pylint: disable=C0116,R0914,R0912,R0915
-# this rules will be fixed by a object-oriented refactoring
+from .promtext import Promtext
 
-import argparse
-from pathlib import Path
-import logging
-import sys
 
-from prometheus_client.parser import text_string_to_metric_families
-from prometheus_client import CollectorRegistry, Gauge, write_to_textfile
-
-
-def promtext():
-    # setup argpars
-    # required file first
-    parser = argparse.ArgumentParser(description="Prometheus textfile helper")
-    parser.add_argument(
-        "filename",
-        type=str,
-        help="Path to existing or new prometheus textfile, will be updated",
-    )
-
-    # metric name, required
-    parser.add_argument("metric", type=str, help="metric name (new or updated)")
-
-    # metric value as int/float, required
-    parser.add_argument("value", type=float, help="metric value")
-
-    # metric documentation as optional argument
-    parser.add_argument(
-        "--docs",
-        type=str,
-        help="metric documentation",
-        default="metric appended by promtext-cli",
-    )
-
-    # labels, key-value, minimum 0, repeatable
-    parser.add_argument(
-        "--label", metavar="KEY=VALUE", help="label key=value pairs", action="append"
-    )
-
-    # log level from argparse
-    parser.add_argument(
-        "-v",
-        "--verbose",
-        action="store_const",
-        dest="loglevel",
-        const=logging.INFO,
-    )
-    args = parser.parse_args()
-    logging.basicConfig(level=args.loglevel)
-    logger = logging.getLogger(__name__)
-
-    # processing: check if file is available, if yes, parse
-    textfile = Path(args.filename)
-
-    registry = CollectorRegistry()
-    metrics = {}
-
-    # check if args.filename exists with pathlib
-    if textfile.is_file():
-        for f in text_string_to_metric_families(textfile.read_text(encoding="utf-8")):
-            # per metric: iterate over samples, create metric in registry
-            m = False
-            samples = []
-            for s in f.samples:
-                samples.append(s)
-            if len(samples) > 0:
-                # if we have samples, use the labelnames from them
-                labelnames = list(samples[0].labels.keys())
-                # metric-type specific init
-                if f.type == "gauge":
-                    m = Gauge(
-                        f.name,
-                        f.documentation,
-                        unit=f.unit,
-                        labelnames=labelnames,
-                        registry=registry,
-                    )
-                else:
-                    # we don't support other types yet, continue in these cases
-                    logger.warning("unsupported metric type %s, dropping", f.type)
-                    continue
-                for s in samples:
-                    if len(labelnames) > 0:
-                        labelvalues = list(s.labels.values())
-                        m.labels(*labelvalues).set(s.value)
-                    else:
-                        m.set(s.value)
-                metrics[f.name] = m
-                logger.info(
-                    "copy gauge metric %s with labels %s from old file",
-                    f.name,
-                    ", ".join(labelnames),
-                )
-            else:
-                logger.warning("got empty metric %s from old file, dropping", f.name)
-
-    # add new metric from commandline
-    m = False
-
-    # figure out labelkey- and values
-    labels = {}
-    if args.label:
-        for lpair in args.label:
-            k, v = lpair.split("=")
-            labels[k] = v
-
-    labelvalues = []
-    # here, we use a new metric
-    if args.metric not in metrics:
-        logger.info("adding new metric %s", args.metric)
-        m = Gauge(args.metric, args.docs, registry=registry, labelnames=labels.keys())
-        labelvalues = labels.values()
-    else:
-        m = metrics[args.metric]
-
-        # There is no way to access existing labelnames directly
-        # pylint: disable=W0212
-        old_labelnames = list(m._labelnames)
-        for la in old_labelnames:
-            logger.info("processing label %s", la)
-            if la in labels:  # labelvalues are needed in order!
-                labelvalues.append(labels[la])
-            else:
-                logger.error("previously known label '%s' missing, cannot update!", la)
-                sys.exit(1)
-        if len(old_labelnames) != len(labels.keys()):
-            logger.error(
-                "labelnames for metric %s not the same, cannot update! Old: %s, New: %s",
-                args.metric,
-                old_labelnames,
-                list(labels.keys()),
-            )
-            sys.exit(1)
-        logger.info("updating metric %s", args.metric)
-
-    # actually set the value
-    if len(labelvalues) > 0:
-        m.labels(*labelvalues).set(args.value)
-    else:
-        m.set(args.value)
-
-    # write to file
-    write_to_textfile(args.filename, registry)
-    logger.info("wrote to %s", args.filename)
+def main():
+    """main method invoking the core class"""
+    Promtext().cli_entrypoint()
 
 
 if __name__ == "__main__":
-    promtext()
+    main()

promtext_cli/promtext.py

@@ -0,0 +1,182 @@
+"""this is the module providing the core functionality"""
+
+import argparse
+from pathlib import Path
+import logging
+import sys
+
+from prometheus_client.parser import text_string_to_metric_families
+from prometheus_client import CollectorRegistry, Gauge, write_to_textfile
+
+
+class Promtext:
+    """Class for the core functionality"""
+
+    def __init__(self):
+        self.textfile = None
+        self.logger = None
+        self.args = None
+        self.registry = CollectorRegistry()
+
+        self.metrics = {}
+
+    def _arguments(self):
+        """load every input item from arguments & define cli"""
+        # required file first
+        parser = argparse.ArgumentParser(description="Prometheus textfile helper")
+        parser.add_argument(
+            "filename",
+            type=str,
+            help="Path to existing or new prometheus textfile, will be updated",
+        )
+
+        # metric name, required
+        parser.add_argument("metric", type=str, help="metric name (new or updated)")
+
+        # metric value as int/float, required
+        parser.add_argument("value", type=float, help="metric value")
+
+        # metric documentation as optional argument
+        parser.add_argument(
+            "--docs",
+            type=str,
+            help="metric documentation",
+            default="metric appended by promtext-cli",
+        )
+
+        # labels, key-value, minimum 0, repeatable
+        parser.add_argument(
+            "--label",
+            metavar="KEY=VALUE",
+            help="label key=value pairs",
+            action="append",
+        )
+
+        # log level from argparse
+        parser.add_argument(
+            "-v",
+            "--verbose",
+            action="store_const",
+            dest="loglevel",
+            const=logging.INFO,
+        )
+        self.args = parser.parse_args()
+
+        # processing: check if file is available, if yes, parse
+        self.textfile = Path(self.args.filename)
+
+    def parse_file(self):
+        """If possible, convert the input textfile to metrics in the registry"""
+        # check if self.args.filename exists with pathlib
+        if self.textfile.is_file():
+            for f in text_string_to_metric_families(
+                self.textfile.read_text(encoding="utf-8")
+            ):
+                # per metric: iterate over samples, create metric in registry
+                m = False
+                samples = []
+                for s in f.samples:
+                    samples.append(s)
+                if len(samples) > 0:
+                    # if we have samples, use the labelnames from them
+                    labelnames = list(samples[0].labels.keys())
+                    # metric-type specific init
+                    if f.type == "gauge":
+                        m = Gauge(
+                            f.name,
+                            f.documentation,
+                            unit=f.unit,
+                            labelnames=labelnames,
+                            registry=self.registry,
+                        )
+                    else:
+                        # we don't support other types yet, continue in these cases
+                        self.logger.warning(
+                            "unsupported metric type %s, dropping", f.type
+                        )
+                        continue
+                    for s in samples:
+                        if len(labelnames) > 0:
+                            labelvalues = list(s.labels.values())
+                            m.labels(*labelvalues).set(s.value)
+                        else:
+                            m.set(s.value)
+                    self.metrics[f.name] = m
+                    self.logger.info(
+                        "copy gauge metric %s with labels %s from old file",
+                        f.name,
+                        ", ".join(labelnames),
+                    )
+                else:
+                    self.logger.warning(
+                        "got empty metric %s from old file, dropping", f.name
+                    )
+
+    def _build_metrics(self):
+        # add new metric from commandline
+        m = False
+
+        # figure out labelkey- and values
+        labels = {}
+        if self.args.label:
+            for lpair in self.args.label:
+                k, v = lpair.split("=")
+                labels[k] = v
+
+        labelvalues = []
+        # here, we use a new metric
+        if self.args.metric not in self.metrics:
+            self.logger.info("adding new metric %s", self.args.metric)
+            m = Gauge(
+                self.args.metric,
+                self.args.docs,
+                registry=self.registry,
+                labelnames=labels.keys(),
+            )
+            labelvalues = labels.values()
+        else:
+            m = self.metrics[self.args.metric]
+
+            # There is no way to access existing labelnames directly
+            # pylint: disable=W0212
+            old_labelnames = list(m._labelnames)
+            for la in old_labelnames:
+                self.logger.info("processing label %s", la)
+                if la in labels:  # labelvalues are needed in order!
+                    labelvalues.append(labels[la])
+                else:
+                    self.logger.error(
+                        "previously known label '%s' missing, cannot update!", la
+                    )
+                    sys.exit(1)
+            if len(old_labelnames) != len(labels.keys()):
+                self.logger.error(
+                    "labelnames for metric %s not the same, cannot update! Old: %s, New: %s",
+                    self.args.metric,
+                    old_labelnames,
+                    list(labels.keys()),
+                )
+                sys.exit(1)
+            self.logger.info("updating metric %s", self.args.metric)
+
+        # actually set the value
+        if len(labelvalues) > 0:
+            m.labels(*labelvalues).set(self.args.value)
+        else:
+            m.set(self.args.value)
+
+    def output_file(self):
+        """Output to a textfile"""
+        write_to_textfile(self.textfile, self.registry)
+        self.logger.info("wrote to %s", self.textfile)
+
+    def cli_entrypoint(self):
+        """Main method called from the CLI"""
+        self._arguments()
+
+        logging.basicConfig(level=self.args.loglevel)
+        self.logger = logging.getLogger(__name__)
+
+        self.parse_file()
+        self._build_metrics()
+        self.output_file()

{promtext_cli-0.1.2.dev96.dist-info → promtext_cli-0.1.2.dev98.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: promtext-cli
-Version: 0.1.2.dev96
+Version: 0.1.2.dev98
 Summary: Prometheus Textfile Tooling
 Project-URL: Documentation, https://codeberg.org/margau/promtext-cli/src/branch/main#readme
 Project-URL: Issues, https://codeberg.org/margau/promtext-cli/issues

promtext_cli-0.1.2.dev98.dist-info/RECORD

@@ -0,0 +1,7 @@
+promtext_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+promtext_cli/main.py,sha256=dSifo05hboJyEDbLv6F9u7ETNndlXmrRn_Ol8r4EEFk,208
+promtext_cli/promtext.py,sha256=BhCZ8ZjyyxD680FeXiaPwU9jXCku4D8fVPRLbZv0_Mc,6402
+promtext_cli-0.1.2.dev98.dist-info/METADATA,sha256=Y_VFIy0JR-yXiIedY5xoK6gpG81CIJj_cmL1lUdJFzQ,3414
+promtext_cli-0.1.2.dev98.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+promtext_cli-0.1.2.dev98.dist-info/entry_points.txt,sha256=qH5Z6dM8QLK7tQ0RrdWZQtRcsdCyaGSU1WXygIcaRRc,52
+promtext_cli-0.1.2.dev98.dist-info/RECORD,,

promtext_cli-0.1.2.dev98.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+promtext = promtext_cli:main.main

promtext_cli-0.1.2.dev96.dist-info/RECORD

@@ -1,6 +0,0 @@
-promtext_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-promtext_cli/main.py,sha256=Ce6GRsfi1IkHyatbENEZ8kTsoWnaoow_gRlwZRKMNYg,5054
-promtext_cli-0.1.2.dev96.dist-info/METADATA,sha256=ksmp_99i-MyCAheZHgGE5saMFYTqIZKfHQANtAVFifk,3414
-promtext_cli-0.1.2.dev96.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-promtext_cli-0.1.2.dev96.dist-info/entry_points.txt,sha256=mIY1sCmFCjCpCfn3px-BajeXOaKZIA7iR5_fCaUJ4gg,56
-promtext_cli-0.1.2.dev96.dist-info/RECORD,,

promtext_cli-0.1.2.dev96.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-promtext = promtext_cli.main:promtext