← Back to branch summary

~ubuntu-branches/ubuntu/maverick/python3.1/maverick

« back to all changes in this revision

Viewing changes to Demo/scripts/lpwatch.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-03-23 00:01:27 UTC
  • Revision ID: james.westby@ubuntu.com-20090323000127-5fstfxju4ufrhthq
Tags: upstream-3.1~a1+20090322
Import upstream version 3.1~a1+20090322

Show diffs side-by-side

added added

removed removed

Lines of Context:
Demo/scripts/lpwatch.py
 
1
#! /usr/bin/env python
 
2
 
 
3
# Watch line printer queue(s).
 
4
# Intended for BSD 4.3 lpq.
 
5
 
 
6
import posix
 
7
import sys
 
8
import time
 
9
import string
 
10
 
 
11
DEF_PRINTER = 'psc'
 
12
DEF_DELAY = 10
 
13
 
 
14
def main():
 
15
    delay = DEF_DELAY # XXX Use getopt() later
 
16
    try:
 
17
        thisuser = posix.environ['LOGNAME']
 
18
    except:
 
19
        thisuser = posix.environ['USER']
 
20
    printers = sys.argv[1:]
 
21
    if printers:
 
22
        # Strip '-P' from printer names just in case
 
23
        # the user specified it...
 
24
        for i in range(len(printers)):
 
25
            if printers[i][:2] == '-P':
 
26
                printers[i] = printers[i][2:]
 
27
    else:
 
28
        if 'PRINTER' in posix.environ:
 
29
            printers = [posix.environ['PRINTER']]
 
30
        else:
 
31
            printers = [DEF_PRINTER]
 
32
    #
 
33
    clearhome = posix.popen('clear', 'r').read()
 
34
    #
 
35
    while 1:
 
36
        text = clearhome
 
37
        for name in printers:
 
38
            text = text + makestatus(name, thisuser) + '\n'
 
39
        print(text)
 
40
        time.sleep(delay)
 
41
 
 
42
def makestatus(name, thisuser):
 
43
    pipe = posix.popen('lpq -P' + name + ' 2>&1', 'r')
 
44
    lines = []
 
45
    users = {}
 
46
    aheadbytes = 0
 
47
    aheadjobs = 0
 
48
    userseen = 0
 
49
    totalbytes = 0
 
50
    totaljobs = 0
 
51
    while 1:
 
52
        line = pipe.readline()
 
53
        if not line: break
 
54
        fields = string.split(line)
 
55
        n = len(fields)
 
56
        if len(fields) >= 6 and fields[n-1] == 'bytes':
 
57
            rank = fields[0]
 
58
            user = fields[1]
 
59
            job = fields[2]
 
60
            files = fields[3:-2]
 
61
            bytes = eval(fields[n-2])
 
62
            if user == thisuser:
 
63
                userseen = 1
 
64
            elif not userseen:
 
65
                aheadbytes = aheadbytes + bytes
 
66
                aheadjobs = aheadjobs + 1
 
67
            totalbytes = totalbytes + bytes
 
68
            totaljobs = totaljobs + 1
 
69
            if user in users:
 
70
                ujobs, ubytes = users[user]
 
71
            else:
 
72
                ujobs, ubytes = 0, 0
 
73
            ujobs = ujobs + 1
 
74
            ubytes = ubytes + bytes
 
75
            users[user] = ujobs, ubytes
 
76
        else:
 
77
            if fields and fields[0] != 'Rank':
 
78
                line = string.strip(line)
 
79
                if line == 'no entries':
 
80
                    line = name + ': idle'
 
81
                elif line[-22:] == ' is ready and printing':
 
82
                    line = name
 
83
                lines.append(line)
 
84
    #
 
85
    if totaljobs:
 
86
        line = '%d K' % ((totalbytes+1023)//1024)
 
87
        if totaljobs != len(users):
 
88
            line = line + ' (%d jobs)' % totaljobs
 
89
        if len(users) == 1:
 
90
            line = line + ' for %s' % (list(users.keys())[0],)
 
91
        else:
 
92
            line = line + ' for %d users' % len(users)
 
93
            if userseen:
 
94
                if aheadjobs == 0:
 
95
                    line =  line + ' (%s first)' % thisuser
 
96
                else:
 
97
                    line = line + ' (%d K before %s)' % (
 
98
                                   (aheadbytes+1023)//1024, thisuser)
 
99
        lines.append(line)
 
100
    #
 
101
    sts = pipe.close()
 
102
    if sts:
 
103
        lines.append('lpq exit status %r' % (sts,))
 
104
    return string.joinfields(lines, ': ')
 
105
 
 
106
if __name__ == "__main__":
 
107
    try:
 
108
        main()
 
109
    except KeyboardInterrupt:
 
110
        pass