def test_read_csv():
assert read_csv('tests/fixtures/framework_results.csv') == [
['123', 'Supplier Name', 'pass'],
['234', 'Supplier Name', 'fail'],
['345', 'Supplier Name', ' PASS'],
['456', 'Supplier Name', ' FAIL'],
['567', 'Supplier Name', ' Yes'],
['Company Name', 'Supplier Name', 'pass'],
['678', 'Supplier Name', 'PasS'],
]
Example:
./generate-framework-agreement-data.py lots.csv declarations.csv framework-outputs localhost:5000 myToken
"""
import sys
sys.path.insert(0, '.')
from docopt import docopt
from dmscripts.generate_framework_agreement_data import check_lots_csv, read_csv, \
check_declarations_csv, build_framework_agreements
from dmutils.apiclient import DataAPIClient
if __name__ == '__main__':
arguments = docopt(__doc__)
supplier_lots = read_csv(arguments['<lots_file>'])
supplier_declarations = read_csv(arguments['<declaration_file>'])
# Remove first column titles row, if it exists
if supplier_lots[0][0] == "Digital Marketplace ID":
del supplier_lots[0]
if supplier_declarations[0][0] == "Digital Marketplace ID":
del supplier_declarations[0]
lot_check = check_lots_csv(supplier_lots)
if not lot_check[0]:
sys.exit("Lots CSV is invalid: {}".format(lot_check[1]))
declaration_check = check_declarations_csv(supplier_declarations)
if not declaration_check[0]:
