utils.py

import itertools
from dataclasses import dataclass
from enum import Enum
from typing import TYPE_CHECKING

from pydantic import BaseModel, Extra

if TYPE_CHECKING:
    from grafana_dashboard.model.dashboard_types_gen import Dashboard


class MyBaseModel(BaseModel):
    def __repr_args__(self):
        return [
            (k, _repr_transform_arg_value(v))
            for k, v in super().__repr_args__()
            if _repr_should_keep_arg(self.__class__, k, v)
        ]

    class Config:
        extra = Extra.allow  # #10131
        # https://github.com/fzyzcjy/yplusplus/issues/10176#issuecomment-1634063216
        allow_population_by_field_name = True


def _repr_transform_arg_value(v):
    # #10118
    if isinstance(v, Enum):
        return _Reprable(str(v))

    return v


def _repr_should_keep_arg(cls, k: str, v):
    # https://github.com/fzyzcjy/yplusplus/issues/10117#issuecomment-1630217216
    if v is None:
        return False

    info = cls.schema().get('properties', {}).get(k)
    # print(f'hi cls={cls} k={k} v={v} info={info}')

    if info is None or 'default' not in info:
        return True

    return v != info['default']


@dataclass
class _Reprable:
    content: str

    def __repr__(self):
        return self.content


def dashboard_auto_panel_ids(dashboard: 'Dashboard'):
    ids = set([panel.id for panel in dashboard.panels if panel.id])
    auto_ids = (i for i in itertools.count(1) if i not in ids)
    for panel in dashboard.panels:
        if panel.id is None:
            panel.id = next(auto_ids)