colors = ["red", "orange", "yellow", "green", "blue", "magenta"]

cmd.ray()
cmd.png("image.png")

def _do(c):
  global colors
  index = 0

  for name in cmd.get_names():
    for chain in cmd.get_chains(name):
      print("name:" + name + " chain:" + chain)
      if c == 0:
        cmd.select("chain_" + chain, "chain " + chain)
        cmd.color(colors[index], "chain_" + chain)

      if c == 1:
        cmd.ray()
        cmd.png("image_chain_" + chain)
        cmd.remove("chain_" + chain)

      index = index + 1


_do(0)
_do(1)