import sys

import argparse

import csv2db

def main(args):

    parser = argparse.ArgumentParser(description='To get arguments')

    parser.add_argument('-pa',

                        '--path',

                        type=str,

                        help='Path of the CSV file',

                        required=True)

    parser.add_argument('-ps',

                        '--password',

                        type=str,

                        help='MySQL password',

                        required=True)

    parser.add_argument('-db',

                        '--database',

                        type=str,

                        help='Database name',

                        required=True)

    parser.add_argument('-t',

                        '--table',

                        type=str,

                        help='Table name',

                        required=True)

    parser.add_argument('-ho',

                        '--host',

                        type=str,

                        default='127.0.0.1',

                        help='Host ip address',

                        required=False)

    parser.add_argument('-p',

                        '--port',

                        type=str,

                        default='3306',

                        help='Port of MySQL',

                        required=False)

    parser.add_argument('-u',

                        '--user',

                        type=str,

                        default='root',

                        help='MySQL user',

                        required=False)

    parser.add_argument('-c',

                        '--columns',

                        nargs='+',

                        type=str,

                        default=None,

                        help='Array of desired columns names. Default is all columns',

                        required=False)

    args = parser.parse_args()

    to_db = csv2db.CSV2DB(args.path,

                          args.password,

                          args.database,

                          args.table,

                          args.host,

                          args.user,

                          args.port,

                          args.columns)

    print('Select columns of dataframe')

    to_db.select_column()

    print('Remove redundant newlines')  

    to_db.rm_multi_newline()

    print('Modify columns name for database')  

    to_db.norm_column_name()

    print('Check strings length in the dataframe')  

    to_db.check_length()

    print('Create a new database')  

    to_db.create_database()

    print('Create a database engine')  

    to_db.create_engine()

    print('Convert dataframe to MySQL database')  

    to_db.to_sql()

    print('Add primary key to the database')  

    to_db.add_primary_key()

    print('Change name column type to VARCHAR from TEXT')  

    to_db.change_type()

if __name__ == '__main__':

    main(sys.argv[1:])