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         #log('%d writing: %d bytes\n' % (os.getpid(), len(data)))
  75         self.outp.write(data)
  76
  77     def ok(self):
  78         self.write('\nok\n')
  79
  80     def check_ok(self):
  81         self.outp.flush()
  82         rl = ''
  83         for rl in linereader(self.inp):
  84             #log('%d got line: %r\n' % (os.getpid(), rl))
  85             if not rl:  # empty line
  86                 continue
  87             elif rl == 'ok':
  88                 return True
  89             else:
  90                 raise Exception('expected "ok", got %r' % rl)
  91         raise Exception('server exited unexpectedly; see errors above')
  92
  93
  94 def linereader(f):
  95     while 1:
  96         line = f.readline()
  97         if not line:
  98             break
  99         yield line[:-1]
 100
 101
 102 def chunkyreader(f, count = None):
 103     if count != None:
 104         while count > 0:
 105             b = f.read(min(count, 65536))
 106             if not b:
 107                 raise IOError('EOF with %d bytes remaining' % count)
 108             yield b
 109             count -= len(b)
 110     else:
 111         while 1:
 112             b = f.read(65536)
 113             if not b: break
 114             yield b
 115
 116
 117 def slashappend(s):
 118     if s and not s.endswith('/'):
 119         return s + '/'
 120     else:
 121         return s
 122