42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import os
|
|
import sys
|
|
import base64
|
|
import pickle
|
|
from pythonvCard4.vcard import Contact
|
|
|
|
input_file = open(sys.argv[1], 'r').readlines()
|
|
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("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
|
|
|
|
output = open('output.vcf', 'w')
|
|
for card in cards:
|
|
vcf_text = card.to_vcard()
|
|
output.write(vcf_text)
|