100 lines
3.7 KiB
Python
100 lines
3.7 KiB
Python
import os
|
|
import sys
|
|
import json
|
|
import re
|
|
from pathlib import Path
|
|
|
|
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])
|
|
copies_dir = Path(work_dir) / "Copies"
|
|
bnot_dir = sys.argv[2]
|
|
target_subdir = os.path.join(work_dir, "A Rendre")
|
|
os.makedirs(target_subdir, exist_ok=True)
|
|
|
|
# --- 1. Load the expected names list ---
|
|
expected_names = set()
|
|
names_path = os.path.join(work_dir, "names")
|
|
if not os.path.exists(names_path):
|
|
names_path = "names" # Fallback to current dir
|
|
|
|
if os.path.exists(names_path):
|
|
with open(names_path, 'r', encoding='utf-8') as f:
|
|
expected_names = {line.strip() for line in f if line.strip()}
|
|
else:
|
|
print(f"Warning: 'names' file not found in {work_dir} or current directory.")
|
|
|
|
# --- 2. Existing Collection Logic ---
|
|
pattern = re.compile(r"^Copie(\d+)\.json$")
|
|
copies_map = defaultdict(list)
|
|
assigned_names = set() # To track which names were successfully linked
|
|
|
|
for filename in os.listdir(copies_dir):
|
|
match = pattern.match(filename)
|
|
if match:
|
|
copie_id = match.group(1)
|
|
json_path = os.path.join(copies_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}")
|
|
|
|
# --- 3. Process and Link ---
|
|
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}")
|
|
|
|
source_folder = None
|
|
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
|
|
elif os.path.exists(os.path.join(path_a, "Concat.jpg")) and os.path.exists(os.path.join(path_a, "score.json")):
|
|
source_folder = path_a
|
|
|
|
if not source_folder:
|
|
continue
|
|
|
|
# If we reached here, the link is possible
|
|
assigned_names.add(name)
|
|
|
|
dest_folder_name = f"{safe_name} ({copie_id})"
|
|
dest_path = os.path.join(target_subdir, dest_folder_name)
|
|
os.makedirs(dest_path, exist_ok=True)
|
|
|
|
links = [("Concat.jpg", f"{safe_name}.jpg"),("Concat_F.pdf", f"{safe_name}.pdf"), ("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)
|
|
if os.path.exists(src_file): os.symlink(src_file, dst_link)
|
|
except Exception as e:
|
|
print(f"Error linking {src_name} for {dest_folder_name}: {e}")
|
|
|
|
# --- 4. Print Unassigned Names ---
|
|
unassigned = expected_names - assigned_names
|
|
if unassigned:
|
|
print("\n" + "!" * 40)
|
|
print("NAMES FROM LIST NOT ASSIGNED:")
|
|
for n in sorted(unassigned):
|
|
print(f" - {n}")
|
|
print("!" * 40)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|