so/doc/py/wso_f.py
2025-02-01 12:53:05 +01:00

121 lines
4.7 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import sys
import json
def generuj_wiersze(wezel, wiersze, poziom=0, rodzic_ogolny=False, intensywnosc=60):
"""
Funkcja rekurencyjna:
- Odczytuje z węzła (wezel) pola nr, opis, punkty.
- Dodaje do listy 'wiersze' wiersz tabeli w stylu:
[wcięcie + nr] & [wcięcie + opis] & [wcięcie + punkty] \\
przy czym każde pole jest dodatkowo wcięte zależnie od poziomu zagnieżdżenia.
- Jeśli węzeł ma opis "Wymagania ogólne", to używana jest lista kolorów dla Wymagań Ogólnych.
- Każde dziecko węzła "Wymagania ogólne" ma kolor o 10% mniej intensywny niż jego rodzic.
"""
nr = wezel.get("nr", "")
opis = wezel.get("opis", "")
punkty = wezel.get("punkty", "")
# Wcięcie w LaTeX-u (np. 0.5 em na poziom) zależnie od poziomu
wciecie = r"\hspace{" + f"{0.5 * poziom}em" + "}" # 0.5em na poziom
# Lista kolorów dla Wymagań Standardowych
kolory_standard = [
r"\cellcolor{gray!20}", # Poziom 0
r"\cellcolor{red!30}", # Poziom 1
r"\cellcolor{blue!20}", # Poziom 2
r"\cellcolor{yellow!20}", # Poziom 3
r"\cellcolor{red!25}", # Poziom 4
r"\cellcolor{orange!20}", # Poziom 5
]
# Sprawdź, czy bieżący węzeł to "Wymagania ogólne"
if opis == "Wymagania ogólne":
rodzic_ogolny = True
intensywnosc = 60 # Początkowa intensywność dla "Wymagania ogólne"
# Wybierz kolor na podstawie grupy i intensywności
if rodzic_ogolny:
kolor = f"\\cellcolor{{green!{intensywnosc}}}" # Kolor dla Wymagań Ogólnych
else:
kolor = kolory_standard[poziom % len(kolory_standard)] # Kolor dla Wymagań Standardowych
# Tworzymy wiersz:
# [wcięcie + nr] & [wcięcie + opis] & [wcięcie + punkty] \\
linia = f"{kolor}{wciecie}{nr} & {kolor}{wciecie}{opis} & {kolor}{wciecie}{punkty} \\\\"
wiersze.append(linia)
# Rekurencja dla dzieci
for dziecko in wezel.get("children", []):
if rodzic_ogolny:
# Dla dzieci Wymagań Ogólnych zmniejsz intensywność o 10
generuj_wiersze(dziecko, wiersze, poziom + 1, rodzic_ogolny, intensywnosc - 20)
else:
# Dla innych węzłów zachowaj standardowe kolory
generuj_wiersze(dziecko, wiersze, poziom + 1, rodzic_ogolny, intensywnosc)
def main():
if len(sys.argv) < 3:
print("Użycie: skrypt.py plik_wejściowy.json plik_wyjściowy.tex")
sys.exit(1)
plik_json = sys.argv[1]
plik_tekstowy = sys.argv[2]
# 1. Wczytanie danych JSON z pliku
with open(plik_json, 'r', encoding='utf-8') as f:
dane = json.load(f)
# 2. Bufor na wszystkie wiersze LaTeX
wiersze = []
# --- Nagłówek dla LaTeX ---
wiersze.append(r"\documentclass{article}")
wiersze.append(r"\usepackage[utf8]{inputenc}") # Obsługa polskich znaków
wiersze.append(r"\usepackage[T1]{fontenc}") # Poprawne wyświetlanie polskich znaków
wiersze.append(r"\usepackage[polish]{babel}") # Polska lokalizacja
wiersze.append(r"\usepackage{longtable}")
wiersze.append(r"\usepackage{array}")
wiersze.append(r"\usepackage{geometry}")
wiersze.append(r"\usepackage[table]{xcolor}") # Dodanie pakietu xcolor do kolorowania komórek
wiersze.append(r"\geometry{a4paper, landscape, margin=1in}") # Format A4 poziomo
wiersze.append(r"\begin{document}")
# --- Tytuł i nagłówek tabeli ---
wiersze.append(r"\title{Lista wymagań}")
wiersze.append(r"\author{}")
wiersze.append(r"\date{}")
wiersze.append(r"\maketitle")
# Dostosowanie szerokości kolumn do 0.1, 0.8 i 0.1 textwidth
wiersze.append(r"\begin{longtable}{|p{0.1\textwidth}|p{0.8\textwidth}|p{0.1\textwidth}|}")
wiersze.append(r"\caption{Lista wymagań (przykład)}\\")
wiersze.append(r"\hline")
wiersze.append(r"\textbf{Nr} & \textbf{Opis} & \textbf{Punkty} \\")
wiersze.append(r"\hline")
# firsthead -> co ma się powtórzyć na kolejnej stronie tabeli
wiersze.append(r"\endfirsthead")
wiersze.append(r"\hline")
wiersze.append(r"\textbf{Nr} & \textbf{Opis} & \textbf{Punkty} \\")
wiersze.append(r"\hline")
wiersze.append(r"\endhead")
# 3. Generowanie wierszy tabeli na podstawie JSON-a
for element in dane.get("items", []):
generuj_wiersze(element, wiersze)
wiersze.append(r"\hline") # Dodanie linii odcinającej po każdej sekcji
# 4. Zakończenie longtable
wiersze.append(r"\hline")
wiersze.append(r"\end{longtable}")
wiersze.append(r"\end{document}")
# 5. Zapis do pliku .tex (z polskimi znakami w UTF-8)
with open(plik_tekstowy, 'w', encoding='utf-8') as f:
f.write("\n".join(wiersze) + "\n")
if __name__ == "__main__":
main()