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 posix.environ.has_key('PRINTER'):
  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 users.has_key(user):
  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' % (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