from typing import Iterable, Optional
import json

from chocula.util import clean_str
from chocula.common import DirectoryLoader
from chocula.database import DirectoryInfo, HomepageUrl


class SzczepanskiLoader(DirectoryLoader):
    """
    CSV Columns:

    """

    source_slug = "szczepanski"

    def open_file(self) -> Iterable:
        return open(self.config.szczepanski.filepath, "r")

    def parse_record(self, row) -> Optional[DirectoryInfo]:

        if not row:
            return None

        row = json.loads(row)

        info = DirectoryInfo(
            directory_slug=self.source_slug,
            issne=row.get("issne"),
            issnp=row.get("issnp"),
            raw_issn=row.get("issn"),
            name=clean_str(row["title"]),
            publisher=clean_str(row.get("ed")),
        )

        info.extra["szczepanski"] = dict(as_of=self.config.szczepanski.date)
        if row.get("extra"):
            info.extra["szczepanski"]["notes"] = row.get("extra")
        for k in ("other_titles", "year_spans", "ed"):
            if row.get(k):
                info.extra["szczepanski"][k] = row[k]

        url = HomepageUrl.from_url(row.get("url"))
        if url:
            info.homepage_urls.append(url)

        return info