getAccessionUDFs.py

import requests
import time
import csv

secretsVersion = input('To edit production server, enter the name of the \
secrets file: ')
if secretsVersion != '':
    try:
        secrets = __import__(secretsVersion)
        print('Editing Production')
    except ImportError:
        secrets = __import__('secrets')
        print('Editing Development')
else:
    print('Editing Development')

startTime = time.time()

baseURL = secrets.baseURL
user = secrets.user
password = secrets.password
repository = secrets.repository

auth = requests.post(baseURL + '/users/' + user + '/login?password='
                     + password).json()
session = auth['session']
headers = {'X-ArchivesSpace-Session': session,
           'Content_Type': 'application/json'}

endpoint = '/repositories/' + repository + '/accessions?all_ids=true'

ids = requests.get(baseURL + endpoint, headers=headers).json()

udfs = []
for id in ids:
    print(id)
    endpoint = '/repositories/' + repository + '/accessions/' + str(id)
    output = requests.get(baseURL + endpoint, headers=headers).json()
    try:
        userDefined = output['user_defined']
        for k, v in userDefined.items():
            if k not in udfs:
                udfs.append(k)
    except ValueError:
        userDefined = ''
udfs.sort()
udfsHeader = ['title', 'uri'] + udfs
f = csv.writer(open('accessionsUdfs.csv', 'w'))
f.writerow(udfsHeader)

for id in ids:
    print(id)
    endpoint = '/repositories/' + repository + '/accessions/' + str(id)
    output = requests.get(baseURL + endpoint, headers=headers).json()
    title = output['title']
    uri = output['uri']
    accessionUdfs = []
    for udf in udfs:
        try:
            keyValue = udf + '|' + output['user_defined'][udf]
        except ValueError:
            keyValue = udf + '|'
        accessionUdfs.append(keyValue)
    accessionUdfs.sort()
    accessionUdfsUpdated = []
    for accessionUdf in accessionUdfs:
        edited = accessionUdf[accessionUdf.index('|') + 1:]
        accessionUdfsUpdated.append(edited)
    accessionUdfsRow = [title, uri] + accessionUdfsUpdated
    f.writerow(accessionUdfsRow)

elapsedTime = time.time() - startTime
m, s = divmod(elapsedTime, 60)
h, m = divmod(m, 60)
print('Total script run time: ', '%d:%02d:%02d' % (h, m, s))