/stdhome

To get this branch, use:
bzr branch http://bzr.ed.am/stdhome

« back to all changes in this revision

Viewing changes to lib/stdhome/command_init.py

  • Committer: Tim Marston
  • Date: 2013-12-08 19:28:11 UTC
  • Revision ID: tim@ed.am-20131208192811-r20qj7cgmn4duw11
initial commit; basic app startup and initial command-line processing

Show diffs side-by-side

added added

removed removed

lib/stdhome/command_init.py
1
 
# init.py
 
1
# command_init.py
2
2
#
3
3
# Copyright (C) 2013 Tim Marston <tim@edm.am>
4
4
#
19
19
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
20
20
 
21
21
 
22
 
import sys, os, re, getopt, shutil, subprocess
23
 
from command import Command
24
 
import stdhome.the as the
25
 
from stdhome.deployment import Deployment
26
 
from stdhome.subprocess import Popen
27
 
 
28
 
 
29
 
class InitCommand( Command ):
30
 
 
31
 
 
32
 
        def __init__( self ):
33
 
                self.repo = None
34
 
 
 
22
import sys, os
 
23
import getopt
 
24
 
 
25
 
 
26
class CommandInit:
35
27
 
36
28
        def print_help( self ):
37
 
                print "Usage: " + the.program.name + " init [URL] [--repo=REPO]"
 
29
                print "Usage: " + self.program + " init [URL] [--repo=REPO]"
38
30
                print
39
31
                #      01234567890123456789012345678901234567890123456789012345678901234567890123456789
40
32
                print "Initialise a local repository."
46
38
                print "hostname, it is internally expanded to scp://HOSTNAME/~/.stdhome/REPO)."
47
39
                print
48
40
                print "Options:"
49
 
                print "  -r, --repo=REPO  select the repo to check-out or create (defaults to 'home')"
50
 
                print "  -v, --verbose    display information about what is being done"
51
 
                print "      --help       display help and exit"
 
41
                print "     --repo     select the repo to check-out or create (defaults to 'home')"
 
42
                print "     --help     display help and exit"
52
43
                exit( 0 )
53
44
 
54
45
 
55
 
        def parse_command_line( self ):
56
 
                opts, args = getopt.gnu_getopt(
57
 
                        sys.argv[ 1: ], "r:v",
58
 
                        [ "repo=", "verbose", "help" ] )
59
 
                for opt, optarg in opts:
60
 
                        if opt in [ '--repo', '-r' ]:
61
 
                                if not re.match( '^[-a-zA-z0-9.]+$', optarg ):
62
 
                                        raise the.program.FatalError(
63
 
                                                'invalid repository name: ' + optarg )
 
46
        def run( self, stdhome ):
 
47
                # parse command line
 
48
                try:
 
49
                        opts, args = getopt.gnu_getopt(
 
50
                                sys.argv[ 1: ], "",
 
51
                                [ "repo:", "help" ] )
 
52
                except getopt.GetoptError as e:
 
53
                        stdhome.print_usage( e )
 
54
                for opt, optargs in opts:
 
55
                        if opt == "--help":
 
56
                                self.print_help()
 
57
                        elif aopt == "--repo":
64
58
                                self.repo = optarg
65
 
                        elif opt in [ '--verbose', '-v' ]:
66
 
                                the.verbose += 1
67
 
                        elif opt == "--help":
68
 
                                self.print_help()
69
 
                
70
 
                # discard first argument (the command)
71
 
                args.pop( 0 )
72
 
 
73
 
                # URL argument
74
 
                self.url = args[ 0 ].strip() if len( args ) else None
75
 
 
76
 
                # remaining arguments
77
 
                if len( args ) > 1:
78
 
                        raise the.program.UsageError( 'too many arguments' )
79
 
 
80
 
 
81
 
        def run( self ):
82
 
 
83
 
                # set up repo and check it *doesn't* already exists
84
 
                the.set_repo( self.repo )
85
 
                the.repo.check_dir_exists( False )
86
 
 
87
 
                # ensure our top-level directory exists
88
 
                if not os.path.exists( the.full_dir ):
89
 
                        os.mkdir( the.full_dir )
90
 
 
91
 
                # checkout a remote repo, or create an empty local one?
92
 
                if self.url:
93
 
 
94
 
                        # expand url if it's a simple hostname
95
 
                        if re.match( '^[0-9a-zA-z.]+$', self.url ):
96
 
                                self.url = 'sftp://%s/%s/%s' % \
97
 
                                                   ( self.url, the.dir, the.repo.name )
98
 
 
99
 
                        # initialise deployment (with an empty repo)
100
 
                        deployment = Deployment()
101
 
                        deployment.copy_in()
102
 
 
103
 
                        # perform bzr checkout
104
 
                        if the.verbose: print "checking out %s" % the.repo.dir
105
 
                        try:
106
 
                                the.repo.vcs.checkout( self.url )
107
 
                        except Exception as e:
108
 
 
109
 
                                # attempt to clean-up repo dir
110
 
                                try:
111
 
                                        shutil.rmtree( the.repo.full_dir )
112
 
                                except OSError:
113
 
                                        pass
114
 
 
115
 
                                raise e
116
 
 
117
 
                        # perform deployment
118
 
                        try:
119
 
                                deployment.copy_out()
120
 
                        except deployment.Conflict as e:
121
 
                                raise the.program.FatalError( e.msg )
122
 
 
123
 
                else:
124
 
 
125
 
                        # perform bzr init
126
 
                        if the.verbose: print 'initialising %s' % the.repo.dir
127
 
                        try:
128
 
                                the.repo.vcs.init()
129
 
                        except Exception as e:
130
 
 
131
 
                                # attempt to clean-up repo dir, and die
132
 
                                try:
133
 
                                        shutil.rmtree( the.repo.full_dir )
134
 
                                except OSError:
135
 
                                        pass
136
 
 
137
 
                                raise e
 
59
 
 
60
                print "init"