mirror of
https://github.com/KnugiHK/WhatsApp-Chat-Exporter.git
synced 2026-01-29 05:40:42 +00:00
ddd0ac3143
Replaces regex-based vCard parsing with dedicated functions for parsing lines, handling quoted-printable encoding, and extracting fields. Adds support for CHARSET and ENCODING parameters, improves handling of multi-line and encoded values, and centralizes vCard entry processing for better maintainability and accuracy.
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
# from contacts_names_from_vcards import readVCardsFile
|
|
|
|
import os
|
|
from Whatsapp_Chat_Exporter.vcards_contacts import normalize_number, read_vcards_file
|
|
|
|
|
|
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) == 6
|
|
# 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"
|
|
|
|
|
|
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'
|