root/hodgestar/PythonCode/SvnTools/svn-filter-logs.py

"""Small script for filtering entries out of SVN logs.
   """

class LogEntry(object):
    def __init__(self, iRevision, sUser,
                 sDate, iLines, sData):
        self.iRevision = iRevision
        self.sUser = sUser
        self.sDate = sDate
        self.iLines = iLines
        self.sData = sData

    @classmethod
    def from_string(self, s):
        aLines = s.split("\n")

        aHeaders = [s.strip() for s in aLines[0].split("|")]
        iRevision = int(aHeaders[0][1:])
        iLines = int(aHeaders[3].split()[0])

        sData = "\n".join(aLines[2:])

        return LogEntry(iRevision, aHeaders[1], aHeaders[2],
                        iLines, sData)

    def __str__(self):
        return """r%d | %s | %s | %i line%s\n""" \
               % (self.iRevision, self.sUser, self.sDate, self.iLines,
                  '' if self.iLines == 1 else 's') + \
               """%s""" % self.sData

    def matches(self, dFilter):
        bMatch = True
        if dFilter.get('user',self.sUser) != self.sUser:
            bMatch = False
        return bMatch

class SvnLogIter(object):
    SEP = "---------------------------------"

    def __init__(self, fFile):
        self._fFile = fFile
        self._aLines = []

    def __iter__(self):
        for sLine in self._fFile:
            if sLine.startswith(self.SEP):
                if self._aLines:
                    yield LogEntry.from_string("".join(self._aLines))
                    self._aLines = []
            else:
                self._aLines.append(sLine)

def main(aArgs):
    sUser = aArgs[1]
    sLogFile = aArgs[2]

    dFilter = {'user': sUser}
    fLogFile = file(sLogFile,"r")

    iTotal, iMatched = 0, 0

    for oLog in SvnLogIter(fLogFile):
        iTotal += 1
        if oLog.matches(dFilter):
            iMatched += 1
            print oLog

    fLogFile.close()

    print SvnLogIter.SEP
    print "%i / %i matched." % (iMatched, iTotal)

    return 0

if __name__ == "__main__":
    import sys
    sys.exit(main(sys.argv))