These files are a subset of the python-2.7.2.tgz distribution from python.org. Changed files from PyMod-2.7.2 have been copied into the corresponding directories of this tree, replacing the original files in the distribution. Signed-off-by: daryl.mcdaniel@intel.com git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@13197 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			478 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			478 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
#! /usr/bin/env python
 | 
						|
 | 
						|
"""Remote CVS -- command line interface"""
 | 
						|
 | 
						|
# XXX To do:
 | 
						|
#
 | 
						|
# Bugs:
 | 
						|
# - if the remote file is deleted, "rcvs update" will fail
 | 
						|
#
 | 
						|
# Functionality:
 | 
						|
# - cvs rm
 | 
						|
# - descend into directories (alraedy done for update)
 | 
						|
# - conflict resolution
 | 
						|
# - other relevant commands?
 | 
						|
# - branches
 | 
						|
#
 | 
						|
# - Finesses:
 | 
						|
# - retain file mode's x bits
 | 
						|
# - complain when "nothing known about filename"
 | 
						|
# - edit log message the way CVS lets you edit it
 | 
						|
# - cvs diff -rREVA -rREVB
 | 
						|
# - send mail the way CVS sends it
 | 
						|
#
 | 
						|
# Performance:
 | 
						|
# - cache remote checksums (for every revision ever seen!)
 | 
						|
# - translate symbolic revisions to numeric revisions
 | 
						|
#
 | 
						|
# Reliability:
 | 
						|
# - remote locking
 | 
						|
#
 | 
						|
# Security:
 | 
						|
# - Authenticated RPC?
 | 
						|
 | 
						|
 | 
						|
from cvslib import CVS, File
 | 
						|
import md5
 | 
						|
import os
 | 
						|
import string
 | 
						|
import sys
 | 
						|
from cmdfw import CommandFrameWork
 | 
						|
 | 
						|
 | 
						|
DEF_LOCAL = 1                           # Default -l
 | 
						|
 | 
						|
 | 
						|
