/usr/share/checkbox/scripts/xrandr_cycle is in checkbox 0.13.7.
This file is owned by root:root, with mode 0o755.
The actual contents of the file can be viewed below.
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 | #!/usr/bin/env python
import subprocess
import argparse
import tarfile
import shutil
import time
import sys
import os
import re
parser = argparse.ArgumentParser()
parser.add_argument('--keyword', default='',
help='A keyword to distinguish the screenshots taken in this run of the script')
args = parser.parse_args()
device_context = '' # track what device's modes we are looking at
modes = [] # keep track of all the devices and modes discovered
current_modes = [] # remember the user's current settings for cleanup later
failures = 0 # count the number of failed modesets
failure_messages = [] # remember which modes failed
success_messages = [] # remember which modes succeeded
# Run xrandr and ask it what devices and modes are supported
xrandrinfo = subprocess.Popen('xrandr -q', shell=True, stdout=subprocess.PIPE)
output = xrandrinfo.communicate()[0].split('\n')
# The results from xrandr are given in terms of the available display devices.
# One device can have zero or more associated modes. Unfortunately xrandr
# indicates this through indentation and is kinda wordy, so we have to keep
# track of the context we see mode names in as we parse the results.
for line in output:
# I haven't seen any blank lines in xrandr's output in my tests, but meh
if line == '':
break;
# luckily the various data from xrandr are separated by whitespace...
foo = line.split()
# Check to see if the second word in the line indicates a new context
# -- if so, keep track of the context of the device we're seeing
if len(foo) >= 2: # throw out any weirdly formatted lines
if foo[1] == 'disconnected':
# we have a new context, but it should be ignored
device_context = ''
if foo[1] == 'connected':
# we have a new context that we want to test
device_context = foo[0]
elif device_context != '': # we've previously seen a 'connected' device
# mode names seem to always be of the format [horiz]x[vert]
# (there can be non-mode information inside of a device context!)
if foo[0].find('x') != -1:
modes.append( (device_context, foo[0]) )
# we also want to remember what the current mode is, which xrandr
# marks with a '*' character, so we can set things back the way
# we found them at the end:
if foo[1].find('*') != -1:
current_modes.append( (device_context, foo[0]) )
# Now we have a list of the modes we need to test. So let's do just that.
profile_path = os.environ['HOME'] + '/.shutter/profiles/'
screenshot_path = os.environ['CHECKBOX_DATA'] + 'xrandr_screens'
if args.keyword: screenshot_path = screenshot_path + '_' + args.keyword
regex = re.compile(r'filename="[^"\r\n]*"')
# Keep the shutter profile in place before starting
try:
os.makedirs(profile_path)
except OSError:
pass
try:
os.makedirs(screenshot_path)
except OSError:
pass
try:
shutil.copy(os.environ['CHECKBOX_SHARE'] + '/data/settings/shutter.xml',
profile_path)
except IOError:
pass
try:
old_profile = open(profile_path + 'shutter.xml', 'r')
content = old_profile.read()
new_profile = open(profile_path + 'shutter.xml', 'w')
# Replace the folder name with the desired one
new_profile.write(re.sub(r'folder="[^"\r\n]*"',
'folder="%s"' % screenshot_path, content))
new_profile.close()
old_profile.close()
except:
pass
for mode in modes:
cmd = 'xrandr --output ' + mode[0] + ' --mode ' + mode[1]
retval = subprocess.call(cmd, shell=True)
if retval != 0:
failures = failures + 1
message = 'Failed to set mode ' + mode[1] + ' for output ' + mode[0]
failure_messages.append(message)
else:
# Update shutter profile to save the image as the right name
mode_string = mode[0] + '_' + mode[1]
try:
old_profile = open(profile_path + 'shutter.xml', 'r')
content = old_profile.read()
new_profile = open(profile_path + 'shutter.xml', 'w')
new_profile.write(regex.sub('filename="%s"' % mode_string, content))
new_profile.close()
old_profile.close()
shuttercmd = ['shutter','--profile=shutter','--full','-e']
retval = subprocess.call(shuttercmd, shell=False)
if retval != 0:
print """Could not capture screenshot -
you may need to install the package 'shutter'."""
except:
print """Could not configure screenshot tool -
you may need to install the package 'shutter',
or check that %s exists.""" % profile_path
message = 'Set mode ' + mode[1] + ' for output ' + mode[0]
success_messages.append(message)
time.sleep(3) # let the hardware recover a bit
# Put things back the way we found them
for mode in current_modes:
cmd = 'xrandr --output ' + mode[0] + ' --mode ' + mode[1]
subprocess.call(cmd, shell=True)
# Tar up the screenshots for uploading
try:
with tarfile.open(screenshot_path + '.tgz', 'w:gz') as screen_tar:
for screen in os.listdir(screenshot_path):
screen_tar.add(screenshot_path + '/' + screen, screen)
except:
pass
# Output some fun facts and knock off for the day
for message in failure_messages:
print >> sys.stderr, message
for message in success_messages:
print message
if failures != 0:
exit(1)
else:
exit(0)
|