Improve Gauge typing for param multiprocess_mode

rafsaf · rafsaf · commit af20d77e449e · 2023-07-12T01:32:19.000+02:00
Signed-off-by: rafsaf &lt;rafal.safin12@gmail.com&gt;
diff --git a/prometheus_client/metrics.py b/prometheus_client/metrics.py
@@ -1,4 +1,5 @@
 import os
+import sys
 from threading import Lock
 import time
 import types
@@ -17,6 +18,9 @@
 from .samples import Exemplar, Sample
 from .utils import floatToGoString, INF
 
+if sys.version_info >= (3, 8, 0):
+    from typing import Literal
+
 T = TypeVar('T', bound='MetricWrapperBase')
 F = TypeVar("F", bound=Callable[..., Any])
 
@@ -357,7 +361,7 @@ def __init__(self,
                  unit: str = '',
                  registry: Optional[CollectorRegistry] = REGISTRY,
                  _labelvalues: Optional[Sequence[str]] = None,
-                 multiprocess_mode: str = 'all',
+                 multiprocess_mode: "Literal['all', 'liveall', 'min', 'livemin', 'max', 'livemax', 'sum', 'livesum']" = 'all',
                  ):
         self._multiprocess_mode = multiprocess_mode
         if multiprocess_mode not in self._MULTIPROC_MODES:
