<filename>bgmi/website/model.py
from operator import attrgetter
from typing import List

import attr

from bgmi.lib.constants import BANGUMI_UPDATE_TIME


@attr.s
class Episode:
    title = attr.ib(type=str)
    download = attr.ib(type=str)
    episode = attr.ib(default=0, type=int)
    time = attr.ib(default=0, type=int)
    subtitle_group = attr.ib(default="", type=str)
    name = attr.ib(default="", type=str)

    @staticmethod
    def remove_duplicated_bangumi(result: List["Episode"]) -> List["Episode"]:
        ret = []
        episodes = list({i.episode for i in result})
        for i in result:
            if i.episode in episodes:
                ret.append(i)
                del episodes[episodes.index(i.episode)]

        return ret


@attr.s
class SubtitleGroup:
    id = attr.ib(type=str)
    name = attr.ib(type=str)


@attr.s
class WebsiteBangumi:
    keyword = attr.ib(type=str)
    update_time = attr.ib(
        default="Unknown", type=str, validator=attr.validators.in_(BANGUMI_UPDATE_TIME)
    )
    name = attr.ib(default="", type=str)
    status = attr.ib(default=0, type=int)
    subtitle_group = attr.ib(factory=list)  # type: List[SubtitleGroup]
    cover = attr.ib(default="", type=str)
    episodes = attr.ib(factory=list)  # type: List[Episode]

    @property
    def max_episode(self) -> int:
        return max(self.episodes, key=attrgetter("episode")).episode

<filename>bgmi/website/model.py
from operator import attrgetter
from typing import List

import attr

from bgmi.lib.constants import BANGUMI_UPDATE_TIME


@attr.s
class Episode:
    title = attr.ib(type=str)
    download = attr.ib(type=str)
    episode = attr.ib(default=0, type=int)
    time = attr.ib(default=0, type=int)
    subtitle_group = attr.ib(default="", type=str)
    name = attr.ib(default="", type=str)

    @staticmethod
    def remove_duplicated_bangumi(result: List["Episode"]) -> List["Episode"]:
        ret = []
        episodes = list({i.episode for i in result})
        for i in result:
            if i.episode in episodes:
                ret.append(i)
                del episodes[episodes.index(i.episode)]

        return ret


@attr.s
class SubtitleGroup:
    id = attr.ib(type=str)
    name = attr.ib(type=str)


@attr.s
class WebsiteBangumi:
    keyword = attr.ib(type=str)
    update_time = attr.ib(
        default="Unknown", type=str, validator=attr.validators.in_(BANGUMI_UPDATE_TIME)
    )
    name = attr.ib(default="", type=str)
    status = attr.ib(default=0, type=int)
    subtitle_group = attr.ib(factory=list)  # type: List[SubtitleGroup]
    cover = attr.ib(default="", type=str)
    episodes = attr.ib(factory=list)  # type: List[Episode]

    @property
    def max_episode(self) -> int:
        return max(self.episodes, key=attrgetter("episode")).episode

# type: List[SubtitleGroup] # type: List[Episode]