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

file = open("contacts.vcf", 'r').readlines()

cards = []

current_card = ""
for line in 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


# contact = Contact.from_vcard(cards[156])
# print(contact.custom["PHOTO"])

# vcf_text = contact.to_vcard()
# print(vcf_text)
# open('test.vcf', 'w').write(vcf_text)

print(len(cards))