"""网页生成器 — 读取数据库,生成静态 HTML 展示页面。""" import os import json import db BASE_DIR = os.path.dirname(os.path.abspath(__file__)) TEMPLATE_PATH = os.path.join(BASE_DIR, "templates", "index.html") STYLE_PATH = os.path.join(BASE_DIR, "static", "style.css") OUTPUT_DIR = os.path.join(BASE_DIR, "output") OUTPUT_PATH = os.path.join(OUTPUT_DIR, "index.html") def _escape(text): """简单的 HTML 转义。""" if not text: return "" return (str(text) .replace("&", "&") .replace("<", "<") .replace(">", ">") .replace('"', """)) def _render_book_card(book): """渲染单本书的 HTML 卡片。""" cover = book.get("cover_url") or "" cover_html = (f'封面' if cover else '
📖
') rating_html = "" if book.get("rating"): stars = "★" * book["rating"] + "☆" * (5 - book["rating"]) rating_html = f'
{stars}
' scores = [] if book.get("douban_score"): scores.append(f'豆瓣 {book["douban_score"]}') if book.get("goodreads_score"): scores.append(f'GR {book["goodreads_score"]}') scores_html = f'
{"".join(scores)}
' if scores else "" tags_html = "" if book.get("tags"): tag_spans = "".join( f'{_escape(t.strip())}' for t in book["tags"].split(",") if t.strip() ) tags_html = f'
{tag_spans}
' meta_parts = [] if book.get("author"): meta_parts.append(f'{_escape(book["author"])}') if book.get("translator"): meta_parts.append(f'译: {_escape(book["translator"])}') if book.get("publisher"): meta_parts.append(f'{_escape(book["publisher"])}') if book.get("pub_date"): meta_parts.append(f'{_escape(book["pub_date"])}') meta_html = f'
{" · ".join(meta_parts)}
' if meta_parts else "" notes_html = "" if book.get("notes"): notes_html = f'
{_escape(book["notes"])}
' dates = [] if book.get("start_date"): dates.append(f'开始: {_escape(book["start_date"])}') if book.get("finish_date"): dates.append(f'完成: {_escape(book["finish_date"])}') dates_html = f'
{" · ".join(dates)}
' if dates else "" format_badge = "📱 电子书" if book.get("format") == "ebook" else "📕 纸质书" return f"""
{cover_html}

{_escape(book.get('title', ''))}

{meta_html} {rating_html} {scores_html} {tags_html} {notes_html} {dates_html}
{format_badge}
""" def build(): """生成静态 HTML 页面,返回输出路径。""" os.makedirs(OUTPUT_DIR, exist_ok=True) books = db.list_books() stats = db.get_stats() all_tags = db.get_all_tags() cards_html = "\n".join(_render_book_card(b) for b in books) # 读取 CSS with open(STYLE_PATH, "r", encoding="utf-8") as f: css = f.read() # 读取模板并填充 with open(TEMPLATE_PATH, "r", encoding="utf-8") as f: template = f.read() # 统计数据 total = stats["total"] reading = stats["by_status"].get("reading", 0) read = stats["by_status"].get("read", 0) to_read = stats["by_status"].get("to-read", 0) paper = stats["by_format"].get("paper", 0) ebook = stats["by_format"].get("ebook", 0) # 标签按钮 tag_buttons = "".join( f'' for t in all_tags ) html = (template .replace("{{CSS}}", css) .replace("{{TOTAL}}", str(total)) .replace("{{READING}}", str(reading)) .replace("{{READ}}", str(read)) .replace("{{TO_READ}}", str(to_read)) .replace("{{PAPER}}", str(paper)) .replace("{{EBOOK}}", str(ebook)) .replace("{{TAG_BUTTONS}}", tag_buttons) .replace("{{BOOK_CARDS}}", cards_html)) with open(OUTPUT_PATH, "w", encoding="utf-8") as f: f.write(html) # 复制封面图片不处理(直接使用 URL) return OUTPUT_PATH