class MyFile(File):
 | 
						|
 | 
						|
    def action(self):
 | 
						|
        """Return a code indicating the update status of this file.
 | 
						|
 | 
						|
        The possible return values are:
 | 
						|
 | 
						|
        '=' -- everything's fine
 | 
						|
        '0' -- file doesn't exist anywhere
 | 
						|
        '?' -- exists locally only
 | 
						|
        'A' -- new locally
 | 
						|
        'R' -- deleted locally
 | 
						|
        'U' -- changed remotely, no changes locally
 | 
						|
               (includes new remotely or deleted remotely)
 | 
						|
        'M' -- changed locally, no changes remotely
 | 
						|
        'C' -- conflict: changed locally as well as remotely
 | 
						|
               (includes cases where the file has been added
 | 
						|
               or removed locally and remotely)
 | 
						|
        'D' -- deleted remotely
 | 
						|
        'N' -- new remotely
 | 
						|
        'r' -- get rid of entry
 | 
						|
        'c' -- create entry
 | 
						|
        'u' -- update entry
 | 
						|
 | 
						|
        (and probably others :-)
 | 
						|
        """
 | 
						|
        if not self.lseen:
 | 
						|
            self.getlocal()
 | 
						|
        if not self.rseen:
 | 
						|
            self.getremote()
 | 
						|
        if not self.eseen:
 | 
						|
            if not self.lsum:
 | 
						|
                if not self.rsum: return '0' # Never heard of
 | 
						|
                else:
 | 
						|
                    return 'N' # New remotely
 | 
						|
            else: # self.lsum
 | 
						|
                if not self.rsum: return '?' # Local only
 | 
						|
                # Local and remote, but no entry
 | 
						|
                if self.lsum == self.rsum:
 | 
						|
                    return 'c' # Restore entry only
 | 
						|
                else: return 'C' # Real conflict
 | 
						|
        else: # self.eseen
 | 
						|
            if not self.lsum:
 | 
						|
                if self.edeleted:
 | 
						|
                    if self.rsum: return 'R' # Removed
 | 
						|
                    else: return 'r' # Get rid of entry
 | 
						|
                else: # not self.edeleted
 | 
						|
                    if self.rsum:
 | 
						|
                        print "warning:",
 | 
						|
                        print self.file,
 | 
						|
                        print "was lost"
 | 
						|
                        return 'U'
 | 
						|
                    else: return 'r' # Get rid of entry
 | 
						|
            else: # self.lsum
 | 
						|
                if not self.rsum:
 | 
						|
                    if self.enew: return 'A' # New locally
 | 
						|
                    else: return 'D' # Deleted remotely
 | 
						|
                else: # self.rsum
 | 
						|
                    if self.enew:
 | 
						|
                        if self.lsum == self.rsum:
 | 
						|
                            return 'u'
 | 
						|
                        else:
 | 
						|
                            return 'C'
 | 
						|
                    if self.lsum == self.esum:
 | 
						|
                        if self.esum == self.rsum:
 | 
						|
                            return '='
 | 
						|
                        else:
 | 
						|
                            return 'U'
 | 
						|
                    elif self.esum == self.rsum:
 | 
						|
                        return 'M'
 | 
						|
                    elif self.lsum == self.rsum:
 | 
						|
                        return 'u'
 | 
						|
                    else:
 | 
						|
                        return 'C'
 | 
						|
 | 
						|
    def update(self):
 | 
						|
        code = self.action()
 | 
						|
        if code == '=': return
 | 
						|
        print code, self.file
 | 
						|
        if code in ('U', 'N'):
 | 
						|
            self.get()
 | 
						|
        elif code == 'C':
 | 
						|
            print "%s: conflict resolution not yet implemented" % \
 | 
						|
                  self.file
 | 
						|
        elif code == 'D':
 | 
						|
            remove(self.file)
 | 
						|
            self.eseen = 0
 | 
						|
        elif code == 'r':
 | 
						|
            self.eseen = 0
 | 
						|
        elif code in ('c', 'u'):
 | 
						|
            self.eseen = 1
 | 
						|
            self.erev = self.rrev
 | 
						|
            self.enew = 0
 | 
						|
            self.edeleted = 0
 | 
						|
            self.esum = self.rsum
 | 
						|
            self.emtime, self.ectime = os.stat(self.file)[-2:]
 | 
						|
            self.extra = ''
 | 
						|
 | 
						|
    def commit(self, message = ""):
 | 
						|
        code = self.action()
 | 
						|
        if code in ('A', 'M'):
 | 
						|
            self.put(message)
 | 
						|
            return 1
 | 
						|
        elif code == 'R':
 | 
						|
            print "%s: committing removes not yet implemented" % \
 | 
						|
                  self.file
 | 
						|
        elif code == 'C':
 | 
						|
            print "%s: conflict resolution not yet implemented" % \
 | 
						|
                  self.file
 | 
						|
 | 
						|
    def diff(self, opts = []):
 | 
						|
        self.action()           # To update lseen, rseen
 | 
						|
        flags = ''
 | 
						|
        rev = self.rrev
 | 
						|
        # XXX should support two rev options too!
 | 
						|
        for o, a in opts:
 | 
						|
            if o == '-r':
 | 
						|
                rev = a
 | 
						|
            else:
 | 
						|
                flags = flags + ' ' + o + a
 | 
						|
        if rev == self.rrev and self.lsum == self.rsum:
 | 
						|
            return
 | 
						|
        flags = flags[1:]
 | 
						|
        fn = self.file
 | 
						|
        data = self.proxy.get((fn, rev))
 | 
						|
        sum = md5.new(data).digest()
 | 
						|
        if self.lsum == sum:
 | 
						|
            return
 | 
						|
        import tempfile
 | 
						|
        tf = tempfile.NamedTemporaryFile()
 | 
						|
        tf.write(data)
 | 
						|
        tf.flush()
 | 
						|
        print 'diff %s -r%s %s' % (flags, rev, fn)
 | 
						|
        sts = os.system('diff %s %s %s' % (flags, tf.name, fn))
 | 
						|
        if sts:
 | 
						|
            print '='*70
 | 
						|
 | 
						|
    def commitcheck(self):
 | 
						|
        return self.action() != 'C'
 | 
						|
 | 
						|
    def put(self, message = ""):
 | 
						|
        print "Checking in", self.file, "..."
 | 
						|
        data = open(self.file).read()
 | 
						|
        if not self.enew:
 | 
						|
            self.proxy.lock(self.file)
 | 
						|
        messages = self.proxy.put(self.file, data, message)
 | 
						|
        if messages:
 | 
						|
            print messages
 | 
						|
        self.setentry(self.proxy.head(self.file), self.lsum)
 | 
						|
 | 
						|
    def get(self):
 | 
						|
        data = self.proxy.get(self.file)
 | 
						|
        f = open(self.file, 'w')
 | 
						|
        f.write(data)
 | 
						|
        f.close()
 | 
						|
        self.setentry(self.rrev, self.rsum)
 | 
						|
 | 
						|
    def log(self, otherflags):
 | 
						|
        print self.proxy.log(self.file, otherflags)
 | 
						|
 | 
						|
    def add(self):
 | 
						|
        self.eseen = 0          # While we're hacking...
 | 
						|
        self.esum = self.lsum
 | 
						|
        self.emtime, self.ectime = 0, 0
 | 
						|
        self.erev = ''
 | 
						|
        self.enew = 1
 | 
						|
        self.edeleted = 0
 | 
						|
        self.eseen = 1          # Done
 | 
						|
        self.extra = ''
 | 
						|
 | 
						|
    def setentry(self, erev, esum):
 | 
						|
        self.eseen = 0          # While we're hacking...
 | 
						|
        self.esum = esum
 | 
						|
        self.emtime, self.ectime = os.stat(self.file)[-2:]
 | 
						|
        self.erev = erev
 | 
						|
        self.enew = 0
 | 
						|
        self.edeleted = 0
 | 
						|
        self.eseen = 1          # Done
 | 
						|
        self.extra = ''
 | 
						|
 | 
						|
 | 
						|
