Random vehicle swapping.

[koperkapel.git] / koperkapel / vehicles / base.py

diff --git a/koperkapel/vehicles/base.py b/koperkapel/vehicles/base.py
index 2ae347cd54e541187b46cd39729da98a7cc86f2b..9eb3eb0cfe714f74f9517d934da6f8e73ba5c7d6 100644 (file)
--- a/koperkapel/vehicles/base.py
+++ b/koperkapel/vehicles/base.py
@@ -1,6 +1,7 @@
 """ Base class for vehicles.  """
 
 import math
+import random
 from itertools import chain, islice, repeat
 from pygame.constants import BLEND_RGBA_MULT
 from pgzero.loaders import images
@@ -73,6 +74,10 @@ class Vehicle:
     def register(cls, vehicle_cls):
         cls._vehicle_types[vehicle_cls.__name__.lower()] = vehicle_cls
 
+    @classmethod
+    def random(cls):
+        return random.choice(list(cls._vehicle_types.keys()))
+
     @classmethod
     def register_all(cls):
         from .walking import Walking