home *** CD-ROM | disk | FTP | other *** search
/ Celestin Apprentice 5 / Apprentice-Release5.iso / Source Code / C / Applications / Python 1.3.3 / Python 133 SRC / Demo / pdist / rrcs.py < prev    next >
Encoding:
Python Source  |  1995-12-21  |  3.4 KB  |  165 lines  |  [TEXT/Pyth]
  1. #! /usr/local/bin/python
  2.  
  3. "Remote RCS -- command line interface"
  4.  
  5. import sys
  6. import os
  7. import getopt
  8. import string
  9. import md5
  10. import tempfile
  11. from rcsclient import openrcsclient
  12.  
  13. def main():
  14.     sys.stdout = sys.stderr
  15.     try:
  16.         opts, rest = getopt.getopt(sys.argv[1:], 'h:p:d:qvL')
  17.         if not rest:
  18.             cmd = 'head'
  19.         else:
  20.             cmd, rest = rest[0], rest[1:]
  21.         if not commands.has_key(cmd):
  22.             raise getopt.error, "unknown command"
  23.         coptset, func = commands[cmd]
  24.         copts, files = getopt.getopt(rest, coptset)
  25.     except getopt.error, msg:
  26.         print msg
  27.         print "usage: rrcs [options] command [options] [file] ..."
  28.         print "where command can be:"
  29.         print "      ci|put      # checkin the given files"
  30.         print "      co|get      # checkout"
  31.         print "      info        # print header info"
  32.         print "      head        # print revision of head branch"
  33.         print "      list        # list filename if valid"
  34.         print "      log         # print full log"
  35.         print "      diff        # diff rcs file and work file"
  36.         print "if no files are given, all remote rcs files are assumed"
  37.         sys.exit(2)
  38.     x = openrcsclient(opts)
  39.     if not files:
  40.         files = x.listfiles()
  41.     for fn in files:
  42.         try:
  43.             func(x, copts, fn)
  44.         except (IOError, os.error), msg:
  45.             print "%s: %s" % (fn, msg)
  46.  
  47. def checkin(x, copts, fn):
  48.     f = open(fn)
  49.     data = f.read()
  50.     f.close()
  51.     new = not x.isvalid(fn)
  52.     if not new and same(x, copts, fn, data):
  53.         print "%s: unchanged since last checkin" % fn
  54.         return
  55.     print "Checking in", fn, "..."
  56.     message = asklogmessage(new)
  57.     messages = x.put(fn, data, message)
  58.     if messages:
  59.         print messages
  60.  
  61. def checkout(x, copts, fn):
  62.     data = x.get(fn)
  63.     f = open(fn, 'w')
  64.     f.write(data)
  65.     f.close()
  66.  
  67. def lock(x, copts, fn):
  68.     x.lock(fn)
  69.  
  70. def unlock(x, copts, fn):
  71.     x.unlock(fn)
  72.  
  73. def info(x, copts, fn):
  74.     dict = x.info(fn)
  75.     keys = dict.keys()
  76.     keys.sort()
  77.     for key in keys:
  78.         print key + ':', dict[key]
  79.     print '='*70
  80.  
  81. def head(x, copts, fn):
  82.     head = x.head(fn)
  83.     print fn, head
  84.  
  85. def list(x, copts, fn):
  86.     if x.isvalid(fn):
  87.         print fn
  88.  
  89. def log(x, copts, fn):
  90.     flags = ''
  91.     for o, a in copts:
  92.         flags = flags + ' ' + o + a
  93.     flags = flags[1:]
  94.     messages = x.log(fn, flags)
  95.     print messages
  96.  
  97. def diff(x, copts, fn):
  98.     if same(x, copts, fn):
  99.         return
  100.     flags = ''
  101.     for o, a in copts:
  102.         flags = flags + ' ' + o + a
  103.     flags = flags[1:]
  104.     data = x.get(fn)
  105.     tfn = tempfile.mktemp()
  106.     try:
  107.         tf = open(tfn, 'w')
  108.         tf.write(data)
  109.         tf.close()
  110.         print 'diff %s -r%s %s' % (flags, x.head(fn), fn)
  111.         sts = os.system('diff %s %s %s' % (flags, tfn, fn))
  112.         if sts:
  113.             print '='*70
  114.     finally:
  115.         remove(tfn)
  116.  
  117. def same(x, copts, fn, data = None):
  118.     if data is None:
  119.         f = open(fn)
  120.         data = f.read()
  121.         f.close()
  122.     lsum = md5.new(data).digest()
  123.     rsum = x.sum(fn)
  124.     return lsum == rsum
  125.  
  126. def asklogmessage(new):
  127.     if new:
  128.         print "enter description,",
  129.     else:
  130.         print "enter log message,",
  131.     print "terminate with single '.' or end of file:"
  132.     if new:
  133.         print "NOTE: This is NOT the log message!"
  134.     message = ""
  135.     while 1:
  136.         sys.stderr.write(">> ")
  137.         sys.stderr.flush()
  138.         line = sys.stdin.readline()
  139.         if not line or line == '.\n': break
  140.         message = message + line
  141.     return message
  142.  
  143. def remove(fn):
  144.     try:
  145.         os.unlink(fn)
  146.     except os.error:
  147.         pass
  148.  
  149. commands = {
  150.     'ci': ('', checkin),
  151.     'put': ('', checkin),
  152.     'co': ('', checkout),
  153.     'get': ('', checkout),
  154.     'info': ('', info),
  155.     'head': ('', head),
  156.     'list': ('', list),
  157.     'lock': ('', lock),
  158.     'unlock': ('', unlock),
  159.     'log': ('bhLRtd:l:r:s:w:V:', log),
  160.     'diff': ('c', diff),
  161.     }
  162.  
  163. if __name__ == '__main__':
  164.     main()
  165.