File: extend_demo01.py

package info (click to toggle)
pymol 3.1.0%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 74,084 kB
  • sloc: cpp: 482,660; python: 89,328; ansic: 29,512; javascript: 6,792; sh: 84; makefile: 25
file content (27 lines) | stat: -rw-r--r-- 573 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

# Python script extend_demo01.py

# A recipe for creating a new PyMOL command which calls your own PyMOL code

from pymol import cmd

# step 1: define your Python function

def my_color(color="red",sele="name CA"):
    print("I am coloring "+color+ " all atoms in selection: "+sele)
    cmd.color(color, sele)

# step 2: add you command to the PyMOL command language

cmd.extend("my_color", my_color)

# step 3: run this script from within PymOL

#PyMOL> run extend_demo01.py

# step 4: try it out!

#PyMOL> my_color
#PyMOL> my_color yellow
#PyMOL> my_color blue, elem c