/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to olive/backend/init.py

  • Committer: Jelmer Vernooij
  • Date: 2006-09-05 01:20:15 UTC
  • mto: (0.8.83 merge)
  • mto: This revision was merged to the branch mainline in revision 83.
  • Revision ID: jelmer@samba.org-20060905012015-afc711bf6c67e963
Integrate olive.backend.init

Show diffs side-by-side

added added

removed removed

Lines of Context:
olive/backend/init.py
1
 
# Copyright (C) 2006 by Szilveszter Farkas (Phanatic) <szilveszter.farkas@gmail.com>
2
 
# Some parts of the code are:
3
 
# Copyright (C) 2005, 2006 by Canonical Ltd
4
 
 
5
 
# This program is free software; you can redistribute it and/or modify
6
 
# it under the terms of the GNU General Public License as published by
7
 
# the Free Software Foundation; either version 2 of the License, or
8
 
# (at your option) any later version.
9
 
 
10
 
# This program is distributed in the hope that it will be useful,
11
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
# GNU General Public License for more details.
14
 
 
15
 
# You should have received a copy of the GNU General Public License
16
 
# along with this program; if not, write to the Free Software
17
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
 
 
19
 
import errno
20
 
import os
21
 
 
22
 
import bzrlib
23
 
import bzrlib.bzrdir as bzrdir
24
 
 
25
 
from bzrlib.branch import Branch
26
 
 
27
 
from bzrlib.errors import (AlreadyBranchError, BranchExistsWithoutWorkingTree)
28
 
 
29
 
def branch(from_location, to_location, revision=None):
30
 
    """ Create a branch from a local/remote location.
31
 
    
32
 
    :param from_location: the original location of the branch
33
 
    
34
 
    :param to_location: the directory where the branch should be created
35
 
    
36
 
    :param revision: if specified, the given revision will be branched
37
 
    
38
 
    :return: number of revisions branched
39
 
    """
40
 
    from bzrlib.transport import get_transport
41
 
 
42
 
    br_from = Branch.open(from_location)
43
 
 
44
 
    br_from.lock_read()
45
 
 
46
 
    try:
47
 
        basis_dir = None
48
 
        if revision is not None:
49
 
            revision_id = br_from.get_rev_id(revision)
50
 
        else:
51
 
            revision_id = None
52
 
 
53
 
        to_location = to_location + '/' + os.path.basename(from_location.rstrip("/\\"))
54
 
        name = None
55
 
        to_transport = get_transport(to_location)
56
 
 
57
 
        to_transport.mkdir('.')
58
 
 
59
 
        try:
60
 
            dir = br_from.bzrdir.sprout(to_transport.base, revision_id, basis_dir)
61
 
            branch = dir.open_branch()
62
 
        except NoSuchRevision:
63
 
            to_transport.delete_tree('.')
64
 
            raise
65
 
 
66
 
        if name:
67
 
            branch.control_files.put_utf8('branch-name', name)
68
 
    finally:
69
 
        br_from.unlock()
70
 
        
71
 
    return branch.revno()
72
 
 
73
 
def checkout(branch_location, to_location, revision=None, lightweight=False):
74
 
    """ Create a new checkout of an existing branch.
75
 
    
76
 
    :param branch_location: the location of the branch you'd like to checkout
77
 
    
78
 
    :param to_location: the directory where the checkout should be created
79
 
    
80
 
    :param revision: the given revision will be checkout'ed (be aware!)
81
 
    
82
 
    :param lightweight: perform a lightweight checkout (be aware!)
83
 
    """
84
 
    source = Branch.open(branch_location)
85
 
    
86
 
    if revision is not None:
87
 
        revision_id = source.get_rev_id(revision)
88
 
    else:
89
 
        revision_id = None
90
 
 
91
 
    # if the source and to_location are the same, 
92
 
    # and there is no working tree,
93
 
    # then reconstitute a branch
94
 
    if (bzrlib.osutils.abspath(to_location) ==
95
 
        bzrlib.osutils.abspath(branch_location)):
96
 
        try:
97
 
            source.bzrdir.open_workingtree()
98
 
        except NoWorkingTree:
99
 
            source.bzrdir.create_workingtree()
100
 
            return
101
 
 
102
 
    to_location = to_location + '/' + os.path.basename(branch_location.rstrip("/\\"))
103
 
    
104
 
    os.mkdir(to_location)
105
 
 
106
 
    old_format = bzrlib.bzrdir.BzrDirFormat.get_default_format()
107
 
    bzrlib.bzrdir.BzrDirFormat.set_default_format(bzrdir.BzrDirMetaFormat1())
108
 
 
109
 
    try:
110
 
        if lightweight:
111
 
            checkout = bzrdir.BzrDirMetaFormat1().initialize(to_location)
112
 
            bzrlib.branch.BranchReferenceFormat().initialize(checkout, source)
113
 
        else:
114
 
            checkout_branch = bzrlib.bzrdir.BzrDir.create_branch_convenience(
115
 
                to_location, force_new_tree=False)
116
 
            checkout = checkout_branch.bzrdir
117
 
            checkout_branch.bind(source)
118
 
            if revision_id is not None:
119
 
                rh = checkout_branch.revision_history()
120
 
                checkout_branch.set_revision_history(rh[:rh.index(revision_id) + 1])
121
 
 
122
 
        checkout.create_workingtree(revision_id)
123
 
    finally:
124
 
        bzrlib.bzrdir.BzrDirFormat.set_default_format(old_format)
125
 
 
126
 
def init(location):
127
 
    """ Initialize a directory.
128
 
    
129
 
    :param location: full path to the directory you want to be versioned
130
 
    """
131
 
    from bzrlib.builtins import get_format_type
132
 
 
133
 
    format = get_format_type('default')
134
 
 
135
 
    if not os.path.exists(location):
136
 
        os.mkdir(location)
137
 
 
138
 
    try:
139
 
        existing_bzrdir = bzrdir.BzrDir.open(location)
140
 
    except NotBranchError:
141
 
        bzrdir.BzrDir.create_branch_convenience(location, format=format)
142
 
    else:
143
 
        if existing_bzrdir.has_branch():
144
 
            if existing_bzrdir.has_workingtree():
145
 
                raise AlreadyBranchError(location)
146
 
            else:
147
 
                raise BranchExistsWithoutWorkingTree(location)
148
 
        else:
149
 
            existing_bzrdir.create_branch()
150
 
            existing_bzrdir.create_workingtree()