Added function for getting latestEdition

Author: Karen Coombs

handler.py

@@ -48,5 +48,13 @@ def addLBDs(event, context):
     csv_read = process_data.addLBDs(processConfig, csv_read)
     handle_files.saveFileToBucket(fileInfo['bucket'], fileInfo['key'] + "_updated", csv_read)   
 
-    return saveFile(bucket, key + "_updated", csv_read) 
+    return saveFile(bucket, key + "_updated", csv_read)
+
+def getLatestEdition(event, context):
+    item_file = handle_files.readFilesFromBucket(event)
+    csv_read = handle_files.loadCSV(item_file)
+    csv_read = process_data.getLatestEdition(processConfig, csv_read)
+    handle_files.saveFileToBucket(fileInfo['bucket'], fileInfo['key'] + "_updated", csv_read)
+
+    return saveFile(bucket, key + "_updated", csv_read)
 

processSheet.py

@@ -12,7 +12,7 @@ def processArgs():
     try:
         parser = argparse.ArgumentParser()
         parser.add_argument('--itemFile', required=True, help='File you want to process')
-        parser.add_argument('--operation', required=True, choices=['getCurrentOCLCNumbers', 'retrieveMergedOCLCNumbers', 'setHoldingsbyOCLCNumber', 'deleteHoldingsbyOCLCNumber', 'addLBDs'], help='Operation to run: getCurrentOCLCNumbers, retrieveMergedOCLCNumbers, setHoldingsbyOCLCNumber, deleteHoldingsbyOCLCNumber, addLBDs')    
+        parser.add_argument('--operation', required=True, choices=['getCurrentOCLCNumbers', 'retrieveMergedOCLCNumbers', 'setHoldingsbyOCLCNumber', 'deleteHoldingsbyOCLCNumber', 'addLBDs', 'getLatestEdition'], help='Operation to run: getCurrentOCLCNumbers, retrieveMergedOCLCNumbers, setHoldingsbyOCLCNumber, deleteHoldingsbyOCLCNumber, addLBDs')
         parser.add_argument('--outputDir', required=True, help='Directory to save output to')        
 
         args = parser.parse_args()
@@ -43,8 +43,10 @@ def process(args):
         elif operation == "deleteHoldingsbyOCLCNumber":
             csv_read = process_data.deleteHoldingsbyOCLCNumber(processConfig, csv_read)
         elif operation == "addLBDs":
-            csv_read = process_data.addLBDs(processConfig, csv_read)            
-    
+            csv_read = process_data.addLBDs(processConfig, csv_read)
+        elif operation == "getLatestEdition":
+            csv_read = process_data.getLatestEdition(processConfig, csv_read)
+
         return handle_files.saveFileLocal(csv_read, output_dir)
 
     except BaseException as err:

src/make_requests.py

@@ -199,3 +199,33 @@ def findBibMatch(config, record):
     except requests.exceptions.HTTPError as err:
         status = "failed"
     return pd.Series([oclcNumber, status])
+
+def getLatestEdition(config, oclcNumber):
+    oauth_session = config.get('oauth-session')
+    requestURL = config.get('metadata_service_url') + "/brief-bibs/" + str(oclcNumber) + "/other-editions?inLanguage=eng&limit=1&orderBy=publicationDateDesc"
+    try:
+        r = requests.get(requestURL, headers={"Accept":"application/json"})
+        r.raise_for_status
+        try:
+            result = r.json()
+            if result.get('briefRecords'):
+                if (str(oclcNumber) == result.get('briefRecords')[0].get('oclcNumber')):
+                    isLatestEdition = "true"
+                elif (result.get('briefRecords')[0].get('mergedOclcNumbers') and str(oclcNumber) in result.get('briefRecords')[0].get('mergedOclcNumbers')):
+                    isLatestEdition = "true"
+                else:
+                    isLatestEdition = "false"
+                latestEditionOCN = result.get('briefRecords')[0].get('oclcNumber')
+                latestEditionYear = result.get('briefRecords')[0].get('date')
+            else:
+                isLatestEdition = ""
+                latestEditionOCN = ""
+                latestEditionYear = ""
+        except json.decoder.JSONDecodeError:
+            isLatestEdition = ""
+            latestEditionOCN = ""
+            latestEditionYear = ""
+    except requests.exceptions.HTTPError as err:
+        status = "failed"
+
+    return pd.Series([isLatestEdition, latestEditionOCN, latestEditionYear, status])

src/process_data.py

@@ -21,5 +21,8 @@ def deleteHoldingsbyOCLCNumber(processConfig, csv_read):
 
 def addLBDs(processConfig, csv_read):  
     csv_read[['oclcnumber', 'lbd_number', 'status']] = csv_read.apply (lambda row: addLBD(row['oclcNumber'], row['note']), axis=1)    
-    return csv_read    
-        
+    return csv_read
+
+def getLatestEdition(processConfig, csv_read):
+    csv_read[['oclcnumber', 'isLatestEdition', 'latestEditionOCN', 'year', 'status']] = csv_read.apply (lambda row: getLatestEdition(row['oclcNumber']), axis=1)
+    return csv_read