extract.py

import csv
import sys


def fix_case(name):
    splitname = name.split()
    return " ".join([n.lower().capitalize() for n in splitname])


def extract(file_path):
    originalcsv = csv.DictReader(open(file_path, 'r'))
    newcsv = csv.writer(open("output.csv", 'w'))
    newcsv.writerow(['MSISDN', 'First Name', 'Last Name'])
    for line in originalcsv:
        splitname = line['S_NAME_CUS'].split()
        firstname, lastname = splitname[0], splitname[-1] if len(splitname) > 1 else ""
        newcsv.writerow(
            [line['MSISDN_CURR'], fix_case(firstname), fix_case(lastname)]
        )

if __name__ == '__main__':
    extract(sys.argv[1])