103 lines
3.9 KiB
Python
103 lines
3.9 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 = []
|
|
|
|
# Dostosowanie szerokości kolumn do 0.1, 0.8 i 0.1 textwidth
|
|
wiersze.append(r"\begin{longtable}{|p{0.1\textwidth}|p{0.74\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}")
|
|
|
|
# 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()
|