Exemple #1
0
    def process_city(self, row):
        city_name = row[self.slices[self.indices['city']]].strip()
        cities = City.execute_select({'name': city_name})

        if cities:
            return cities[0].id

        if self.indices['country']:
            country_code = row[self.slices[self.indices['country']]].strip()
            country_id = Country.select_attrs(['id'], {'code': country_code})[0]
        else:
            country_id = 1

        return City.execute_insert([{'name': city_name, 'country_id': country_id}])[0]