homeassistant/custom_components/hacs/helpers/classes/repositorydata.py

"""Repository data."""
from datetime import datetime
from typing import List

import attr


@attr.s(auto_attribs=True)
class RepositoryData:
    """RepositoryData class."""

    archived: bool = False
    authors: List[str] = []
    category: str = ""
    content_in_root: bool = False
    country: List[str] = []
    config_flow: bool = False
    default_branch: str = None
    description: str = ""
    domain: str = ""
    domains: List[str] = []
    downloads: int = 0
    file_name: str = ""
    filename: str = ""
    first_install: bool = False
    fork: bool = False
    full_name: str = ""
    hacs: str = None  # Minimum HACS version
    hide: bool = False
    hide_default_branch: bool = False
    homeassistant: str = None  # Minimum Home Assistant version
    id: int = 0
    iot_class: str = None
    installed: bool = False
    installed_commit: str = None
    installed_version: str = None
    open_issues: int = 0
    last_commit: str = None
    last_version: str = None
    last_updated: str = 0
    manifest_name: str = None
    new: bool = True
    persistent_directory: str = None
    pushed_at: str = ""
    releases: bool = False
    render_readme: bool = False
    published_tags: List[str] = []
    selected_tag: str = None
    show_beta: bool = False
    stargazers_count: int = 0
    topics: List[str] = []
    zip_release: bool = False

    @property
    def stars(self):
        """Return the stargazers count."""
        return self.stargazers_count or 0

    @property
    def name(self):
        """Return the name."""
        if self.category in ["integration", "netdaemon"]:
            return self.domain
        return self.full_name.split("/")[-1]

    def to_json(self):
        """Export to json."""
        return attr.asdict(self)

    @staticmethod
    def create_from_dict(source: dict):
        """Set attributes from dicts."""
        data = RepositoryData()
        for key in source:
            print(key)
            if key in data.__dict__:
                if key == "pushed_at":
                    if source[key] == "":
                        continue
                    if "Z" in source[key]:
                        setattr(
                            data,
                            key,
                            datetime.strptime(source[key], "%Y-%m-%dT%H:%M:%SZ"),
                        )
                    else:
                        setattr(
                            data,
                            key,
                            datetime.strptime(source[key], "%Y-%m-%dT%H:%M:%S"),
                        )
                elif key == "id":
                    setattr(data, key, str(source[key]))
                elif key == "country":
                    if isinstance(source[key], str):
                        setattr(data, key, [source[key]])
                    else:
                        setattr(data, key, source[key])
                else:
                    setattr(data, key, source[key])
        return data

    def update_data(self, data: dict):
        """Update data of the repository."""
        for key in data:
            if key in self.__dict__:
                if key == "pushed_at":
                    if data[key] == "":
                        continue
                    if "Z" in data[key]:
                        setattr(
                            self,
                            key,
                            datetime.strptime(data[key], "%Y-%m-%dT%H:%M:%SZ"),
                        )
                    else:
                        setattr(
                            self, key, datetime.strptime(data[key], "%Y-%m-%dT%H:%M:%S")
                        )
                elif key == "id":
                    setattr(self, key, str(data[key]))
                elif key == "country":
                    if isinstance(data[key], str):
                        setattr(self, key, [data[key]])
                    else:
                        setattr(self, key, data[key])
                else:
                    setattr(self, key, data[key])