annotate lib/localdisplay.py @ 886:2d055bce2e60

Xorg display number fetcher for use by xidle and dpms Ignore-this: eaef4d7fa8277e88c65d83343a46f866 darcs-hash:20130529074436-312f9-82b723dfcb487deee25ca2beab344631b09c0243
author drewp <drewp@bigasterisk.com>
date Wed, 29 May 2013 00:44:36 -0700
parents
children 9155e9a605ec
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
886
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
1 import psutil, os
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
2
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
3 def setDisplayToLocalX():
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
4 """
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
5 set DISPLAY env var in this process to the id of the X process on localhost
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
6 """
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
7 for pid in psutil.get_pid_list():
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
8 try:
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
9 proc = psutil.Process(pid)
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
10 if proc.exe not in ['/usr/bin/Xorg', '/usr/bin/X']:
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
11 continue
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
12 except (psutil.error.AccessDenied, psutil.error.NoSuchProcess):
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
13 continue
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
14 display = proc.cmdline[1]
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
15 assert display.startswith(':'), display
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
16 os.environ['DISPLAY'] = display
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
17 break
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
18 else:
2d055bce2e60 Xorg display number fetcher for use by xidle and dpms
drewp <drewp@bigasterisk.com>
parents:
diff changeset
19 raise ValueError("didn't find an Xorg process")