Homec4science
Diffusion py-licenser (master) Browse

py-licenser/VersionInfo.py
9f2960d072demaster

Last Change
Raw File

VersionInfo.py
View Options

#!/usr/bin/python
# -*- coding: utf-8 -*-
__all__ = ["VersionInfo", "SVN", "GIT"]
import os
import pysvn
import re
import datetime as dt
try:
from termcolor import colored
except ImportError:
def colored(string, color, **kwargs):
return string
SVN = 0
GIT = 1
class VersionInfo(object):
__backend_str_conv = {'svn': SVN, 'git': GIT}
# members
_authors = set()
_author_list = dict()
_modification_dates = []
_name = ""
_filename = ""
_revisions = []
_names = dict()
oldest_name = ""
def __new__(cls, filename, backend = SVN, **kwargs):
"""
Factory constructor depending on the choosen backend
"""
if type(backend) == str or type(backend) == unicode:
backend = cls.__backend_str_conv[backend]
if(backend == SVN):
return super(VersionInfo, cls).__new__(SVNInfo, filename, **kwargs)
elif(backend == GIT):
return super(VersionInfo, cls).__new__(GITInfo, filename, **kwargs)
else:
raise(Exception("{0} not a known backend".format(backend)))
def __init__(self, filename, backend = SVN): pass
def getName(self):
return self._name
def creationDate(self):
return min(self._modification_dates)
def lastModificationDate(self):
return max(self._modification_dates)
def getAuthors(self):
return self._authors
def numberOfModifications(self,author):
return len(self._author_list[author])
def modificationsByAuthor(self, author):
a = self._author_list[author]
res = []
for d, r, msg in a:
date = dt.datetime.fromtimestamp(d).strftime("%d-%m-%Y")
info = colored("@ {0} rev {1} msg {2}".format(date, r.number ,msg), 'blue', attrs=['bold'])
modif = self.getModifications(r)
res.append( [ info, modif, modif.split("\n") ] )
return res
@classmethod
def getNameFromFilename(cls, filename, backend = SVN):
if type(backend) == str or type(backend) == unicode:
backend = cls.__backend_str_conv[backend]
if(backend == SVN):
return SVNInfo.getNameFromFilename(filename)
elif(backend == GIT):
return GITInfo.getNameFromFilename(filename)
else:
raise(Exception("{0} not a known backend".format(backend)))
class SVNInfo(VersionInfo):
def __init__(self, filename, rev_to = None, **kwargs):
self._authors = set()
self._author_list = dict()
self._modification_dates = []
self._revisions = []
self._names = dict()
self._filename = os.path.expanduser(filename)
self._client = pysvn.Client()
self._root_url = self._client.root_url_from_path(self._filename)
self._infos = self._client.info(self._filename)
if rev_to is not None:
rev_start = pysvn.Revision(pysvn.opt_revision_kind.date, rev_to.strftime('%s'))
else:
rev_start = pysvn.Revision(pysvn.opt_revision_kind.head)
self._name = self._infos.url.replace(self._root_url, '')
self._logs = self._client.log(self._filename,
strict_node_history=False,
discover_changed_paths=True,
revision_start=rev_start)
prev_name = self._name
self._logs = sorted(self._logs, key=lambda log: log["revision"].number, reverse=True)
for entry in self._logs:
d = entry["date"]
self._modification_dates.append(d)
# checking if the name as changed
old_names = [ ch_path for ch_path in entry["changed_paths"] if
(re.match("{0}.*".format(ch_path["path"]), prev_name)) ]
if len(old_names) > 0 and old_names[0]["copyfrom_path"] is not None:
m = re.match("{0}(.*)".format(old_names[0]["path"]), prev_name)
tmp_prev_name = prev_name
prev_name = "{0}{1}".format(old_names[0]["copyfrom_path"],
m.group(1) if m else "")
self._names[old_names[0]["copyfrom_revision"].number] = tmp_prev_name
self._names[0] = prev_name
# print("\n".join(["{0} {1}".format(n, r) for r, n in self._names.items() ]))
def oldest_name(self):
return self._names[0]
def populate(self, author_db, rev_from=None, rev_to=None, **kwargs):
if(not rev_from == None):
rev_end = pysvn.Revision(pysvn.opt_revision_kind.date, rev_from.strftime('%s'))
else:
rev_end = pysvn.Revision(pysvn.opt_revision_kind.number, 0)
if(not rev_to == None):
rev_start = pysvn.Revision(pysvn.opt_revision_kind.date, rev_to.strftime('%s'))
else:
rev_start = pysvn.Revision(pysvn.opt_revision_kind.head)
self._logs = self._client.log(self._filename,
strict_node_history=False,
revision_start=rev_start,
revision_end=rev_end)
self._logs = sorted(self._logs, key=lambda log: log["revision"].number)
for entry in self._logs:
auth = entry["author"]
rev = entry["revision"]
d = entry["date"]
msg = entry["message"] if "message" in entry else ""
self._revisions.append(rev)
if auth not in self._author_list:
self._author_list[auth] = []
a = author_db.findByUserName(auth)
if not author_db.isAuthorInIgnoreList(a):
self._authors.add(a)
self._author_list[auth].append((d, rev, msg))
@classmethod
def getNameFromFilename(cls, filename, **kwargs):
client = pysvn.Client()
root_url = client.root_url_from_path(filename)
infos = client.info(filename)
return infos.url.replace(root_url, '')
def __findPreviousRelease(self, r):
revs = [ rev.number for rev in self._revisions if rev.number < r]
rev = 0 if len(revs) == 0 else max(revs)
return pysvn.Revision(pysvn.opt_revision_kind.number, rev)
def getModifications(self, revision):
r_pre = pysvn.Revision(pysvn.opt_revision_kind.number, revision.number -1)
#self.__findPreviousRelease(revision.number)
relevant_rev = max([ n for n in self._names.keys() if revision.number > n ])
try:
#print("{0} r{1}:{2}".format(self._root_url + self._names[relevant_rev],
# r_pre.number,
# revision.number))
res_diff = self._client.diff("/tmp", self._root_url + self._names[relevant_rev],
revision1=r_pre,
revision2=revision,
diff_options=['-b', '-w'])
return res_diff
except Exception as e: pass
return ""
c4science · Help