Python/armylist.py at main · grahamharrison68/Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
from html import escape
from pathlib import Path

import pandas as pd

import warnings

warnings.filterwarnings(
    "ignore",
    message="Data Validation extension is not supported and will be removed",
    category=UserWarning,
)

PAGE_STYLE = """
body { font-family: Arial, sans-serif; margin: 0; padding: 2rem; color: #102a43; background: #eef5fb; }
main { max-width: 1100px; margin: 0 auto; }
h1 { color: #0b3d91; }
.tiles { display: grid; grid-template-columns: repeat(auto-fit, minmax(180px, 1fr)); gap: 1rem; }
.tile { box-sizing: border-box; display: flex; align-items: center; justify-content: center; min-height: 180px; padding: 1rem; border: 1px solid #0b3d91; border-radius: 8px; background: #1976d2; color: white; font-size: 1.1rem; font-weight: bold; text-align: center; text-decoration: none; box-shadow: 0 3px 8px #102a4326; }
.tile:hover { background: #0b3d91; }
"""


def page(title, content):
    """Return a complete HTML page containing the supplied content."""
    return f"""<!doctype html>
<html lang="en">
<head>
<meta charset="utf-8">
<meta name="viewport" content="width=device-width, initial-scale=1">
<title>{escape(str(title))}</title>
<style>{PAGE_STYLE}</style>
</head>
<body><main>{content}</main></body>
</html>
"""


def validate_columns(dataframes):
    required = {
        "df_army": {"Name", "Filename"},
        "df_unit": {"Unit", "URL"},
        "df_army_list": {"Army", "Unit", "Points"},
    }
    for name, columns in required.items():
        missing = columns - set(dataframes[name].columns)
        if missing:
            raise ValueError(f"{name} is missing columns: {', '.join(sorted(missing))}")


def generate_pages(df_army, df_unit, df_army_list, output_dir):
    """Generate the army index and one linked page for every army."""
    validate_columns({"df_army": df_army, "df_unit": df_unit, "df_army_list": df_army_list})
    output_dir.mkdir(parents=True, exist_ok=True)

    unit_urls = dict(zip(df_unit["Unit"], df_unit["URL"]))
    army_tiles = []

    for army in df_army.itertuples(index=False):
        filename = Path(str(army.Filename)).name
        army_rows = df_army_list[df_army_list["Army"] == army.Name].sort_values(
            by="Unit", key=lambda values: values.astype(str).str.casefold()
        )
        unit_tiles = []
        for unit in army_rows.itertuples(index=False):
            unit_name = str(unit.Unit)
            unit_url = unit_urls.get(unit_name)
            if not unit_url or pd.isna(unit_url):
                raise ValueError(f"No URL found in df_unit for unit: {unit_name}")
            unit_tiles.append(
                f'<a class="tile" href="{escape(str(unit_url), quote=True)}" '
                f'target="_blank" rel="noopener noreferrer">'
                f'{escape(unit_name)} ({escape(str(unit.Points))} points)</a>'
            )

        army_content = (
            f"<h1>{escape(str(army.Name))}</h1>"
            f'<section class="tiles">{"".join(unit_tiles)}</section>'
        )
        (output_dir / filename).write_text(page(army.Name, army_content), encoding="utf-8")
        army_tiles.append(
            f'<a class="tile" href="{escape(filename, quote=True)}" '
            f'target="_blank" rel="noopener noreferrer">{escape(str(army.Name))}</a>'
        )

    index_content = f'<h1>Army Lists</h1><section class="tiles">{"".join(army_tiles)}</section>'
    (output_dir / "index.html").write_text(page("Army Lists", index_content), encoding="utf-8")


def main():
    """Load the workbook and generate the HTML pages."""
    project_dir = Path(__file__).resolve().parent
    # input_file = project_dir / "data" / "in" / "ArmyList.xlsx"
    # output_dir = project_dir / "data" / "out"

    input_file = project_dir / "ArmyList.xlsx"
    output_dir = project_dir

    print("Loading army list data...")
    df_army = pd.read_excel(input_file, sheet_name="Army")
    df_unit = pd.read_excel(input_file, sheet_name="Unit")
    df_army_list = pd.read_excel(input_file, sheet_name="ArmyList")
    generate_pages(df_army, df_unit, df_army_list, output_dir)
    print(f"Generated army pages in {output_dir}")


if __name__ == "__main__":
    main()