-
Notifications
You must be signed in to change notification settings - Fork 4
/
updatePersnames.py
59 lines (54 loc) · 3.21 KB
/
updatePersnames.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import requests, csv, json, urllib.parse, runtime, authenticate
viafURL = 'http://viaf.org/viaf/search?query=local.personalNames+%3D+%22'
# # print instructions
print('This script queries existing person agent records in ArchivesSpace with the source of "viaf" and updates them with the proper/updated name form from VIAF (if one exists) and appends the VIAF URI to the existing records. Please note: This is a PROOF OF CONCEPT script, and should not be used in production settings without thinking this through!')
input('Press Enter to continue...')
# This is where we connect to ArchivesSpace. See authenticate.py
baseURL, headers = authenticate.login()
# search AS for person agents with source "viaf"
query = '/search?page=1&filter={"query":{"jsonmodel_type":"boolean_query","op":"AND","subqueries":[{"jsonmodel_type":"field_query","field":"primary_type","value":"agent_person","literal":true},{"jsonmodel_type":"field_query","field":"source","value":"viaf","literal":true}]}}'
ASoutput = requests.get(baseURL + query, headers=headers).json()
print('Found ' + str(len(ASoutput['results'])) + ' agents.')
# grab uri out of agent
for person in ASoutput['results']:
uri = person['uri']
personRecord = requests.get(baseURL + uri, headers=headers).json()
lockVersion = str(personRecord['lock_version'])
primary_name = personRecord['names'][0]['primary_name']
try:
secondary_name = personRecord['names'][0]['rest_of_name']
except:
secondary_name = ''
try:
dates = personRecord['names'][0]['dates']
except:
dates = ''
searchName = primary_name + ', ' + secondary_name + ', ' + dates
nameEdited = urllib.parse.quote(searchName.strip())
url = viafURL+nameEdited+'%22+and+local.sources+%3D+%22lc%22&sortKeys=holdingscount&maximumRecords=1&httpAccept=application/rdf+json'
# first need to treat the response as text since we get an xml resopnse (with json embedded inside)
response = requests.get(url).text
try:
response = response[response.index('<recordData xsi:type="ns1:stringOrXmlFragment">')+47:response.index('</recordData>')].replace('"','"')
response = json.loads(response)
properName = response['mainHeadings']['data'][0]['text']
nameArray = properName.split(',')
properPrimary = nameArray[0]
try:
properSecondary = nameArray[1]
except:
properSecondary = ''
try:
properDates = nameArray[2]
except:
properDates = ''
viafid = response['viafID']
except:
label = ''
viafid = ''
if viafid != '':
links = json.loads(requests.get('http://viaf.org/viaf/'+viafid+'/justlinks.json').text)
viafid = 'http://viaf.org/viaf/'+viafid
toPost = '{"lock_version": ' + lockVersion + ',"names": [{"primary_name":"' + properPrimary.strip() + '","rest_of_name":"' + properSecondary.strip() + '","dates":"' + properDates.strip() + '","sort_name":"' + properName + '","authorized":true, "is_display_name": true, "source": "viaf", "rules": "dacs", "name_order": "inverted", "jsonmodel_type": "name_person", "authority_id":"' + viafid + '"}]}'
post = requests.post(baseURL + uri, headers=headers, data=toPost).json()
print(post)