Exemple #1
0
    def handle(self, *args, **options):

        # check for necessary parameters
        if 'filename' not in options:
            raise ValueError("no filename given")
        if 'scenario_id' not in options:
            raise ValueError("no scenario_id given")

        filename = options['filename']
        logger.info('starting to import file {}'.format(filename))

        # get scenario and root tile
        try:
            scenario_id = options['scenario_id']
            scenario = Scenario.objects.get(pk=scenario_id)
        except ObjectDoesNotExist:
            logger.error('invalid scenario id: {}'.format(scenario_id))
            raise ValueError(
                'Scenario with id {} does not exist'.format(scenario_id))

        order = 0
        with fiona.open(filename) as shapefile:
            for feature in shapefile:
                coordinates = feature['geometry']['coordinates']
                geometry = Point(coordinates)
                name = feature['properties'][NAME_FIELD]

                location = Location()
                location.scenario = scenario
                location.name = name
                location.direction = 0  # TODO: currently looking north all the time
                location.location = geometry
                location.order = order
                location.save()

                order += 1  # currently there is no order field in the shapefile

        logger.info("finished import to scenario_id {}".format(scenario_id))