144 lines
4.4 KiB
Python
144 lines
4.4 KiB
Python
import pygraphviz as pgv
|
|
import json
|
|
import os
|
|
from PIL import Image, ImageDraw, ImageFont, ImageOps
|
|
|
|
def create_avatar_with_name(user_id, name, avatar_path, output_path):
|
|
"""Crée une image combinant avatar rond et pseudo en dessous"""
|
|
try:
|
|
avatar = Image.open(avatar_path).convert("RGBA")
|
|
avatar = avatar.resize((150, 150))
|
|
|
|
mask = Image.new('L', (150, 150), 0)
|
|
draw = ImageDraw.Draw(mask)
|
|
draw.ellipse((0, 0, 150, 150), fill=255)
|
|
rounded_avatar = ImageOps.fit(avatar, mask.size, centering=(0.5, 0.5))
|
|
rounded_avatar.putalpha(mask)
|
|
|
|
combined = Image.new('RGBA', (200, 200), (0, 0, 0, 0))
|
|
combined.paste(rounded_avatar, (25, 0), rounded_avatar)
|
|
|
|
draw = ImageDraw.Draw(combined)
|
|
try:
|
|
font = ImageFont.truetype("arial.ttf", 14)
|
|
except:
|
|
font = ImageFont.load_default()
|
|
|
|
text_width = draw.textlength(name, font=font)
|
|
draw.text(((200 - text_width) / 2, 160), name, font=font, fill="white")
|
|
|
|
combined.save(output_path)
|
|
return True
|
|
except Exception as e:
|
|
print(f"Erreur création avatar+texte {user_id}: {e}")
|
|
return False
|
|
|
|
def generate_tree(input_file, output_file, root_id=None):
|
|
with open(input_file) as f:
|
|
data = json.load(f)
|
|
|
|
G = pgv.AGraph(
|
|
directed=True,
|
|
rankdir="TB",
|
|
nodesep="0.8",
|
|
ranksep="1.5",
|
|
bgcolor="#000000",
|
|
splines="true"
|
|
)
|
|
|
|
node_style = {
|
|
"shape": "none",
|
|
"imagescale": "false",
|
|
"labelloc": "b",
|
|
"width": "2",
|
|
"height": "2",
|
|
"fixedsize": "true"
|
|
}
|
|
|
|
edge_style = {
|
|
"color": "#FFFFFF",
|
|
"penwidth": "3",
|
|
"arrowsize": "1.2"
|
|
}
|
|
|
|
os.makedirs("temp_avatars", exist_ok=True)
|
|
|
|
# Traitement de tous les membres
|
|
for user_id, info in data["members"].items():
|
|
original_avatar = f"avatars/{user_id}.png"
|
|
combined_avatar = f"temp_avatars/combined_{user_id}.png"
|
|
|
|
if os.path.exists(original_avatar):
|
|
create_avatar_with_name(
|
|
user_id,
|
|
info["name"],
|
|
original_avatar,
|
|
combined_avatar
|
|
)
|
|
G.add_node(
|
|
user_id,
|
|
image=combined_avatar,
|
|
**node_style
|
|
)
|
|
else:
|
|
G.add_node(
|
|
user_id,
|
|
label=info["name"],
|
|
**{**node_style, "fontcolor": "white"}
|
|
)
|
|
|
|
# Nouveau : Dictionnaire pour forcer les rangs des couples
|
|
forced_ranks = {}
|
|
|
|
# Traitement des couples
|
|
for couple in data.get("couples", []):
|
|
if all(m in data["members"] for m in couple):
|
|
member1, member2 = couple
|
|
|
|
# Style spécial pour les liens de couple
|
|
G.add_edge(member1, member2,
|
|
style="dashed",
|
|
color="#FF69B4",
|
|
penwidth="2.5",
|
|
dir="none")
|
|
|
|
# Force les partenaires au même rang
|
|
partner_group = f"couple_{min(couple)}_{max(couple)}"
|
|
G.add_subgraph(couple, name=partner_group, rank="same")
|
|
forced_ranks[member1] = partner_group
|
|
forced_ranks[member2] = partner_group
|
|
|
|
# Organisation hiérarchique améliorée
|
|
if root_id:
|
|
roots = [root_id]
|
|
else:
|
|
roots = data.get("roots", [k for k,v in data["members"].items() if not v.get("parents")])
|
|
|
|
# Nouvelle approche pour les niveaux
|
|
levels = {}
|
|
for user_id in data["members"]:
|
|
gen = data["members"][user_id].get("generation",
|
|
len(data["members"][user_id].get("parents", [])))
|
|
if gen not in levels:
|
|
levels[gen] = []
|
|
levels[gen].append(user_id)
|
|
|
|
# Création des sous-graphes par niveau
|
|
for level, members in levels.items():
|
|
filtered_members = [m for m in members if m not in forced_ranks]
|
|
if filtered_members:
|
|
G.add_subgraph(filtered_members, name=f"rank{level}", rank="same")
|
|
|
|
# Connexions parent-enfant
|
|
for user_id, info in data["members"].items():
|
|
for parent in info.get("parents", []):
|
|
if parent["id"] in data["members"]:
|
|
G.add_edge(parent["id"], user_id, **edge_style)
|
|
|
|
G.layout(prog="dot")
|
|
G.draw(output_file)
|
|
|
|
# Nettoyage
|
|
for f in os.listdir("temp_avatars"):
|
|
os.remove(f"temp_avatars/{f}")
|