helpers.py

   1 import sys, os, pwd, subprocess, errno
   2
   3
   4 def log(s):
   5     sys.stderr.write(s)
   6
   7
   8 def mkdirp(d):
   9     try:
  10         os.makedirs(d)
  11     except OSError, e:
  12         if e.errno == errno.EEXIST:
  13             pass
  14         else:
  15             raise
  16
  17
  18 def readpipe(argv):
  19     p = subprocess.Popen(argv, stdout=subprocess.PIPE)
  20     r = p.stdout.read()
  21     p.wait()
  22     return r
  23
  24
  25 _username = None
  26 def username():
  27     global _username
  28     if not _username:
  29         uid = os.getuid()
  30         try:
  31             _username = pwd.getpwuid(uid)[0]
  32         except KeyError:
  33             _username = 'user%d' % uid
  34     return _username
  35
  36
  37 _userfullname = None
  38 def userfullname():
  39     global _userfullname
  40     if not _userfullname:
  41         uid = os.getuid()
  42         try:
  43             _userfullname = pwd.getpwuid(uid)[4].split(',')[0]
  44         except KeyError:
  45             _userfullname = 'user%d' % uid
  46     return _userfullname
  47
  48
  49 _hostname = None
  50 def hostname():
  51     global _hostname
  52     if not _hostname:
  53         try:
  54             _hostname = readpipe(['hostname', '-f']).strip()
  55         except OSError:
  56             pass
  57     return _hostname or 'localhost'
  58
  59
  60 class Conn:
  61     def __init__(self, inp, outp):
  62         self.inp = inp
  63         self.outp = outp
  64
  65     def read(self, size):
  66         self.outp.flush()
  67         return self.inp.read(size)
  68
  69     def readline(self):
  70         self.outp.flush()
  71         return self.inp.readline()
  72
  73     def write(self, data):
  74         self.outp.write(data)
  75
  76     def ok(self):
  77         self.write('\nok\n')
  78
  79     def check_ok(self):
  80         self.outp.flush()
  81         rl = ''
  82         for rl in linereader(self.inp):
  83             if not rl:
  84                 continue
  85             elif rl == 'ok':
  86                 return True
  87             else:
  88                 break
  89         raise Exception('expected "ok", got %r' % rl)
  90
  91
  92 def linereader(f):
  93     while 1:
  94         line = f.readline()
  95         if not line:
  96             break
  97         yield line[:-1]
  98
  99
 100 def chunkyreader(f, count = None):
 101     if count != None:
 102         while count > 0:
 103             b = f.read(min(count, 65536))
 104             if not b:
 105                 raise IOError('EOF with %d bytes remaining' % count)
 106             yield b
 107             count -= len(b)
 108     else:
 109         while 1:
 110             b = f.read(65536)
 111             if not b: break
 112             yield b