update_doi.py

def update_doi(socrata_4x4, temp_table, draft=True):
    """Updates existing DOI's metadata"""

    url = 'https://api.datacite.org/dois/'
    datacite_user = os.environ['datacite_user']
    datacite_pass = os.environ['datacite_pass']
    print('Updating: ' + socrata_4x4)

    doi_assets = pd.read_json(assets_filename)

    payload, doi, xml, metadata = assemble_payload(socrata_4x4, temp_table, draft, update=True)

    r = requests.put('{}{}'.format(url, doi), json=payload, auth=(datacite_user, datacite_pass))
    response_dict = json.loads(r.content)
    print('DataCite Response:')
    print(response_dict)
    if 'data' in response_dict:
        # TODO add git commit of file to push to github repo
        # if successful update doi table
        doi_assets.loc[doi_assets['socrata_4x4'] == socrata_4x4, 'metadata_xml'] = xml
        doi_assets.loc[doi_assets['socrata_4x4'] == socrata_4x4, 'last_updated'] = str(datetime.datetime.now())
        doi_assets.to_json(filename)
        return True
    else:
        print('Error, could not update: ' + socrata_4x4)
        return False