Add --safe argument to spawn_npc.py

This commit is contained in:
nsubiron 2018-11-03 12:28:14 +01:00
parent dca0bf86b0
commit 7891a0b138
1 changed files with 8 additions and 0 deletions

View File

@ -53,6 +53,10 @@ def main():
default=2.0,
type=float,
help='delay in seconds between spawns (default: 2.0)')
argparser.add_argument(
'--safe',
action='store_true',
help='avoid spawning vehicles prone to accidents')
args = argparser.parse_args()
actor_list = []
@ -64,6 +68,10 @@ def main():
world = client.get_world()
blueprints = world.get_blueprint_library().filter('vehicle.*')
if args.safe:
blueprints = [x for x in blueprints if int(x.get_attribute('number_of_wheels')) == 4]
blueprints = [x for x in blueprints if not x.id.endswith('isetta')]
def try_spawn_random_vehicle_at(transform):
blueprint = random.choice(blueprints)
if blueprint.has_attribute('color'):