helpers.py

   1 import sys, os, pwd, subprocess, errno, socket
   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         _hostname = socket.getfqdn()
  54     return _hostname
  55
  56
  57 class Conn:
  58     def __init__(self, inp, outp):
  59         self.inp = inp
  60         self.outp = outp
  61
  62     def read(self, size):
  63         self.outp.flush()
  64         return self.inp.read(size)
  65
  66     def readline(self):
  67         self.outp.flush()
  68         return self.inp.readline()
  69
  70     def write(self, data):
  71         #log('%d writing: %d bytes\n' % (os.getpid(), len(data)))
  72         self.outp.write(data)
  73
  74     def ok(self):
  75         self.write('\nok\n')
  76
  77     def check_ok(self):
  78         self.outp.flush()
  79         rl = ''
  80         for rl in linereader(self.inp):
  81             #log('%d got line: %r\n' % (os.getpid(), rl))
  82             if not rl:  # empty line
  83                 continue
  84             elif rl == 'ok':
  85                 return True
  86             else:
  87                 raise Exception('expected "ok", got %r' % rl)
  88         raise Exception('server exited unexpectedly; see errors above')
  89
  90
  91 def linereader(f):
  92     while 1:
  93         line = f.readline()
  94         if not line:
  95             break
  96         yield line[:-1]
  97
  98
  99 def chunkyreader(f, count = None):
 100     if count != None:
 101         while count > 0:
 102             b = f.read(min(count, 65536))
 103             if not b:
 104                 raise IOError('EOF with %d bytes remaining' % count)
 105             yield b
 106             count -= len(b)
 107     else:
 108         while 1:
 109             b = f.read(65536)
 110             if not b: break
 111             yield b
 112
 113
 114 def slashappend(s):
 115     if s and not s.endswith('/'):
 116         return s + '/'
 117     else:
 118         return s
 119