Guido van Rossum | f06ee5f | 1996-11-27 19:52:01 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 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(): |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 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: |
Collin Winter | 6f2df4d | 2007-07-17 20:59:35 +0000 | [diff] [blame^] | 28 | if 'PRINTER' in posix.environ: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 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' |
Collin Winter | 6f2df4d | 2007-07-17 20:59:35 +0000 | [diff] [blame^] | 39 | print(text) |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 40 | time.sleep(delay) |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 41 | |
Guido van Rossum | 7030d0a | 1991-12-18 13:38:58 +0000 | [diff] [blame] | 42 | def makestatus(name, thisuser): |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 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 |
Collin Winter | 6f2df4d | 2007-07-17 20:59:35 +0000 | [diff] [blame^] | 69 | if user in users: |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 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: |
Neal Norwitz | 3bd844e | 2006-08-29 04:39:12 +0000 | [diff] [blame] | 77 | if fields and fields[0] != 'Rank': |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 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) |
Neal Norwitz | 3bd844e | 2006-08-29 04:39:12 +0000 | [diff] [blame] | 87 | if totaljobs != len(users): |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 88 | line = line + ' (%d jobs)' % totaljobs |
| 89 | if len(users) == 1: |
Collin Winter | 6f2df4d | 2007-07-17 20:59:35 +0000 | [diff] [blame^] | 90 | line = line + ' for %s' % (list(users.keys())[0],) |
Tim Peters | e6ddc8b | 2004-07-18 05:56:09 +0000 | [diff] [blame] | 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, ': ') |
Guido van Rossum | ec758ea | 1991-06-04 20:36:54 +0000 | [diff] [blame] | 105 | |
Johannes Gijsbers | 7a8c43e | 2004-09-11 16:34:35 +0000 | [diff] [blame] | 106 | if __name__ == "__main__": |
| 107 | try: |
| 108 | main() |
| 109 | except KeyboardInterrupt: |
| 110 | pass |