X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=koperkapel%2Fvehicles%2Fbase.py;h=9eb3eb0cfe714f74f9517d934da6f8e73ba5c7d6;hb=e275a42b8cc75a4b3989096171b4dc17939b9de0;hp=2ae347cd54e541187b46cd39729da98a7cc86f2b;hpb=b0e9b722445f62943284044e1b5f9638db89a4e7;p=koperkapel.git diff --git a/koperkapel/vehicles/base.py b/koperkapel/vehicles/base.py index 2ae347c..9eb3eb0 100644 --- 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