cic-stack/apps/data-seeding/cic_ussd/import_ussd_data.py

# standard imports
import argparse
import csv
import logging
import os
import psycopg2

# external imports
from confini import Config

# local imports


default_config_dir = './config'
logging.basicConfig(level=logging.WARNING)
logg = logging.getLogger()

arg_parser = argparse.ArgumentParser(description='Pins import script.')
arg_parser.add_argument('-c', type=str, default=default_config_dir, help='config root to use.')
arg_parser.add_argument('--env-prefix',
                        default=os.environ.get('CONFINI_ENV_PREFIX'),
                        dest='env_prefix',
                        type=str,
                        help='environment prefix for variables to overwrite configuration.')
arg_parser.add_argument('import_dir', default='out', type=str, help='user export directory')
arg_parser.add_argument('-v', help='be verbose', action='store_true')
arg_parser.add_argument('-vv', help='be more verbose', action='store_true')

args = arg_parser.parse_args()

if args.vv:
    logging.getLogger().setLevel(logging.DEBUG)
elif args.v:
    logging.getLogger().setLevel(logging.INFO)

config = Config(args.c, args.env_prefix)
config.process()
config.censor('PASSWORD', 'DATABASE')
logg.debug(f'config loaded from {args.c}:\n{config}')


def main():
    with open(f'{args.import_dir}/ussd_data.csv') as ussd_data_file:
        ussd_data = [tuple(row) for row in csv.reader(ussd_data_file)]

    db_conn = psycopg2.connect(
        database=config.get('DATABASE_NAME'),
        host=config.get('DATABASE_HOST'),
        port=config.get('DATABASE_PORT'),
        user=config.get('DATABASE_USER'),
        password=config.get('DATABASE_PASSWORD')
    )
    db_cursor = db_conn.cursor()
    sql = 'UPDATE account SET status = %s, preferred_language = %s WHERE phone_number = %s'
    for element in ussd_data:
        status = 2 if int(element[1]) == 1 else 1
        preferred_language = element[2]
        phone_number = element[0]
        db_cursor.execute(sql, (status, preferred_language, phone_number))
        logg.debug(f'Updating account:{phone_number} with: preferred language: {preferred_language} status: {status}.')
    db_conn.commit()
    db_cursor.close()
    db_conn.close()


if __name__ == '__main__':
    main()