# Copyright (C) 2006 by Szilveszter Farkas (Phanatic) <szilveszter.farkas@gmail.com>
# Some parts of the code are:
# Copyright (C) 2005, 2006 by Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
import time
import bzrlib
from bzrlib import (
bzrdir,
diff,
errors,
osutils,
urlutils,
)
from bzrlib.missing import find_unmerged
def _repo_rel_url(repo_url, inner_url):
"""Return path with common prefix of repository path removed.
If path is not part of the repository, the original path is returned.
If path is equal to the repository, the current directory marker '.' is
returned.
Otherwise, a relative path is returned, with trailing '/' stripped.
"""
inner_url = urlutils.normalize_url(inner_url)
repo_url = urlutils.normalize_url(repo_url)
if inner_url == repo_url:
return '.'
result = urlutils.relative_url(repo_url, inner_url)
if result != inner_url:
result = result.rstrip('/')
return result
def get_location_info(repository, branch=None, working=None):
""" Get known locations for working, branch and repository.
:return: a dictionary containing the needed infos
"""
ret = {}
repository_path = repository.bzrdir.root_transport.base
if working and branch:
working_path = working.bzrdir.root_transport.base
branch_path = branch.bzrdir.root_transport.base
if working_path != branch_path:
# lightweight checkout
ret['lightcoroot'] = working_path
if repository.is_shared():
# lightweight checkout of branch in shared repository
ret['sharedrepo'] = repository_path
ret['repobranch'] = _repo_rel_url(repository_path, branch_path)
else:
# lightweight checkout of standalone branch
ret['cobranch'] = branch_path
elif repository.is_shared():
# branch with tree inside shared repository
ret['sharedrepo'] = repository_path
ret['repoco'] = _repo_rel_url(repository_path, branch_path)
elif branch.get_bound_location():
# normal checkout
ret['coroot'] = working_path
ret['cobranch'] = branch.get_bound_location()
else:
# standalone
ret['branchroot'] = working_path
elif branch:
branch_path = branch.bzrdir.root_transport.base
if repository.is_shared():
# branch is part of shared repository
ret['sharedrepo'] = repository_path
ret['repobranch'] = _repo_rel_url(repository_path, branch_path)
else:
# standalone branch
ret['branchroot'] = branch_path
else:
# shared repository
assert repository.is_shared()
ret['sharedrepo'] = repository_path
return ret
def get_related_info(branch):
""" Get parent and push location of branch.
:return: a dictionary containing the needed infos
"""
ret = {}
if branch.get_parent() or branch.get_push_location():
if branch.get_parent():
ret['parentbranch'] = branch.get_parent()
if branch.get_push_location():
ret['publishbranch'] = branch.get_push_location()
return ret
def get_format_info(control=None, repository=None, branch=None, working=None):
""" Get known formats for control, working, branch and repository.
:return: a dictionary containing the needed infos
"""
ret = {}
if control:
ret['control'] = control._format.get_format_description()
if working:
ret['workingtree'] = working._format.get_format_description()
if branch:
ret['branch'] = branch._format.get_format_description()
if repository:
ret['repository'] = repository._format.get_format_description()
return ret
def get_locking_info(repository, branch=None, working=None):
""" Get locking status of working, branch and repository.
:return: a dictionary containing the needed infos
"""
ret = {}
if (repository.get_physical_lock_status() or
(branch and branch.get_physical_lock_status()) or
(working and working.get_physical_lock_status())):
if working:
if working.get_physical_lock_status():
status = 'locked'
else:
status = 'unlocked'
ret['workingtree'] = status
if branch:
if branch.get_physical_lock_status():
status = 'locked'
else:
status = 'unlocked'
ret['branch'] = status
if repository:
if repository.get_physical_lock_status():
status = 'locked'
else:
status = 'unlocked'
ret['repository'] = status
return ret
def get_missing_revisions_branch(branch):
""" Get missing master revisions in branch.
:return: a dictionary containing the needed infos
"""
ret = {}
# Try with inaccessible branch ?
master = branch.get_master_branch()
if master:
local_extra, remote_extra = find_unmerged(branch, master)
if remote_extra:
ret = len(remote_extra)
return ret
def get_missing_revisions_working(working):
""" Get missing revisions in working tree.
:return: a dictionary containing the needed infos
"""
if (bzrlib.version_info[0] == 0) and (bzrlib.version_info[1] < 9):
# function deprecated after 0.9
from bzrlib.delta import compare_trees
ret = {}
branch = working.branch
basis = working.basis_tree()
if (bzrlib.version_info[0] == 0) and (bzrlib.version_info[1] < 9):
delta = compare_trees(basis, working, want_unchanged=True)
else:
delta = working.changes_from(basis, want_unchanged=True)
history = branch.revision_history()
tree_last_id = working.last_revision()
if len(history) and tree_last_id != history[-1]:
tree_last_revno = branch.revision_id_to_revno(tree_last_id)
missing_count = len(history) - tree_last_revno
ret = missing_count
return ret
def get_working_stats(working):
""" Get statistics about a working tree.
:return: a dictionary containing the needed infos
"""
if (bzrlib.version_info[0] == 0) and (bzrlib.version_info[1] < 9):
# function deprecated after 0.9
from bzrlib.delta import compare_trees
ret = {}
basis = working.basis_tree()
if (bzrlib.version_info[0] == 0) and (bzrlib.version_info[1] < 9):
delta = compare_trees(basis, working, want_unchanged=True)
else:
delta = working.changes_from(basis, want_unchanged=True)
ret['unchanged'] = len(delta.unchanged)
ret['modified'] = len(delta.modified)
ret['added'] = len(delta.added)
ret['removed'] = len(delta.removed)
ret['renamed'] = len(delta.renamed)
ignore_cnt = unknown_cnt = 0
for path in working.extras():
if working.is_ignored(path):
ignore_cnt += 1
else:
unknown_cnt += 1
ret['unknown'] = unknown_cnt
ret['ignored'] = ignore_cnt
dir_cnt = 0
for path, ie in working.iter_entries_by_dir():
if ie.kind == 'directory':
dir_cnt += 1
ret['subdirs'] = dir_cnt
return ret
def get_branch_stats(branch):
""" Get statistics about a branch.
:return: a dictionary containing the needed infos
"""
ret = {}
repository = branch.repository
history = branch.revision_history()
revno = len(history)
ret['revno'] = revno
committers = {}
for rev in history:
committers[repository.get_revision(rev).committer] = True
ret['commiters'] = len(committers)
if revno > 0:
firstrev = repository.get_revision(history[0])
age = int((time.time() - firstrev.timestamp) / 3600 / 24)
ret['age'] = age
ret['firstrev'] = osutils.format_date(firstrev.timestamp,
firstrev.timezone)
lastrev = repository.get_revision(history[-1])
ret['lastrev'] = osutils.format_date(lastrev.timestamp,
lastrev.timezone)
return ret
def get_repository_stats(repository):
""" Get statistics about a repository.
:return: a dictionary containing the needed infos
"""
ret = {}
if repository.bzrdir.root_transport.listable():
c, t = repository._revision_store.total_size(repository.get_transaction())
ret['revisions'] = c
ret['size'] = t
return ret
def diff_helper(tree, specific_files, external_diff_options,
old_revision_spec=None, new_revision_spec=None,
old_label='a/', new_label='b/', output=None):
""" Helper for diff.
:param tree: a WorkingTree
:param specific_files: the specific files to compare, or None
:param external_diff_options: if non-None, run an external diff, and pass it these options
:param old_revision_spec: if None, use basis tree as old revision, otherwise use the tree for the specified revision.
:param new_revision_spec: if None, use working tree as new revision, otherwise use the tree for the specified revision.
"""
import sys
from bzrlib.diff import show_diff_trees
if output == None:
output = sys.stdout
def spec_tree(spec):
revision_id = spec.in_store(tree.branch).rev_id
return tree.branch.repository.revision_tree(revision_id)
if old_revision_spec is None:
old_tree = tree.basis_tree()
else:
old_tree = spec_tree(old_revision_spec)
if new_revision_spec is None:
new_tree = tree
else:
new_tree = spec_tree(new_revision_spec)
return show_diff_trees(old_tree, new_tree, output, specific_files,
external_diff_options,
old_label=old_label, new_label=new_label)
syntax highlighted by Code2HTML, v. 0.9.1