SENDMAIL = "/usr/lib/sendmail -t"
 | 
						|
MAILFORM = """To: %s
 | 
						|
Subject: CVS changes: %s
 | 
						|
 | 
						|
...Message from rcvs...
 | 
						|
 | 
						|
Committed files:
 | 
						|
        %s
 | 
						|
 | 
						|
Log message:
 | 
						|
        %s
 | 
						|
"""
 | 
						|
 | 
						|
 | 
						|
class RCVS(CVS):
 | 
						|
 | 
						|
    FileClass = MyFile
 | 
						|
 | 
						|
    def __init__(self):
 | 
						|
        CVS.__init__(self)
 | 
						|
 | 
						|
    def update(self, files):
 | 
						|
        for e in self.whichentries(files, 1):
 | 
						|
            e.update()
 | 
						|
 | 
						|
    def commit(self, files, message = ""):
 | 
						|
        list = self.whichentries(files)
 | 
						|
        if not list: return
 | 
						|
        ok = 1
 | 
						|
        for e in list:
 | 
						|
            if not e.commitcheck():
 | 
						|
                ok = 0
 | 
						|
        if not ok:
 | 
						|
            print "correct above errors first"
 | 
						|
            return
 | 
						|
        if not message:
 | 
						|
            message = raw_input("One-liner: ")
 | 
						|
        committed = []
 | 
						|
        for e in list:
 | 
						|
            if e.commit(message):
 | 
						|
                committed.append(e.file)
 | 
						|
        self.mailinfo(committed, message)
 | 
						|
 | 
						|
    def mailinfo(self, files, message = ""):
 | 
						|
        towhom = "sjoerd@cwi.nl, jack@cwi.nl" # XXX
 | 
						|
        mailtext = MAILFORM % (towhom, string.join(files),
 | 
						|
                                string.join(files), message)
 | 
						|
        print '-'*70
 | 
						|
        print mailtext
 | 
						|
        print '-'*70
 | 
						|
        ok = raw_input("OK to mail to %s? " % towhom)
 | 
						|
        if string.lower(string.strip(ok)) in ('y', 'ye', 'yes'):
 | 
						|
            p = os.popen(SENDMAIL, "w")
 | 
						|
            p.write(mailtext)
 | 
						|
            sts = p.close()
 | 
						|
            if sts:
 | 
						|
                print "Sendmail exit status %s" % str(sts)
 | 
						|
            else:
 | 
						|
                print "Mail sent."
 | 
						|
        else:
 | 
						|
            print "No mail sent."
 | 
						|
 | 
						|
    def report(self, files):
 | 
						|
        for e in self.whichentries(files):
 | 
						|
            e.report()
 | 
						|
 | 
						|
    def diff(self, files, opts):
 | 
						|
        for e in self.whichentries(files):
 | 
						|
            e.diff(opts)
 | 
						|
 | 
						|
    def add(self, files):
 | 
						|
        if not files:
 | 
						|
            raise RuntimeError, "'cvs add' needs at least one file"
 | 
						|
        list = []
 | 
						|
        for e in self.whichentries(files, 1):
 | 
						|
            e.add()
 | 
						|
 | 
						|
    def rm(self, files):
 | 
						|
        if not files:
 | 
						|
            raise RuntimeError, "'cvs rm' needs at least one file"
 | 
						|
        raise RuntimeError, "'cvs rm' not yet imlemented"
 | 
						|
 | 
						|
    def log(self, files, opts):
 | 
						|
        flags = ''
 | 
						|
        for o, a in opts:
 | 
						|
            flags = flags + ' ' + o + a
 | 
						|
        for e in self.whichentries(files):
 | 
						|
            e.log(flags)
 | 
						|
 | 
						|
    def whichentries(self, files, localfilestoo = 0):
 | 
						|
        if files:
 | 
						|
            list = []
 | 
						|
            for file in files:
 | 
						|
                if self.entries.has_key(file):
 | 
						|
                    e = self.entries[file]
 | 
						|
                else:
 | 
						|
                    e = self.FileClass(file)
 | 
						|
                    self.entries[file] = e
 | 
						|
                list.append(e)
 | 
						|
        else:
 | 
						|
            list = self.entries.values()
 | 
						|
            for file in self.proxy.listfiles():
 | 
						|
                if self.entries.has_key(file):
 | 
						|
                    continue
 | 
						|
                e = self.FileClass(file)
 | 
						|
                self.entries[file] = e
 | 
						|
                list.append(e)
 | 
						|
            if localfilestoo:
 | 
						|
                for file in os.listdir(os.curdir):
 | 
						|
                    if not self.entries.has_key(file) \
 | 
						|
                       and not self.ignored(file):
 | 
						|
                        e = self.FileClass(file)
 | 
						|
                        self.entries[file] = e
 | 
						|
                        list.append(e)
 | 
						|
            list.sort()
 | 
						|
        if self.proxy:
 | 
						|
            for e in list:
 | 
						|
                if e.proxy is None:
 | 
						|
                    e.proxy = self.proxy
 | 
						|
        return list
 | 
						|
 | 
						|
 | 
						|
