From ad30bee143b960725472270553088530989d1b13 Mon Sep 17 00:00:00 2001 From: diegantobass Date: Wed, 7 May 2025 21:42:33 +0200 Subject: [PATCH] clean duplicate pictures --- vcf_cleaner.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/vcf_cleaner.py b/vcf_cleaner.py index 744e251..f3bc3a2 100644 --- a/vcf_cleaner.py +++ b/vcf_cleaner.py @@ -9,18 +9,29 @@ check_for_multiple_pictures = False cards = [] current_card = "" + for line in input_file: + current_card += line + if "END:VCARD" in line: contact = Contact.from_vcard(current_card) + if check_for_multiple_pictures: if "PHOTO" in contact.custom and len(contact.custom["PHOTO"]) > 1: - os.makedirs("photo/" + contact.fn, exist_ok=True) + os.makedirs("multiple_pictures/" + contact.fn, exist_ok=True) for image in range(len(contact.custom["PHOTO"])): print(contact.custom["PHOTO"][image]) with open("photo/" + contact.fn + "/" + str(image) + ".jpg", "wb") as f: f.write(base64.decodebytes(str.encode(contact.custom["PHOTO"][image]))) + if "PHOTO" in contact.custom: + os.makedirs("pictures/" + contact.fn, exist_ok=True) + with open("pictures/" + contact.fn + "/profile.jpg", "wb") as f: + f.write(base64.decodebytes(str.encode(contact.custom["PHOTO"][0]))) + contact.photo_path = "pictures/" + contact.fn + "/profile.jpg" + contact.custom = {} + cards.append(contact) current_card = "" continue