from nullptr.roles.trader import assign_trader
from nullptr.roles.probe import assign_probe
from nullptr.roles.siphon import assign_siphon
from nullptr.roles.hauler import assign_hauler
from nullptr.roles.surveyor import assign_surveyor
from nullptr.roles.miner import assign_miner

def assign_mission(c, s):
  if s.role == 'trader':
    assign_trader(c, s)
  elif s.role == 'probe':
    assign_probe(c, s)
  elif s.role == 'siphon':
    assign_siphon(c, s)
  elif s.role == 'hauler':
    assign_hauler(c, s)
  elif s.role == 'surveyor':
    assign_surveyor(c, s)
  elif s.role == 'miner':
    assign_miner(c, s)