#!/usr/bin/env python3

import argparse
import logging
from typing import TextIO

from src.logic.neo4j import get_identities
from src.utils import read_emails, write_identity_ids

logging.basicConfig(level=logging.INFO)



def main(input_file: TextIO, output_file: TextIO, key: str) -> None:
    emails = list(read_emails(input_file, key))
    email_identity_pairs = get_identities(emails)
    write_identity_ids(output_file, email_identity_pairs)



if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Email to Identity')
    parser.add_argument('-if', dest='input_file', help='input file', type=argparse.FileType('r'), required=True)
    parser.add_argument('-of', dest='output_file', help='output file', type=argparse.FileType('w'), required=True)
    parser.add_argument('-n', '--column-name', dest='key', default='email', help='column name')


    args = parser.parse_args()
    main(**vars(args))