feat: use argparse for CLI

abcddb2vcard.py

@@ -1,21 +1,43 @@
 #!/usr/bin/env python3
+'''
+Extract data from AddressBook database (.abcddb) to Contacts VCards file (.vcf)
+'''
+import os
 import sys
 from ABCDDB import ABCDDB
 from pathlib import Path
+from argparse import ArgumentParser
 
-if len(sys.argv) != 2:
-    print('  Usage:', Path(__file__).name, 'outfile.vcf')
-    exit(0)
 
-outfile = Path(sys.argv[1])
-if not outfile.parent.exists():
-    print('Output directory does not exist.', file=sys.stderr)
+DB_FILE = str(Path.home().joinpath(
+    'Library', 'Application Support', 'AddressBook', 'AddressBook-v22.abcddb'))
+
+cli = ArgumentParser(description=__doc__)
+cli.add_argument('output', type=str, metavar='outfile.vcf',
+                 help='VCard output file.')
+cli.add_argument('-f', '--force', action='store_true',
+                 help='Overwrite existing output file.')
+cli.add_argument('-i', '--input', type=str, metavar='AddressBook.abcddb',
+                 default=DB_FILE, help='Specify another abcddb input file.'
+                 ' Default: ' + DB_FILE)
+args = cli.parse_args()
+
+# check input args
+if not os.path.isfile(args.input):
+    print('AddressBook "{}" does not exist.'.format(args.input),
+          file=sys.stderr)
+    exit(1)
+elif not os.path.isdir(os.path.dirname(args.output) or os.curdir):
+    print('Output parent directory does not exist.', file=sys.stderr)
+    exit(1)
+elif os.path.isfile(args.output) and not args.force:
+    print('Output file already exist. Use -f to force overwrite.',
+          file=sys.stderr)
     exit(1)
 
-contacts = ABCDDB.load(Path.home().joinpath(
-    'Library/Application Support/AddressBook/AddressBook-v22.abcddb'))
-# contacts = [list(contacts)[-1]]  # test on last imported contact
-with open(outfile, 'w') as f:
+# perform export
+contacts = ABCDDB.load(args.input)
+with open(args.output, 'w') as f:
     for rec in contacts:
         f.write(rec.makeVCard())
     print(len(contacts), 'contacts.')