13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
17
"""bzr upgrade logic."""
20
from bzrlib.bzrdir import BzrDir, BzrDirFormat
20
from bzrlib.bzrdir import BzrDir, BzrDirFormat, format_registry
21
21
import bzrlib.errors as errors
22
22
from bzrlib.remote import RemoteBzrDir
23
23
from bzrlib.transport import get_transport
27
27
class Convert(object):
29
def __init__(self, url, format):
31
format = BzrDirFormat.get_default_format()
29
def __init__(self, url, format=None):
32
30
self.format = format
33
31
self.bzrdir = BzrDir.open_unsupported(url)
34
32
if isinstance(self.bzrdir, RemoteBzrDir):
53
51
branch.bzrdir.root_transport.base)
55
53
except (errors.NotBranchError, errors.IncompatibleRepositories):
56
# might not be a format we can open without upgrading; see e.g.
54
# might not be a format we can open without upgrading; see e.g.
57
55
# https://bugs.launchpad.net/bzr/+bug/253891
59
if not self.bzrdir.needs_format_conversion(self.format):
57
if self.format is None:
59
rich_root = self.bzrdir.find_repository()._format.rich_root_data
60
except errors.NoRepositoryPresent:
61
rich_root = False # assume no rich roots
63
format_name = "default-rich-root"
65
format_name = "default"
66
format = format_registry.make_bzrdir(format_name)
69
if not self.bzrdir.needs_format_conversion(format):
60
70
raise errors.UpToDateFormat(self.bzrdir._format)
61
71
if not self.bzrdir.can_convert_format():
62
72
raise errors.BzrError("cannot upgrade from bzrdir format %s" %
63
73
self.bzrdir._format)
64
self.bzrdir.check_conversion_target(self.format)
74
self.bzrdir.check_conversion_target(format)
65
75
self.pb.note('starting upgrade of %s', self.transport.base)
66
76
self.bzrdir.backup_bzrdir()
67
while self.bzrdir.needs_format_conversion(self.format):
68
converter = self.bzrdir._format.get_converter(self.format)
77
while self.bzrdir.needs_format_conversion(format):
78
converter = self.bzrdir._format.get_converter(format)
69
79
self.bzrdir = converter.convert(self.bzrdir, self.pb)
70
80
self.pb.note("finished")