summaryrefslogtreecommitdiff
path: root/morphlib/workspace.py
blob: 3a2269c809ef855762c40bbb80261a107822668b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# Copyright (C) 2013  Codethink Limited
#
# 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; version 2 of the License.
#
# 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.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# =*= License: GPL-2 =*=


'''A module to create, query, and manipulate Morph workspaces.'''


import os

import morphlib


class WorkspaceDirExists(morphlib.Error):

    def __init__(self, dirname):
        self.msg = (
            'can only initialize empty directory as a workspace: %s' %
            dirname)


class NotInWorkspace(morphlib.Error):

    def __init__(self, dirname):
        self.msg = (
            "Can't find the workspace directory.\n"
            "Morph must be built and deployed within the "
                "system branch checkout within the workspace directory.")


class Workspace(object):

    '''A Morph workspace.

    This class should be instantiated with the open() or create()
    functions in this module.

    '''

    def __init__(self, root_directory):
        self.root = root_directory


def open(dirname):
    '''Open an existing workspace.

    The given directory name may be to a subdirectory of the
    workspace. This makes it easy to instantiate the Workspace
    class even when the user invokes Morph in a subdirectory.
    The workspace MUST exist already, or NotInWorkspace is
    raised.

    Return a Workspace instance.

    '''

    root = _find_root(dirname)
    if root is None:
        raise NotInWorkspace(dirname)
    return Workspace(root)


def create(dirname):
    '''Create a new workspace.

    The given directory must not be inside an existing workspace.
    The workspace directory is created, unless it already exists.  If it
    does exist, it must be empty. Otherwise WorkspaceDirExists is raised.

    '''

    root = _find_root(dirname)
    if root is not None:
        raise WorkspaceDirExists(root)

    if os.path.exists(dirname):
        if os.listdir(dirname):
            raise WorkspaceDirExists(dirname)
    else:
        os.mkdir(dirname)
    os.mkdir(os.path.join(dirname, '.morph'))
    return Workspace(dirname)


def _find_root(dirname):
    '''Find the workspace root directory at or above a given directory.'''

    dirname = os.path.normpath(os.path.abspath(dirname))
    while not os.path.isdir(os.path.join(dirname, '.morph')):
        if dirname == '/':
            return None
        dirname = os.path.dirname(dirname)
    return dirname