mirror of
https://github.com/KnugiHK/WhatsApp-Chat-Exporter.git
synced 2026-04-01 10:39:01 +00:00
2ebb389ad1
Parse and match vCard properties that use grouping prefixes (e.g. item1.TEL) by extracting the property name correctly. Regression caused by the removal of the vobject dependency.
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
# from contacts_names_from_vcards import readVCardsFile
|
|
|
|
import os
|
|
from Whatsapp_Chat_Exporter.vcards_contacts import normalize_number, read_vcards_file, get_vcard_value
|
|
|
|
|
|
def test_readVCardsFile():
|
|
data_dir = os.path.join(os.path.dirname(__file__), "data")
|
|
data = read_vcards_file(os.path.join(data_dir, "contacts.vcf"), "852")
|
|
if data:
|
|
print("Found Names")
|
|
print("-----------------------")
|
|
for count, contact_tuple in enumerate(data, start=1):
|
|
# The name is the second element of the tuple (at index 1)
|
|
name = contact_tuple[1]
|
|
|
|
# Print the count and the name
|
|
print(f"{count}. {name}")
|
|
print(data)
|
|
assert len(data) == 8
|
|
# Test simple contact name
|
|
assert data[0][1] == "Sample Contact"
|
|
# Test complex name
|
|
assert data[1][1] == "Yard Lawn Guy, Jose Lopez"
|
|
# Test name with emoji
|
|
assert data[2][1] == "John Butler 🌟💫🌟"
|
|
# Test note with multi-line encoding
|
|
assert data[3][1] == "Airline Contact #'s"
|
|
# Test address with multi-line encoding
|
|
assert data[4][1] == "James Peacock Elementary"
|
|
# Test business entry using ORG but not F/FN
|
|
assert data[5][1] == "AAA Car Service"
|
|
# Test grouped entry
|
|
assert data[6][1] == "Racing Team (1)"
|
|
assert data[7][1] == "Racing Team (2)"
|
|
|
|
|
|
def test_grouping_mechanism():
|
|
no_group_vcf = """
|
|
BEGIN:VCARD
|
|
VERSION:2.1
|
|
TEL;CELL:7777777778
|
|
TEL;CELL:7777777779
|
|
TEL;CELL:7777777780
|
|
ORG:Racing Team
|
|
END:VCARD"""
|
|
group_vcf = """
|
|
BEGIN:VCARD
|
|
VERSION:2.1
|
|
item1.TEL;CELL:7777777778
|
|
item2.TEL;CELL:7777777779
|
|
item3.TEL;CELL:7777777780
|
|
ORG:Racing Team
|
|
END:VCARD"""
|
|
assert get_vcard_value(no_group_vcf, "TEL") == ["7777777778", "7777777779", "7777777780"]
|
|
assert get_vcard_value(group_vcf, "TEL") == ["7777777778", "7777777779", "7777777780"]
|
|
|
|
|
|
|
|
def test_create_number_to_name_dicts():
|
|
pass
|
|
|
|
|
|
def test_fuzzy_match_numbers():
|
|
pass
|
|
|
|
|
|
def test_normalize_number():
|
|
assert normalize_number('0531234567', '1') == '1531234567'
|
|
assert normalize_number('001531234567', '2') == '1531234567'
|
|
assert normalize_number('+1531234567', '34') == '1531234567'
|
|
assert normalize_number('053(123)4567', '34') == '34531234567'
|
|
assert normalize_number('0531-234-567', '58') == '58531234567'
|