def _populate()

in libraries/edge/opensearch/isoresponsebysolr.py [0:0]
48 lines of code
14 McCabe index (conditional complexity)

    def _populate(self, solrDatasetResponse, solrGranuleResponse = None):
        if solrDatasetResponse is not None:
            solrJson = json.loads(solrDatasetResponse)

            logging.debug('dataset count: '+str(len(solrJson['response']['docs'])))

            if len(solrJson['response']['docs']) == 1:
                # ok now populate variables!
                doc = solrJson['response']['docs'][0]

                #self.variables['Dataset_ShortName'] = doc['Dataset-ShortName'][0]
                #self.variables['Dataset_ShortName'] = u'unko'
                
                self.variables['doc'] = doc
                
                # Format dates
                try:
                    self.variables['DatasetCitation_ReleaseDate'] = date.fromtimestamp(float(doc['DatasetCitation-ReleaseDateLong'][0]) / 1000).strftime('%Y%m%d')
                    self.variables['DatasetCoverage_StartTime'] = self._convertTimeLongToISO(doc['DatasetCoverage-StartTimeLong'][0])
                    self.variables['DatasetCoverage_StopTime'] = self._convertTimeLongToISO(doc['DatasetCoverage-StopTimeLong'][0])
                except:
                    pass
                
                try:
                    # Create list of unique dataset sensor
                    self.variables['UniqueDatasetSensor'] = {}
                    for i, x in enumerate(doc['DatasetSource-Sensor-ShortName']):
                        self.variables['UniqueDatasetSensor'][x] = i
                    self.variables['UniqueDatasetSensor'] = list(self.variables['UniqueDatasetSensor'].values())
                    
                    # Create list of unique dataset source
                    self.variables['UniqueDatasetSource'] = {}
                    for i, x in enumerate(doc['DatasetSource-Source-ShortName']):
                        self.variables['UniqueDatasetSource'][x] = i
                    self.variables['UniqueDatasetSource'] = list(self.variables['UniqueDatasetSource'].values())
                    
                    # Replace all none, None values with empty string
                    doc['DatasetParameter-VariableDetail'] = [self._filterString(variableDetail) for variableDetail in doc['DatasetParameter-VariableDetail']]
                    
                    # Current date
                    self.variables['DateStamp'] = datetime.utcnow().strftime('%Y%m%d')
                    
                    # Data format version
                    self.variables['DatasetPolicy_DataFormat_Version'] = self._getDataFormatVersion(doc['DatasetPolicy-DataFormat'][0])
                except Exception as e:
                    logging.debug("Problem generating ISO " + str(e))
                    del self.variables['doc']
                
                if solrGranuleResponse is not None:
                    solrGranuleJson = json.loads(solrGranuleResponse)
                    
                    logging.debug('granule count: '+str(len(solrGranuleJson['response']['docs'])))
                    
                    for doc in solrGranuleJson['response']['docs']:
                        self._populateItem(solrGranuleResponse, doc, None)
                        
                        doc['Granule-StartTimeLong'][0] = self._convertTimeLongToISO(doc['Granule-StartTimeLong'][0])
                        doc['Granule-StopTimeLong'][0] = self._convertTimeLongToISO(doc['Granule-StopTimeLong'][0])
                        doc['Granule-ArchiveTimeLong'][0] = self._convertTimeLongToISO(doc['Granule-ArchiveTimeLong'][0])
                        doc['Granule-CreateTimeLong'][0] = self._convertTimeLongToISO(doc['Granule-CreateTimeLong'][0])
                        
                        # Create dictionary for bounding box extent
                        '''
                        if ('GranuleReal-Value' in doc and 'GranuleReal-DatasetElement-Element-ShortName' in doc):
                            self.variables['GranuleBoundingBox'] = dict(zip(doc['GranuleReal-DatasetElement-Element-ShortName'], doc['GranuleReal-Value']))
                        '''
                        if 'GranuleSpatial-NorthLat' in doc and 'GranuleSpatial-EastLon' in doc and 'GranuleSpatial-SouthLat' in doc and 'GranuleSpatial-WestLon' in doc:
                            self.variables['GranuleBoundingBox'] = dict([('southernmostLatitude', doc['GranuleSpatial-SouthLat'][0]), 
                                                              ('northernmostLatitude', doc['GranuleSpatial-NorthLat'][0]),
                                                              ('westernmostLongitude', doc['GranuleSpatial-WestLon'][0]),
                                                              ('easternmostLongitude', doc['GranuleSpatial-EastLon'][0])])
                        break
                        
                    self.variables['granules'] = solrGranuleJson['response']['docs']