csv2vcf is a small command line tool to convert CSV files to VCard (.vcf) files.

csv2vcf.py

"""
Original Author: Mridul Ahuja
Co-Author: Jakob Schöttl
GitHub: https://github.com/schoettl/csv2vcf
Description: A small command line tool to convert CSV files to VCard files

Changes:
- python3
"""

import os
import sys
import csv
import json


def convert_to_vcard(input_file, single_output, input_file_format):

    FN = input_file_format['name']-1 if 'name' in input_file_format else None
    NICKNAME = input_file_format['nickname']-1 if 'nickname' in input_file_format else None
    ORG = input_file_format['org']-1 if 'org' in input_file_format else None
    TEL = input_file_format['tel']-1 if 'tel' in input_file_format else None
    URL = input_file_format['url']-1 if 'url' in input_file_format else None
    BDAY = input_file_format['bday']-1 if 'bday' in input_file_format else None
    ROLE = input_file_format['role']-1 if 'role' in input_file_format else None
    EMAIL = input_file_format['email']-1 if 'email' in input_file_format else None

    # if single output option is selected
    if single_output:
        with open( input_file, 'r' ) as source_file:
            reader = csv.reader( source_file )
            with open('csv2vcf/all_contacts.vcf', 'w') as single_vcf:
               i = 0
               for row in reader:

                  FN_VAL = row[FN] if FN is not None else ''
                  NICKNAME_VAL = row[NICKNAME] if NICKNAME is not None else ''
                  ORG_VAL = row[ORG] if ORG is not None else ''
                  TEL_VAL = row[TEL] if TEL is not None else ''
                  URL_VAL = row[URL] if URL is not None else ''
                  BDAY_VAL = row[BDAY] if BDAY is not None else ''
                  ROLE_VAL = row[ROLE] if ROLE is not None else ''
                  EMAIL_VAL = row[EMAIL] if EMAIL is not None else ''

                  print('BEGIN:VCARD')
                  print('VERSION:3.0')
                  print('N:' + FN_VAL)
                  print('FN:' + FN_VAL)
                  print('NICKNAME:' + NICKNAME_VAL)
                  print('TEL;HOME;VOICE:' + TEL_VAL)
                  print('EMAIL:' + EMAIL_VAL)
                  print('BDAY:' + BDAY_VAL)
                  print('ORG:' + ORG_VAL)
                  print('ROLE:' + ROLE_VAL)
                  print('URL:' + URL_VAL)
                  print('END:VCARD')
                  print('----------------------')

                  # write the single file
                  single_vcf.write('BEGIN:VCARD' + "\n")
                  single_vcf.write('VERSION:3.0' + "\n")
                  single_vcf.write('N:' + FN_VAL + ';' + "\n")
                  single_vcf.write('FN:' + FN_VAL + "\n")
                  single_vcf.write('NICKNAME:' + NICKNAME_VAL + "\n")
                  single_vcf.write('TEL;HOME;VOICE:' + TEL_VAL + "\n")
                  single_vcf.write('EMAIL:' + EMAIL_VAL + "\n")
                  single_vcf.write('BDAY:' + BDAY_VAL + "\n")
                  single_vcf.write('ORG:' + ORG_VAL + "\n")
                  single_vcf.write('ROLE:' + ROLE_VAL + "\n")
                  single_vcf.write('URL:' + URL_VAL + "\n")
                  single_vcf.write('END:VCARD' + "\n")
                  single_vcf.write("\n")

                  i += 1

            print(str(i) + " VCARDS written")
            print('----------------------')

    # default ( multi-file output )
    else:
        with open( input_file, 'r' ) as source_file:
            reader = csv.reader( source_file )
            i = 0
            for row in reader:

                FN_VAL = row[FN] if FN is not None else ''
                NICKNAME_VAL = row[NICKNAME] if NICKNAME is not None else ''
                ORG_VAL = row[ORG] if ORG is not None else ''
                TEL_VAL = row[TEL] if TEL is not None else ''
                URL_VAL = row[URL] if URL is not None else ''
                BDAY_VAL = row[BDAY] if BDAY is not None else ''
                ROLE_VAL = row[ROLE] if ROLE is not None else ''
                EMAIL_VAL = row[EMAIL] if EMAIL is not None else ''

                print('BEGIN:VCARD')
                print('VERSION:3.0')
                print('N:' + FN_VAL)
                print('FN:' + FN_VAL)
                print('NICKNAME:' + NICKNAME_VAL)
                print('TEL;HOME;VOICE:' + TEL_VAL)
                print('EMAIL:' + EMAIL_VAL)
                print('BDAY:' + BDAY_VAL)
                print('ORG:' + ORG_VAL)
                print('ROLE:' + ROLE_VAL)
                print('URL:' + URL_VAL)
                print('END:VCARD')
                print('----------------------')

                # write each entry
                each_vcf = open('csv2vcf/' + FN_VAL + '_' + TEL_VAL + ".vcf", 'w')
                each_vcf.write( 'BEGIN:VCARD' + "\n")
                each_vcf.write( 'VERSION:3.0' + "\n")
                each_vcf.write( 'N:' + FN_VAL + ';' + "\n")
                each_vcf.write( 'FN:' + FN_VAL + "\n")
                each_vcf.write( 'NICKNAME:' + NICKNAME_VAL + "\n")
                each_vcf.write( 'TEL;HOME;VOICE:' + TEL_VAL + "\n")
                each_vcf.write( 'EMAIL:' + EMAIL_VAL + "\n")
                each_vcf.write( 'BDAY:' + BDAY_VAL + "\n")
                each_vcf.write( 'ORG:' + ORG_VAL + "\n")
                each_vcf.write( 'ROLE:' + ROLE_VAL + "\n")
                each_vcf.write( 'URL:' + URL_VAL + "\n")
                each_vcf.write( 'END:VCARD' + "\n")
                each_vcf.write("\n")
                each_vcf.close()

                i += 1

            print(str(i) + " VCARDS written")
            print('----------------------')


def main(args):
    args_len = len(args)

    if args_len < 3 or args_len > 4:
        print("Usage:")
        print(args[0] + " filename")
        sys.exit(1)

    if args_len == 3:
        input_file = args[1]

        try:
            input_file_format = json.loads(args[2])
        except Exception as e:
            print('\033[91m'+"ERROR: json could not be parsed"+'\033[0m')
            sys.exit(1)

        single_output = 0
    elif args_len == 4:
        input_file = args[1]

        if args[2] == '-s' or args[2] == '--single':
            single_output = 1
        else:
            print('\033[91m'+"ERROR: invalid argument `" + args[2] + "`"+'\033[0m')
            sys.exit(1)

        try:
            input_file_format = json.loads(args[3])
        except Exception as e:
            print('\033[91m'+"ERROR: json could not be parsed"+'\033[0m')
            sys.exit(1)

    if not os.path.exists(input_file):
        print('\033[91m'+"ERROR: file `" + input_file + "` not found"+'\033[0m')
        sys.exit(1)

    if not os.path.exists('csv2vcf'):
        os.makedirs('csv2vcf')

    convert_to_vcard(input_file, single_output, input_file_format)

if __name__ == '__main__':
    main(sys.argv)