96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
import os
|
|
import sys
|
|
import json
|
|
import re
|
|
from collections import defaultdict
|
|
|
|
def main():
|
|
if len(sys.argv) < 3:
|
|
print("Usage: python giving_names.py <directory_path> <Bnot/BGnot>")
|
|
sys.exit(1)
|
|
|
|
work_dir = os.path.abspath(sys.argv[1])
|
|
bnot_dir = sys.argv[2]
|
|
target_subdir = os.path.join(work_dir, "A Rendre")
|
|
|
|
os.makedirs(target_subdir, exist_ok=True)
|
|
|
|
pattern = re.compile(r"^Copie(\d+)\.json$")
|
|
copies_map = defaultdict(list)
|
|
|
|
# 1. Collect all data
|
|
for filename in os.listdir(work_dir):
|
|
match = pattern.match(filename)
|
|
if match:
|
|
copie_id = match.group(1)
|
|
json_path = os.path.join(work_dir, filename)
|
|
|
|
try:
|
|
with open(json_path, 'r', encoding='utf-8') as f:
|
|
data = json.load(f)
|
|
name = data.get("name", "Unknown").strip()
|
|
copies_map[name].append(copie_id)
|
|
except Exception as e:
|
|
print(f"Error processing {filename}: {e}")
|
|
|
|
# 2. Check constraints and symlink files
|
|
for name, ids in copies_map.items():
|
|
if name == "Unknown":
|
|
print(f"ALERT: 'Unknown' name found for IDs: {', '.join(ids)}")
|
|
elif len(ids) > 1:
|
|
print(f"ALERT: Name '{name}' assigned to multiple IDs: {', '.join(ids)}")
|
|
|
|
safe_name = re.sub(r'[<>:"/\\|?*]', '', name).strip()
|
|
|
|
for copie_id in ids:
|
|
path_b = os.path.join(work_dir, f"{bnot_dir}/Copie{copie_id}")
|
|
path_a = os.path.join(work_dir, f"Anot/Copie{copie_id}")
|
|
|
|
# Determine source: must contain both files
|
|
source_folder = None
|
|
|
|
# Check Bnot first
|
|
if (os.path.exists(os.path.join(path_b, "Concat.jpg")) and
|
|
os.path.exists(os.path.join(path_b, "score.json"))):
|
|
source_folder = path_b
|
|
# Fallback to Anot
|
|
elif (os.path.exists(os.path.join(path_a, "Concat.jpg")) and
|
|
os.path.exists(os.path.join(path_a, "score.json"))):
|
|
print(f"Debug : couldn't find {path_b} falling back to Anot path")
|
|
source_folder = path_a
|
|
|
|
if not source_folder:
|
|
print(f"Skipping ID {copie_id}: Files missing in both Anot and {bnot_dir}.")
|
|
continue
|
|
|
|
# Create destination directory
|
|
dest_folder_name = safe_name if len(ids) == 1 else f"{safe_name} ({copie_id})"
|
|
dest_path = os.path.join(target_subdir, dest_folder_name)
|
|
os.makedirs(dest_path, exist_ok=True)
|
|
|
|
common = os.path.commonpath([source_folder, dest_path])
|
|
s = os.path.relpath(source_folder, common)
|
|
d = os.path.relpath(dest_path, common)
|
|
|
|
print(f"Linking '{s}' -> '{d}'")
|
|
|
|
# Link configuration: (source_filename, dest_filename)
|
|
links = [
|
|
("Concat.jpg", f"{safe_name}.jpg"),
|
|
("score.json", "score.json")
|
|
]
|
|
|
|
for src_name, dst_name in links:
|
|
src_file = os.path.join(source_folder, src_name)
|
|
dst_link = os.path.join(dest_path, dst_name)
|
|
|
|
try:
|
|
if os.path.lexists(dst_link):
|
|
os.remove(dst_link)
|
|
os.symlink(src_file, dst_link)
|
|
except Exception as e:
|
|
print(f"Error linking {src_name} for {dest_folder_name}: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|