make_report.py

import sys
import json
import argparse


import gspread


from config import DEFAULT_FILE_PATH, \
    WORK_SHEET, DOCUMENT_ID, OUTPUT_FILE
from storage_api import DiskStorage, GoogleDocsStorage, \
    get_work_sheet, append_row


def load_data(file_name):
    with open(file_name) as f:
        data = f.read()
        return json.loads(data)


# getting worksheet from sheet or create it with specified column names.


def make_report(email, password, data):
    gc = gspread.login(email, password)
    sh = gc.open_by_key(DOCUMENT_ID)

    work_sheet = get_work_sheet(sh, WORK_SHEET, data.keys())
    append_row(work_sheet, data)


def parse_args(argv):
    parser = argparse.ArgumentParser()
    parser.add_argument('-n', '--name', help='data file path',
                        default=DEFAULT_FILE_PATH)
    parser.add_argument('-e', '--email', help='user email',
                        default="aaa@gmail.com")
    parser.add_argument('-p', '--password', help='user password',
                        default="1234")
    parser.add_argument('-m', '--mode', help='mode type local or global',
                        default='local')
    return parser.parse_args(argv)


def process_results(file_name, email, password, mode):
    data = load_data(file_name)

    if mode == 'local':
        storage = DiskStorage(OUTPUT_FILE)
    else:
        storage = GoogleDocsStorage(DOCUMENT_ID, WORK_SHEET, email, password)

    storage.store(data)


def main(argv):
    opts = parse_args(argv)

    process_results(opts.name,
                    opts.email,
                    opts.password,
                    opts.mode)
    return 0


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