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         for rl in linereader(self.inp):
  82             if not rl:
  83                 continue
  84             elif rl == 'ok':
  85                 return True
  86             else:
  87                 raise Exception('expected "ok", got %r' % rl)
  88
  89
  90 def linereader(f):
  91     while 1:
  92         line = f.readline()
  93         if not line:
  94             break
  95         yield line[:-1]
  96
  97
  98 def chunkyreader(f, count):
  99     while count > 0:
 100         b = f.read(min(count, 65536))
 101         if not b:
 102             raise IOError('EOF with %d bytes remaining' % count)
 103         yield b
 104         count -= len(b)