/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/smart/repository.py

Allow Repository tests to be backed onto a specific VFS as needed.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/smart/repository.py
 
1
# Copyright (C) 2006 Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
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
 
16
 
 
17
"""Server-side repository related request implmentations."""
 
18
 
 
19
 
 
20
from bzrlib import errors
 
21
from bzrlib.bzrdir import BzrDir
 
22
from bzrlib.smart.request import SmartServerRequest, SmartServerResponse
 
23
 
 
24
 
 
25
class SmartServerRepositoryRequest(SmartServerRequest):
 
26
    """Common base class for Repository requests."""
 
27
 
 
28
    def do(self, path, *args):
 
29
        """Execute a repository request.
 
30
        
 
31
        The repository must be at the exact path - no searching is done.
 
32
 
 
33
        The actual logic is delegated to self.do_repository_request.
 
34
 
 
35
        :param path: The path for the repository.
 
36
        :return: A smart server from self.do_repository_request().
 
37
        """
 
38
        transport = self._backing_transport.clone(path)
 
39
        bzrdir = BzrDir.open_from_transport(transport)
 
40
        repository = bzrdir.open_repository()
 
41
        return self.do_repository_request(repository, *args)
 
42
 
 
43
 
 
44
class SmartServerRequestHasRevision(SmartServerRepositoryRequest):
 
45
 
 
46
    def do_repository_request(self, repository, revision_id):
 
47
        """Return ok if a specific revision is in the repository at path.
 
48
 
 
49
        :param repository: The repository to query in.
 
50
        :param revision_id: The utf8 encoded revision_id to lookup.
 
51
        :return: A smart server response of ('ok', ) if the revision is
 
52
            present.
 
53
        """
 
54
        decoded_revision_id = revision_id.decode('utf8')
 
55
        if repository.has_revision(decoded_revision_id):
 
56
            return SmartServerResponse(('ok', ))
 
57
        else:
 
58
            return SmartServerResponse(('no', ))
 
59
 
 
60
 
 
61
class SmartServerRepositoryIsShared(SmartServerRepositoryRequest):
 
62
 
 
63
    def do_repository_request(self, repository):
 
64
        """Return the result of repository.is_shared().
 
65
 
 
66
        :param repository: The repository to query in.
 
67
        :return: A smart server response of ('yes', ) if the repository is
 
68
            shared, and ('no', ) if it is not.
 
69
        """
 
70
        if repository.is_shared():
 
71
            return SmartServerResponse(('yes', ))
 
72
        else:
 
73
            return SmartServerResponse(('no', ))