Page MenuHomeSoftware Heritage
Files F8396429

__init__.py
No OneTemporary
Actions

__init__.py
View Options

# Copyright (C) 2021 The Software Heritage developers
# See the AUTHORS file at the top-level directory of this distribution
# License: GNU General Public License version 3, or any later version
# See top-level LICENSE file for more information
from __future__ import annotations
import importlib
from typing import TYPE_CHECKING, Any, Dict
if TYPE_CHECKING:
from swh.counters.interface import CountersInterface, HistoryInterface
COUNTERS_IMPLEMENTATIONS = {
"redis": ".redis.Redis",
"remote": ".api.client.RemoteCounters",
}
HISTORY_IMPLEMENTATIONS = {
"prometheus": ".history.History",
}
def get_counters(cls: str, **kwargs: Dict[str, Any]) -> CountersInterface:
"""Get an counters object of class `cls` with arguments `args`.
Args:
cls: counters's class, either 'local' or 'remote'
args: dictionary of arguments passed to the
counters class constructor
Returns:
an instance of swh.counters's classes (either local or remote)
Raises:
ValueError if passed an unknown counters class.
"""
class_path = COUNTERS_IMPLEMENTATIONS.get(cls)
if class_path is None:
raise ValueError(
"Unknown counters class `%s`. Supported: %s"
% (cls, ", ".join(COUNTERS_IMPLEMENTATIONS))
)
(module_path, class_name) = class_path.rsplit(".", 1)
module = importlib.import_module(module_path, package=__package__)
Counters = getattr(module, class_name)
return Counters(**kwargs)
def get_history(cls: str, **kwargs: Dict[str, Any]) -> HistoryInterface:
"""Get a history object of class `cls` with arguments `kwargs`.
Args:
cls: history's class, only 'prometheus' is supported actually
kwargs: dictionary of arguments passed to the
counters class constructor
Returns:
an instance of swh.counters.history's classes (either local or remote)
Raises:
ValueError if passed an unknown history class.
"""
class_path = HISTORY_IMPLEMENTATIONS.get(cls)
if class_path is None:
raise ValueError(
"Unknown history class `%s`. Supported: %s"
% (cls, ", ".join(HISTORY_IMPLEMENTATIONS))
)
(module_path, class_name) = class_path.rsplit(".", 1)
module = importlib.import_module(module_path, package=__package__)
History = getattr(module, class_name)
return History(**kwargs)

File Metadata

Mime Type
text/x-python
Expires
Jun 4 2025, 7:53 PM (12 w, 5 d ago)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3364355

Event Timeline

About · Developer information · Legal