mandos.model.utils.hit_utils

Complexity

Total Complexity

4

Size/Duplication

Total Lines	35
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	HitUtils.hits_to_df()	0	9	2
A	HitUtils.df_to_hits()	0	12	2

from typing import Sequence

Missing module docstring

import pandas as pd

Unable to import 'pandas'

from mandos.model.hits import AbstractHit, HitFrame
from mandos.model.concrete_hits import HIT_CLASSES


class HitUtils:

Missing class docstring

    @classmethod
    def hits_to_df(cls, hits: Sequence[AbstractHit]) -> HitFrame:

Missing function or method docstring

        data = []
        for hit in hits:
            x = {f: getattr(hit, f) for f in hit.__class__.fields()}

Variable name "x" doesn't conform to snake_case naming style ('([^\\W\\dA-Z][^\\WA-Z]2,|_[^\\WA-Z]*|__[^\\WA-Z\\d_][^\\WA-Z]+__)$' pattern)

            x["universal_id"] = hit.universal_id
            x["hit_class"] = hit.hit_class
            data.append(x)
        return HitFrame([pd.Series(x) for x in data])

    @classmethod
    def df_to_hits(cls, self: HitFrame) -> Sequence[AbstractHit]:

Missing function or method docstring

        hits = []
        for row in self.itertuples():
            clazz = HIT_CLASSES[row.hit_class]
            # ignore extra columns
            # if cols are missing, let it fail on clazz.__init__
            data = {f: getattr(row, f) for f in clazz.fields()}
            # noinspection PyArgumentList
            hit = clazz(**data)
            hits.append(hit)
        return hits


__all__ = ["HitUtils"]