#!/usr/bin/env python
# coding: utf-8

# In[ ]:


from pymol import cmd

cmd.fetch("1a02")
cmd.remove("not (chain F+J or polymer.nucleic)")
cmd.color("bluewhite","polymer.nucleic")
cmd.scene("s1","store")

cmd.hide("everything")
cmd.show("ribbon", "1a02")
cmd.select("set1"," resn DA+DT+DC+DG and name O3'+O4'+O5'")
cmd.color("firebrick", "set1")
cmd.show("spheres", "set1")
cmd.scene("s2","store")

cmd.hide("spheres", "set1")
cmd.select("set2"," resn DA+DT+DC+DG and name OP1+OP2")
cmd.color("tv_orange", "set2")
cmd.show("spheres", "set2")
cmd.scene("s3","store")

cmd.hide("spheres", "set2")
cmd.select("set3"," resn DA+DT+DC+DG and name N1+N2+N3+N4+N6+N7+N9")
cmd.color("tv_blue", "set3")
cmd.show("spheres", "set3")
cmd.scene("s4","store")

cmd.set("scene_animation", 1)
cmd.set("scene_animation_duration", 1)
cmd.set("movie_fps", 30)

cmd.mset("1 x120")

cmd.mdo(1, "scene s1, recall")
cmd.mdo(30, "scene s2, recall")
cmd.mdo(60, "scene s3, recall")
cmd.mdo(90, "scene s4, recall")

cmd.mplay()