33 lines
928 B
Python
33 lines
928 B
Python
from nullptr.missions.survey import SurveyMission
|
|
from nullptr.missions.mine import MiningMission
|
|
from nullptr.missions.trade import TradeMission
|
|
from nullptr.missions.travel import TravelMission
|
|
from nullptr.missions.probe import ProbeMission
|
|
from nullptr.missions.idle import IdleMission
|
|
from nullptr.missions.siphon import SiphonMission
|
|
from nullptr.missions.haul import HaulMission
|
|
from nullptr.missions.sit import SitMission
|
|
|
|
def get_mission_class( mtype):
|
|
types = {
|
|
'survey': SurveyMission,
|
|
'mine': MiningMission,
|
|
'trade': TradeMission,
|
|
'travel': TravelMission,
|
|
'probe': ProbeMission,
|
|
'idle': IdleMission,
|
|
'siphon': SiphonMission,
|
|
'haul': HaulMission,
|
|
'sit': SitMission,
|
|
|
|
}
|
|
if mtype not in types:
|
|
raise ValueError(f'invalid mission type {mtype}')
|
|
return types[mtype]
|
|
|
|
def create_mission(mtype, ship, c):
|
|
typ = get_mission_class(mtype)
|
|
m = typ(ship, c)
|
|
return m
|
|
|