import os
import sys
import base64
import pickle
from pythonvCard4.vcard import Contact

input_file = open(sys.argv[1], 'r').readlines()

cards = []
current_card = ""
for line in input_file:
    current_card += line
    if "END:VCARD" in line:
        contact = Contact.from_vcard(current_card)
        # if "PHOTO" in contact.custom and len(contact.custom["PHOTO"]) > 1:
        #     os.makedirs("photo/" + contact.fn, exist_ok=True)
        #     for image in range(len(contact.custom["PHOTO"])):
        #         with open("photo/" + contact.fn + "/" + str(image) + ".jpg", "wb") as f:
        #             f.write(base64.decodebytes(str.encode(contact.custom["PHOTO"][image])))

        cards.append(contact)
        current_card = ""
        continue

output = open('output.vcf', 'w')
for card in cards:
    vcf_text = card.to_vcard()
    print(vcf_text)
    output.write(vcf_text)