generate.py

from __future__ import annotations
from dataclasses import dataclass
import json
from pathlib import Path

import requests
from jinja2 import Environment, FileSystemLoader
from jinja_markdown2 import MarkdownExtension

URL = 'https://lospec.com/palette-list/load?colorNumberFilterType=exact&colorNumber=16&page={page}&tag=&sortingType=downloads'
ROOT = Path(__file__).parent


def download() -> list[dict]:
    result = []
    for page in range(5):
        resp = requests.get(URL.format(page=page))
        resp.raise_for_status()
        result.extend(resp.json()["palettes"])
    return result


def get_palettes() -> list[dict]:
    path = ROOT / ".palettes.json"
    if path.exists():
        return json.loads(path.read_text())
    result = download()
    path.write_text(json.dumps(result))
    return result


palettes = get_palettes()
env = Environment(loader=FileSystemLoader('templates'))
env.add_extension(MarkdownExtension)
public_dir = Path('public')
public_dir.mkdir(exist_ok=True)

template = env.get_template('index.html.j2')
content = template.render(palettes=palettes)
Path('public', 'index.html').write_text(content)


@dataclass
class Color:
    index: int
    rgb: str

    @property
    def r_hex(self):
        return self.rgb[:2]

    @property
    def g_hex(self):
        return self.rgb[2:4]

    @property
    def b_hex(self):
        return self.rgb[4:]


template = env.get_template('palette.html.j2')
for p in palettes:
    slug = p['slug']
    colors = [Color(i+1, c) for i, c in enumerate(p['colors'])]
    content = template.render(p=p, colors=colors)
    Path('public', f'{slug}.html').write_text(content)