class rcvs(CommandFrameWork):
 | 
						|
 | 
						|
    GlobalFlags = 'd:h:p:qvL'
 | 
						|
    UsageMessage = \
 | 
						|
"usage: rcvs [-d directory] [-h host] [-p port] [-q] [-v] [subcommand arg ...]"
 | 
						|
    PostUsageMessage = \
 | 
						|
            "If no subcommand is given, the status of all files is listed"
 | 
						|
 | 
						|
    def __init__(self):
 | 
						|
        """Constructor."""
 | 
						|
        CommandFrameWork.__init__(self)
 | 
						|
        self.proxy = None
 | 
						|
        self.cvs = RCVS()
 | 
						|
 | 
						|
    def close(self):
 | 
						|
        if self.proxy:
 | 
						|
            self.proxy._close()
 | 
						|
        self.proxy = None
 | 
						|
 | 
						|
    def recurse(self):
 | 
						|
        self.close()
 | 
						|
        names = os.listdir(os.curdir)
 | 
						|
        for name in names:
 | 
						|
            if name == os.curdir or name == os.pardir:
 | 
						|
                continue
 | 
						|
            if name == "CVS":
 | 
						|
                continue
 | 
						|
            if not os.path.isdir(name):
 | 
						|
                continue
 | 
						|
            if os.path.islink(name):
 | 
						|
                continue
 | 
						|
            print "--- entering subdirectory", name, "---"
 | 
						|
            os.chdir(name)
 | 
						|
            try:
 | 
						|
                if os.path.isdir("CVS"):
 | 
						|
                    self.__class__().run()
 | 
						|
                else:
 | 
						|
                    self.recurse()
 | 
						|
            finally:
 | 
						|
                os.chdir(os.pardir)
 | 
						|
                print "--- left subdirectory", name, "---"
 | 
						|
 | 
						|
    def options(self, opts):
 | 
						|
        self.opts = opts
 | 
						|
 | 
						|
    def ready(self):
 | 
						|
        import rcsclient
 | 
						|
        self.proxy = rcsclient.openrcsclient(self.opts)
 | 
						|
        self.cvs.setproxy(self.proxy)
 | 
						|
        self.cvs.getentries()
 | 
						|
 | 
						|
    def default(self):
 | 
						|
        self.cvs.report([])
 | 
						|
 | 
						|
    def do_report(self, opts, files):
 | 
						|
        self.cvs.report(files)
 | 
						|
 | 
						|
    def do_update(self, opts, files):
 | 
						|
        """update [-l] [-R] [file] ..."""
 | 
						|
        local = DEF_LOCAL
 | 
						|
        for o, a in opts:
 | 
						|
            if o == '-l': local = 1
 | 
						|
            if o == '-R': local = 0
 | 
						|
        self.cvs.update(files)
 | 
						|
        self.cvs.putentries()
 | 
						|
        if not local and not files:
 | 
						|
            self.recurse()
 | 
						|
    flags_update = '-lR'
 | 
						|
    do_up = do_update
 | 
						|
    flags_up = flags_update
 | 
						|
 | 
						|
    def do_commit(self, opts, files):
 | 
						|
        """commit [-m message] [file] ..."""
 | 
						|
        message = ""
 | 
						|
        for o, a in opts:
 | 
						|
            if o == '-m': message = a
 | 
						|
        self.cvs.commit(files, message)
 | 
						|
        self.cvs.putentries()
 | 
						|
    flags_commit = 'm:'
 | 
						|
    do_com = do_commit
 | 
						|
    flags_com = flags_commit
 | 
						|
 | 
						|
    def do_diff(self, opts, files):
 | 
						|
        """diff [difflags] [file] ..."""
 | 
						|
        self.cvs.diff(files, opts)
 | 
						|
    flags_diff = 'cbitwcefhnlr:sD:S:'
 | 
						|
    do_dif = do_diff
 | 
						|
    flags_dif = flags_diff
 | 
						|
 | 
						|
    def do_add(self, opts, files):
 | 
						|
        """add file ..."""
 | 
						|
        if not files:
 | 
						|
            print "'rcvs add' requires at least one file"
 | 
						|
            return
 | 
						|
        self.cvs.add(files)
 | 
						|
        self.cvs.putentries()
 | 
						|
 | 
						|
    def do_remove(self, opts, files):
 | 
						|
        """remove file ..."""
 | 
						|
        if not files:
 | 
						|
            print "'rcvs remove' requires at least one file"
 | 
						|
            return
 | 
						|
        self.cvs.remove(files)
 | 
						|
        self.cvs.putentries()
 | 
						|
    do_rm = do_remove
 | 
						|
 | 
						|
    def do_log(self, opts, files):
 | 
						|
        """log [rlog-options] [file] ..."""
 | 
						|
        self.cvs.log(files, opts)
 | 
						|
    flags_log = 'bhLNRtd:s:V:r:'
 | 
						|
 | 
						|
 | 
						|
def remove(fn):
 | 
						|
    try:
 | 
						|
        os.unlink(fn)
 | 
						|
    except os.error:
 | 
						|
        pass
 | 
						|
 | 
						|
 | 
						|
def main():
 | 
						|
    r = rcvs()
 | 
						|
    try:
 | 
						|
        r.run()
 | 
						|
    finally:
 | 
						|
        r.close()
 | 
						|
 | 
						|
 | 
						|
if __name__ == "__main__":
 | 
						|
    main()
 |