Esempio n. 1
0
 def __init__(self, country, year, entryPath): # complete
     spreadsheetID = entryPath + "ipdConflict" + country + str(year)\
     + '.csv'
     self.table = Spreadsheet()
     if self.table.isPresent(spreadsheetID):
         self.table.open(spreadsheetID)
     else:
         self.table.make(spreadsheetID)
Esempio n. 2
0
class Entries:
    def __init__(self, country, year, entryPath): # complete
        spreadsheetID = entryPath + "ipdConflict" + country + str(year)\
        + '.csv'
        self.table = Spreadsheet()
        if self.table.isPresent(spreadsheetID):
            self.table.open(spreadsheetID)
        else:
            self.table.make(spreadsheetID)

    def getNumEntries(self): # complete
        return self.table.numRows() - 1 # return number of rows - 1 (excluding column labels)

    def getEntry(self, pk): # complete
        if pk > self.table.numRows() or pk < 2: # if the primary key is out of bounds, raise exception
            raise ValueError("pk out of bounds")

        row = self.table.getRow(pk)
        entry = self.__processRow(row)
        return entry
    
    def makeEntry(self, location): # complete
        row = self.__getRow(location)
        self.table.append(row)

    def changeEntry(self, pk, location): # complete
        if self.table.numRows() < pk or pk < 2:
            raise ValueError("pk out of bounds")
        rowNumToChange = pk
        entry = self.__getRow(location)
        self.table.modify(rowNumToChange, entry)

    def __getRow(self, location): # complete
        # make an array of items to insert into row, pass that array to spreadsheet while specifying row 
        # order: resource, minetype, locationname, stdmeasure, annlLocCapacity in stdMeasure
        #        Jan ppu, yrlLocValue, capOfLocInKg, ppKilogram, long, lat, precisCode
        #        arbit field 1, arbit field 2, ...

        row = []

        row.extend([location.getResource(), location.getMineType(), location.getLocationName()])
        row.extend([location.getStdMeasure(), location.getAnnlLocCapacity()])
        row.extend([location.getPpu(), location.getYrlyLocValue()])
        row.extend([location.getKgCapacity(), location.getPpk(), location.getLong()])
        row.extend([location.getLat(), location.getPrecisCode()])
        # must implement arbit fields!!
        arbitFields = location.getArbitFields()

        keys = arbitFields.keys().sort()
        
        if keys != None:
            for key in keys:
                row.extend(key + '!' +  arbitFields[key])
        
        return row

    def __processRow(self, row): # complete #given row, returns location
        locToReturn = Location()
        l = locToReturn

        functionsToCall = [l.setResource, l.setMineType, l.setLocName, l.setStdMeasure,\
                           l.setAnnlLocCapacity, l.setPpu, l.setLongLat, l.setPrecisCode]
        
        i = 0

        while i <= 5:
            functionsToCall[i](row[i])
            i += 1

        l.setLongLat(row[9], row[10])
        l.setPrecisCode(row[11])

        i = 12

        while i < len(row):
            arbitKeyValue = row[i]
            key, value = arbitKeyValue.partition('!')[0], arbitKeyValue.partition('!')[2]
            l.setArbitField(key, value)
            
        return locToReturn        

    def save(self):
        self.table.save()

    def close(self):
        self.table.close()