File: extend_demo01.py

package info (click to toggle)
pymol 2.4.0%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 43,312 kB
  • sloc: cpp: 480,106; python: 79,860; ansic: 28,343; javascript: 6,792; sh: 47; makefile: 30; csh: 8
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