Merge pull request #3 from ehanson8/master

update

Author: ehanson8

README.md

@@ -10,6 +10,9 @@ All of these scripts require a secrets.py file in the same directory that must c
 
 ## Scripts
 
+#### [getArchivalObjectsByResource.py](/getArchivalObjectsByResource.py)
+A GET script to extract all of the archival objects associated with a particular resource. Upon running the script, you will be prompted enter the resource ID (just the number, not the full URI).
+
 #### [getArrayPropertiesFromAgentsPeopleCSV.py](/getArrayPropertiesFromAgentsPeopleCSV.py)
 This GET script retrieves specific properties, including proprerties that have arrays as values, from the JSON of ArchivesSpace agent_people records. In this example, the 'dates_of existence' property contains an array that must be iterated over. This requires a second level of iteration with 'for j in range (...)' on line 20, which is in addition to the iteration function 'for i in range (...)' on line 19, which was also found in the getPropertiesFromAgentsPeopleCSV.py script. As with the previous script, it also writes the properties' values into a CSV file which is specified in variable 'f' on line 17.
 
@@ -29,6 +32,3 @@ This POST script will post new records to a generic API endpoint based the recor
 
 #### [postOverwrite.py](/postOverwrite.py)
 This POST script will overwrite existing ArchivesSpace records based the 'uri' and can be used with any ArchivesSpace record type (e.g. resource, accession, subject, agent_people, agent_corporate_entity, archival_object, etc.). It requires a properly formatted JSON file (specified where [JSON File] appears in the 'records' variable on line 13) for the particular ArchivesSpace record type you are trying to post. 
-
-#### [searchArchivalObjectsByResource.py](/searchArchivalObjectsByResource.py)
-A GET script to extract all of the archival objects associated with a particular resource. To set the resource you want to search for, adjust the 'resourceNumber' variable on line 13

getArchivalObjectsByResource.py

@@ -0,0 +1,48 @@
+import json
+import requests
+import secrets
+import time
+
+startTime = time.time()
+
+def findKey(d, key):
+    if key in d:
+        yield d[key]
+    for k in d:
+        if isinstance(d[k], list):
+            for i in d[k]:
+                for j in findKey(i, key):
+                    yield j
+
+baseURL = secrets.baseURL
+user = secrets.user
+password = secrets.password
+
+resourceID= raw_input('Enter resource ID: ')
+
+auth = requests.post(baseURL + '/users/'+user+'/login?password='+password).json()
+session = auth["session"]
+headers = {'X-ArchivesSpace-Session':session, 'Content_Type':'application/json'}
+
+endpoint = '/repositories/3/resources/'+resourceID+'/tree'
+
+output = requests.get(baseURL + endpoint, headers=headers).json()
+
+archivalObjects = []
+for value in findKey(output, 'record_uri'):
+    if 'archival_objects' in value:
+        archivalObjects.append(value)
+
+records = []
+for archivalObject in archivalObjects:
+    output = requests.get(baseURL + archivalObject, headers=headers).json()
+    records.append(output)
+
+f=open('archivalObjects.json', 'w')
+json.dump(records, f)
+f.close()
+
+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)