Home | History | Annotate | Download | only in examples
      1 #!/usr/bin/env python
      2 
      3 """This displays uptime information using uptime. This is redundant,
      4 but it demonstrates expecting for a regular expression that uses subgroups.
      5 
      6 $Id: uptime.py 489 2007-11-28 23:40:34Z noah $
      7 """
      8 
      9 import pexpect
     10 import re
     11 
     12 # There are many different styles of uptime results. I try to parse them all. Yeee!
     13 # Examples from different machines:
     14 # [x86] Linux 2.4 (Redhat 7.3)
     15 #  2:06pm  up 63 days, 18 min,  3 users,  load average: 0.32, 0.08, 0.02
     16 # [x86] Linux 2.4.18-14 (Redhat 8.0)
     17 #  3:07pm  up 29 min,  1 user,  load average: 2.44, 2.51, 1.57
     18 # [PPC - G4] MacOS X 10.1 SERVER Edition
     19 # 2:11PM  up 3 days, 13:50, 3 users, load averages: 0.01, 0.00, 0.00
     20 # [powerpc] Darwin v1-58.corefa.com 8.2.0 Darwin Kernel Version 8.2.0
     21 # 10:35  up 18:06, 4 users, load averages: 0.52 0.47 0.36
     22 # [Sparc - R220] Sun Solaris (8)
     23 #  2:13pm  up 22 min(s),  1 user,  load average: 0.02, 0.01, 0.01
     24 # [x86] Linux 2.4.18-14 (Redhat 8)
     25 # 11:36pm  up 4 days, 17:58,  1 user,  load average: 0.03, 0.01, 0.00
     26 # AIX jwdir 2 5 0001DBFA4C00
     27 #  09:43AM   up  23:27,  1 user,  load average: 0.49, 0.32, 0.23
     28 # OpenBSD box3 2.9 GENERIC#653 i386
     29 #  6:08PM  up 4 days, 22:26, 1 user, load averages: 0.13, 0.09, 0.08
     30 
     31 # This parses uptime output into the major groups using regex group matching.
     32 p = pexpect.spawn ('uptime')
     33 p.expect('up\s+(.*?),\s+([0-9]+) users?,\s+load averages?: ([0-9]+\.[0-9][0-9]),?\s+([0-9]+\.[0-9][0-9]),?\s+([0-9]+\.[0-9][0-9])')
     34 duration, users, av1, av5, av15 = p.match.groups()
     35 
     36 # The duration is a little harder to parse because of all the different
     37 # styles of uptime. I'm sure there is a way to do this all at once with
     38 # one single regex, but I bet it would be hard to read and maintain.
     39 # If anyone wants to send me a version using a single regex I'd be happy to see it.
     40 days = '0'
     41 hours = '0'
     42 mins = '0'
     43 if 'day' in duration:
     44     p.match = re.search('([0-9]+)\s+day',duration)
     45     days = str(int(p.match.group(1)))
     46 if ':' in duration:
     47     p.match = re.search('([0-9]+):([0-9]+)',duration)
     48     hours = str(int(p.match.group(1)))
     49     mins = str(int(p.match.group(2)))
     50 if 'min' in duration:
     51     p.match = re.search('([0-9]+)\s+min',duration)
     52     mins = str(int(p.match.group(1)))
     53 
     54 # Print the parsed fields in CSV format.
     55 print 'days, hours, minutes, users, cpu avg 1 min, cpu avg 5 min, cpu avg 15 min'
     56 print '%s, %s, %s, %s, %s, %s, %s' % (days, hours, mins, users, av1, av5, av15)
     57 
     58