Merge pull request #1 from dheles/delete_bitstreams

Add script to delete an item's bitstreams

Author: ehanson8

.gitignore

@@ -52,4 +52,16 @@ data/*
 !data/.keep
 .profile
 *.csv
-*.json
+*.json
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Rope project settings
+.ropeproject

data/.keep

@@ -0,0 +1,72 @@
+import json
+import requests
+import secrets
+import time
+import csv
+from datetime import datetime
+
+secretsVersion = raw_input('To edit production server, enter the name of the secrets file: ')
+if secretsVersion != '':
+    try:
+        secrets = __import__(secretsVersion)
+        print 'Editing Production'
+    except ImportError:
+        print 'Editing Stage'
+else:
+    print 'Editing Stage'
+
+baseURL = secrets.baseURL
+email = secrets.email
+password = secrets.password
+filePath = secrets.filePath
+verify = secrets.verify
+
+requests.packages.urllib3.disable_warnings()
+
+itemHandle = raw_input('Enter item handle: ')
+
+
+startTime = time.time()
+data = json.dumps({'email':email,'password':password})
+header = {'content-type':'application/json','accept':'application/json'}
+session = requests.post(baseURL + '/rest/login', headers=header, verify=verify, data=data).content
+headerAuth = {'content-type':'application/json','accept':'application/json', 'rest-dspace-token':session}
+print 'authenticated'
+
+bitstreamList = []
+endpoint = baseURL + '/rest/handle/' + itemHandle
+item = requests.get(endpoint, headers=headerAuth, verify=verify).json()
+itemID = item['id']
+print 'itemID = %s' % itemID
+bitstreams = ''
+url = baseURL + '/rest/items/' + str(itemID) + '/bitstreams?expand=bitstreams'
+bitstreams = requests.get(url, headers=headerAuth, verify=verify)
+while bitstreams.status_code != 200:
+    time.sleep(5)
+    bitstreams = requests.get(url, headers=headerAuth, verify=verify)
+bitstreams = bitstreams.json()
+print 'found %d bitstreams' % len(bitstreams)
+for k in range(0, len(bitstreams)):
+    bitstreamID = bitstreams[k]['id']
+    bitstreamList.append(bitstreamID)
+elapsedTime = time.time() - startTime
+m, s = divmod(elapsedTime, 60)
+h, m = divmod(m, 60)
+print 'Bitstreams list creation time: ', '%d:%02d:%02d' % (h, m, s)
+print(bitstreamList)
+
+f=csv.writer(open(filePath + 'deletedBitstreams' + datetime.now().strftime('%Y-%m-%d %H.%M.%S') + '.csv', 'wb'))
+f.writerow(['bitstreamID'] + ['delete'])
+for number, bitstreamID in enumerate(bitstreamList):
+    bitstreamsRemaining = len(bitstreamList) - number
+    print 'Bitstreams remaining: ', bitstreamsRemaining, 'bitstreamID: ', bitstreamID
+    delete = requests.delete(baseURL + '/rest/bitstreams/' + str(bitstreamID), headers=headerAuth, verify=verify)
+    print delete
+    f.writerow([bitstreamID] + [delete])
+
+logout = requests.post(baseURL + '/rest/logout', headers=headerAuth, verify=verify)
+
+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)