~azzar1/unity/add-show-desktop-key

« back to all changes in this revision

Viewing changes to setup.py

  • Committer: Matt Giuca
  • Date: 2010-02-18 08:33:51 UTC
  • Revision ID: matt.giuca@gmail.com-20100218083351-l6be6fpe244u2x4c
svnlogservice: utf-8 decoding of the relevant parts of the dictionary, so they are put into JSON as unicode strings, not UTF-8 strings which are decoded wrongly. svnlog now displays unicode characters correctly. This is the final fix in Launchpad Bug #523495 -- all Subversion filename issues fixed (except the crash on log and diff directly on a non-ASCII filename, covered by bug #523500).

Show diffs side-by-side

added added

removed removed

Lines of Context:
21
21
# Date:   12/12/2007
22
22
 
23
23
# This is a command-line application, for use by the administrator.
24
 
# This program configures, builds and installs IVLE in three separate steps.
 
24
# This program is a frontend for the modules in the setup packages that 
 
25
# build and install IVLE in separate steps.
25
26
# It is called with at least one argument, which specifies which operation to
26
27
# take.
27
28
 
28
 
# setup.py listmake (for developer use only)
29
 
# Recurses through the source tree and builds a list of all files which should
30
 
# be copied upon installation. This should be run by the developer before
31
 
# cutting a distribution, and the listfile it generates should be included in
32
 
# the distribution, avoiding the administrator having to run it.
33
 
 
34
 
# setup.py config [args]
35
 
# Configures IVLE with machine-specific details, most notably, various paths.
36
 
# Either prompts the administrator for these details or accepts them as
37
 
# command-line args.
38
 
# Creates lib/conf/conf.py and trampoline/conf.h.
39
 
 
40
 
# setup.py build
41
 
# Compiles all files and sets up a jail template in the source directory.
42
 
# Details:
43
 
# Compiles (GCC) trampoline/trampoline.c to trampoline/trampoline.
44
 
# Creates jail/.
45
 
# Creates standard subdirs inside the jail, eg bin, opt, home, tmp.
46
 
# Copies console/ to a location within the jail.
47
 
# Copies OS programs and files to corresponding locations within the jail
48
 
#   (eg. python and Python libs, ld.so, etc).
49
 
# Generates .pyc files for all the IVLE .py files.
50
 
 
51
 
# setup.py install [--nojail] [--dry|n]
52
 
# (Requires root)
53
 
# Create target install directory ($target).
54
 
# Create $target/bin.
55
 
# Copy trampoline/trampoline to $target/bin.
56
 
# chown and chmod the installed trampoline.
57
 
# Copy www/ to $target.
58
 
# Copy jail/ to jails template directory (unless --nojail specified).
59
 
 
60
 
import os
61
 
import stat
62
 
import shutil
63
29
import sys
64
 
import getopt
65
 
import string
66
 
import errno
67
 
import mimetypes
68
 
import compileall
69
 
import getopt
70
 
 
71
 
# Import modules from the website is tricky since they're in the www
72
 
# directory.
73
 
sys.path.append(os.path.join(os.getcwd(), 'lib'))
74
 
import conf
75
 
import common.makeuser
76
 
 
77
 
# Determine which Python version (2.4 or 2.5, for example) we are running,
78
 
# and use that as the filename to the Python directory.
79
 
# Just get the first 3 characters of sys.version.
80
 
PYTHON_VERSION = sys.version[0:3]
81
 
 
82
 
# Operating system files to copy over into the jail.
83
 
# These will be copied from the given place on the OS file system into the
84
 
# same place within the jail.
85
 
JAIL_FILES = [
86
 
    '/lib/ld-linux.so.2',
87
 
    '/lib/tls/i686/cmov/libc.so.6',
88
 
    '/lib/tls/i686/cmov/libdl.so.2',
89
 
    '/lib/tls/i686/cmov/libm.so.6',
90
 
    '/lib/tls/i686/cmov/libpthread.so.0',
91
 
    '/lib/tls/i686/cmov/libutil.so.1',
92
 
    '/etc/ld.so.conf',
93
 
    '/etc/ld.so.cache',
94
 
    # These 2 files do not exist in Ubuntu
95
 
    #'/etc/ld.so.preload',
96
 
    #'/etc/ld.so.nohwcap',
97
 
    # UNIX commands
98
 
    '/usr/bin/strace',
99
 
    '/bin/ls',
100
 
    '/bin/echo',
101
 
    # Needed by python
102
 
    '/usr/bin/python%s' % PYTHON_VERSION,
103
 
    # Needed by fileservice
104
 
    '/lib/libcom_err.so.2',
105
 
    '/lib/libcrypt.so.1',
106
 
    '/lib/libkeyutils.so.1',
107
 
    '/lib/libresolv.so.2',
108
 
    '/lib/librt.so.1',
109
 
    '/lib/libuuid.so.1',
110
 
    '/usr/lib/libapr-1.so.0',
111
 
    '/usr/lib/libaprutil-1.so.0',
112
 
    '/usr/lib/libdb-4.4.so',
113
 
    '/usr/lib/libexpat.so.1',
114
 
    '/usr/lib/libgcrypt.so.11',
115
 
    '/usr/lib/libgnutls.so.13',
116
 
    '/usr/lib/libgpg-error.so.0',
117
 
    '/usr/lib/libgssapi_krb5.so.2',
118
 
    '/usr/lib/libk5crypto.so.3',
119
 
    '/usr/lib/libkrb5.so.3',
120
 
    '/usr/lib/libkrb5support.so.0',
121
 
    '/usr/lib/liblber.so.2',
122
 
    '/usr/lib/libldap_r.so.2',
123
 
    '/usr/lib/libneon.so.26',
124
 
    '/usr/lib/libpq.so.5',
125
 
    '/usr/lib/libsasl2.so.2',
126
 
    '/usr/lib/libsqlite3.so.0',
127
 
    '/usr/lib/libsvn_client-1.so.1',
128
 
    '/usr/lib/libsvn_delta-1.so.1',
129
 
    '/usr/lib/libsvn_diff-1.so.1',
130
 
    '/usr/lib/libsvn_fs-1.so.1',
131
 
    '/usr/lib/libsvn_fs_base-1.so.1',
132
 
    '/usr/lib/libsvn_fs_fs-1.so.1',
133
 
    '/usr/lib/libsvn_ra-1.so.1',
134
 
    '/usr/lib/libsvn_ra_dav-1.so.1',
135
 
    '/usr/lib/libsvn_ra_local-1.so.1',
136
 
    '/usr/lib/libsvn_ra_svn-1.so.1',
137
 
    '/usr/lib/libsvn_repos-1.so.1',
138
 
    '/usr/lib/libsvn_subr-1.so.1',
139
 
    '/usr/lib/libsvn_wc-1.so.1',
140
 
    '/usr/lib/libtasn1.so.3',
141
 
    '/usr/lib/libxml2.so.2',
142
 
    # Needed by matplotlib
143
 
    '/usr/lib/i686/cmov/libssl.so.0.9.8',
144
 
    '/usr/lib/i686/cmov/libcrypto.so.0.9.8',
145
 
    '/lib/tls/i686/cmov/libnsl.so.1',
146
 
    '/usr/lib/libz.so.1',
147
 
    '/usr/lib/atlas/liblapack.so.3',
148
 
    '/usr/lib/atlas/libblas.so.3',
149
 
    '/usr/lib/libg2c.so.0',
150
 
    '/usr/lib/libstdc++.so.6',
151
 
    '/usr/lib/libfreetype.so.6',
152
 
    '/usr/lib/libpng12.so.0',
153
 
    '/usr/lib/libBLT.2.4.so.8.4',
154
 
    '/usr/lib/libtk8.4.so.0',
155
 
    '/usr/lib/libtcl8.4.so.0',
156
 
    '/usr/lib/tcl8.4/init.tcl',
157
 
    '/usr/lib/libX11.so.6',
158
 
    '/usr/lib/libXau.so.6',
159
 
    '/usr/lib/libXdmcp.so.6',
160
 
    '/lib/libgcc_s.so.1',
161
 
    '/etc/matplotlibrc',
162
 
    # Needed for resolv
163
 
    '/lib/libnss_dns.so.2',
164
 
    #'/lib/libnss_mdns4.so',
165
 
    '/etc/hosts',
166
 
    '/etc/resolv.conf',
167
 
    #'/etc/hosts.conf',
168
 
    #'/etc/hostname',
169
 
    '/etc/nsswitch.conf',
170
 
    '/lib/libnss_files.so.2',
171
 
]
172
 
# Symlinks to make within the jail. Src mapped to dst.
173
 
JAIL_LINKS = {
174
 
    'python%s' % PYTHON_VERSION: 'jail/usr/bin/python',
175
 
}
176
 
# Trees to copy. Src mapped to dst (these will be passed to action_copytree).
177
 
JAIL_COPYTREES = {
178
 
    '/usr/lib/python%s' % PYTHON_VERSION:
179
 
        'jail/usr/lib/python%s' % PYTHON_VERSION,
180
 
    '/usr/share/matplotlib': 'jail/usr/share/matplotlib',
181
 
    '/etc/ld.so.conf.d': 'jail/etc/ld.so.conf.d',
182
 
}
183
 
 
184
 
class ConfigOption:
185
 
    """A configuration option; one of the things written to conf.py."""
186
 
    def __init__(self, option_name, default, prompt, comment):
187
 
        """Creates a configuration option.
188
 
        option_name: Name of the variable in conf.py. Also name of the
189
 
            command-line argument to setup.py conf.
190
 
        default: Default value for this variable.
191
 
        prompt: (Short) string presented during the interactive prompt in
192
 
            setup.py conf.
193
 
        comment: (Long) comment string stored in conf.py. Each line of this
194
 
            string should begin with a '#'.
195
 
        """
196
 
        self.option_name = option_name
197
 
        self.default = default
198
 
        self.prompt = prompt
199
 
        self.comment = comment
200
 
 
201
 
# Configuration options, defaults and descriptions
202
 
config_options = []
203
 
config_options.append(ConfigOption("root_dir", "/",
204
 
    """Root directory where IVLE is located (in URL space):""",
205
 
    """
206
 
# In URL space, where in the site is IVLE located. (All URLs will be prefixed
207
 
# with this).
208
 
# eg. "/" or "/ivle"."""))
209
 
config_options.append(ConfigOption("ivle_install_dir", "/opt/ivle",
210
 
    'Root directory where IVLE will be installed (on the local file '
211
 
    'system):',
212
 
    """
213
 
# In the local file system, where IVLE is actually installed.
214
 
# This directory should contain the "www" and "bin" directories."""))
215
 
config_options.append(ConfigOption("jail_base", "/home/informatics/jails",
216
 
    """Location of Directories
217
 
=======================
218
 
Root directory where the jails (containing user files) are stored
219
 
(on the local file system):""",
220
 
    """
221
 
# In the local file system, where are the student/user file spaces located.
222
 
# The user jails are expected to be located immediately in subdirectories of
223
 
# this location."""))
224
 
config_options.append(ConfigOption("subjects_base",
225
 
    "/home/informatics/subjects",
226
 
    """Root directory where the subject directories (containing worksheets
227
 
and other per-subject files) are stored (on the local file system):""",
228
 
    """
229
 
# In the local file system, where are the per-subject file spaces located.
230
 
# The individual subject directories are expected to be located immediately
231
 
# in subdirectories of this location."""))
232
 
config_options.append(ConfigOption("exercises_base",
233
 
    "/home/informatics/exercises",
234
 
    """Root directory where the exercise directories (containing
235
 
subject-independent exercise sheets) are stored (on the local file
236
 
system):""",
237
 
    """
238
 
# In the local file system, where are the subject-independent exercise sheet
239
 
# file spaces located."""))
240
 
config_options.append(ConfigOption("public_host", "public.localhost",
241
 
    """Hostname which will cause the server to go into "public mode",
242
 
providing login-free access to student's published work:""",
243
 
    """
244
 
# The server goes into "public mode" if the browser sends a request with this
245
 
# host. This is for security reasons - we only serve public student files on a
246
 
# separate domain to the main IVLE site.
247
 
# Public mode does not use cookies, and serves only public content.
248
 
# Private mode (normal mode) requires login, and only serves files relevant to
249
 
# the logged-in user."""))
250
 
config_options.append(ConfigOption("allowed_uids", "33",
251
 
    """UID of the web server process which will run IVLE.
252
 
Only this user may execute the trampoline. May specify multiple users as
253
 
a comma-separated list.
254
 
    (eg. "1002,78")""",
255
 
    """
256
 
# The User-ID of the web server process which will run IVLE, and any other
257
 
# users who are allowed to run the trampoline. This is stores as a string of
258
 
# comma-separated integers, simply because it is not used within Python, only
259
 
# used by the setup program to write to conf.h (see setup.py config)."""))
260
 
config_options.append(ConfigOption("db_host", "localhost",
261
 
    """PostgreSQL Database config
262
 
==========================
263
 
Hostname of the DB server:""",
264
 
    """
265
 
### PostgreSQL Database config ###
266
 
# Database server hostname"""))
267
 
config_options.append(ConfigOption("db_port", "5432",
268
 
    """Port of the DB server:""",
269
 
    """
270
 
# Database server port"""))
271
 
config_options.append(ConfigOption("db_dbname", "ivle",
272
 
    """Database name:""",
273
 
    """
274
 
# Database name"""))
275
 
config_options.append(ConfigOption("db_user", "postgres",
276
 
    """Username for DB server login:""",
277
 
    """
278
 
# Database username"""))
279
 
config_options.append(ConfigOption("db_password", "",
280
 
    """Password for DB server login:
281
 
    (Caution: This password is stored in plaintext in lib/conf/conf.py)""",
282
 
    """
283
 
# Database password"""))
284
 
config_options.append(ConfigOption("auth_modules", "ldap_auth",
285
 
    """Authentication config
286
 
=====================
287
 
Comma-separated list of authentication modules. Only "ldap" is available
288
 
by default.""",
289
 
    """
290
 
# Comma-separated list of authentication modules.
291
 
# These refer to importable Python modules in the www/auth directory.
292
 
# Modules "ldap" and "guest" are available in the source tree, but
293
 
# other modules may be plugged in to auth against organisation-specific
294
 
# auth backends."""))
295
 
config_options.append(ConfigOption("ldap_url", "ldaps://www.example.com",
296
 
    """(LDAP options are only relevant if "ldap" is included in the list of
297
 
auth modules).
298
 
URL for LDAP authentication server:""",
299
 
    """
300
 
# URL for LDAP authentication server"""))
301
 
config_options.append(ConfigOption("ldap_format_string",
302
 
    "uid=%s,ou=users,o=example",
303
 
    """Format string for LDAP auth request:
304
 
    (Must contain a single "%s" for the user's login name)""",
305
 
    """
306
 
# Format string for LDAP auth request
307
 
# (Must contain a single "%s" for the user's login name)"""))
308
 
config_options.append(ConfigOption("svn_addr", "http://svn.localhost/",
309
 
    """Subversion config
310
 
=================
311
 
The base url for accessing subversion repositories:""",
312
 
    """
313
 
# The base url for accessing subversion repositories."""))
314
 
config_options.append(ConfigOption("svn_conf", "/opt/ivle/svn/svn.conf",
315
 
    """The location of the subversion configuration file used by apache
316
 
to host the user repositories:""",
317
 
    """
318
 
# The location of the subversion configuration file used by
319
 
# apache to host the user repositories."""))
320
 
config_options.append(ConfigOption("svn_repo_path", "/home/informatics/repositories",
321
 
    """The root directory for the subversion repositories:""",
322
 
    """
323
 
# The root directory for the subversion repositories."""))
324
 
config_options.append(ConfigOption("svn_auth_ivle", "/opt/ivle/svn/ivle.auth",
325
 
    """The location of the password file used to authenticate users
326
 
of the subversion repository from the ivle server:""",
327
 
    """
328
 
# The location of the password file used to authenticate users
329
 
# of the subversion repository from the ivle server."""))
330
 
config_options.append(ConfigOption("svn_auth_local", "/opt/ivle/svn/local.auth",
331
 
    """The location of the password file used to authenticate local users
332
 
of the subversion repository:""",
333
 
    """
334
 
# The location of the password file used to authenticate local users
335
 
# of the subversion repository."""))
336
 
config_options.append(ConfigOption("usrmgt_host", "localhost",
337
 
    """User Management Server config
338
 
============================
339
 
The hostname where the usrmgt-server runs:""",
340
 
    """
341
 
# The hostname where the usrmgt-server runs."""))
342
 
config_options.append(ConfigOption("usrmgt_port", "2178",
343
 
    """The port where the usrmgt-server runs:""",
344
 
    """
345
 
# The port where the usrmgt-server runs."""))
346
 
config_options.append(ConfigOption("usrmgt_magic", "",
347
 
    """The password for the usrmgt-server:""",
348
 
    """
349
 
# The password for the usrmgt-server."""))
350
 
 
351
 
# Try importing existing conf, but if we can't just set up defaults
352
 
# The reason for this is that these settings are used by other phases
353
 
# of setup besides conf, so we need to know them.
354
 
# Also this allows you to hit Return to accept the existing value.
355
 
try:
356
 
    confmodule = __import__("lib/conf/conf")
357
 
    for opt in config_options:
358
 
        try:
359
 
            globals()[opt.option_name] = confmodule.__dict__[opt.option_name]
360
 
        except:
361
 
            globals()[opt.option_name] = opt.default
362
 
except ImportError:
363
 
    # Just set reasonable defaults
364
 
    for opt in config_options:
365
 
        globals()[opt.option_name] = opt.default
366
 
 
367
 
# Try importing install_list, but don't fail if we can't, because listmake can
368
 
# function without it.
369
 
try:
370
 
    import install_list
371
 
except:
372
 
    pass
373
 
 
374
 
# Mime types which will automatically be placed in the list by listmake.
375
 
# Note that listmake is not intended to be run by the final user (the system
376
 
# administrator who installs this), so the developers can customize the list
377
 
# as necessary, and include it in the distribution.
378
 
listmake_mimetypes = ['text/x-python', 'text/html',
379
 
    'application/x-javascript', 'application/javascript',
380
 
    'text/css', 'image/png', 'image/gif', 'application/xml']
381
 
 
382
 
# Main function skeleton from Guido van Rossum
383
 
# http://www.artima.com/weblogs/viewpost.jsp?thread=4829
 
30
import setup.build
 
31
import setup.install
384
32
 
385
33
def main(argv=None):
386
34
    if argv is None:
389
37
    # Print the opening spiel including the GPL notice
390
38
 
391
39
    print """IVLE - Informatics Virtual Learning Environment Setup
392
 
Copyright (C) 2007-2008 The University of Melbourne
 
40
Copyright (C) 2007-2009 The University of Melbourne
393
41
IVLE comes with ABSOLUTELY NO WARRANTY.
394
42
This is free software, and you are welcome to redistribute it
395
43
under certain conditions. See LICENSE.txt for details.
405
53
        help([])
406
54
        return 1
407
55
 
408
 
    # Disallow run as root unless installing
409
 
    if (operation != 'install' and operation != 'updatejails'
410
 
        and os.geteuid() == 0):
411
 
        print >>sys.stderr, "I do not want to run this stage as root."
412
 
        print >>sys.stderr, "Please run as a normal user."
413
 
        return 1
 
56
    oper_func = call_operator(operation)
 
57
    return oper_func(argv[2:])
 
58
 
 
59
def help(args):
 
60
    if len(args)!=1:
 
61
        print """Usage: python setup.py operation [options]
 
62
Operation can be:
 
63
    help [operation]
 
64
    build
 
65
    install
 
66
 
 
67
    For help and options for a specific operation use 'help [operation]'."""
 
68
    else:
 
69
        operator = args[0]
 
70
        oper_func = call_operator(operator)
 
71
        oper_func(['operator','--help'])
 
72
 
 
73
def call_operator(operation):
414
74
    # Call the requested operation's function
415
75
    try:
416
76
        oper_func = {
417
77
            'help' : help,
418
 
            'config' : conf,
419
 
            'build' : build,
420
 
            'listmake' : listmake,
421
 
            'install' : install,
422
 
            'updatejails' : updatejails,
 
78
            'build' : setup.build.build,
 
79
            'install' : setup.install.install,
423
80
        }[operation]
424
81
    except KeyError:
425
82
        print >>sys.stderr, (
426
83
            """Invalid operation '%s'. Try python setup.py help."""
427
84
            % operation)
428
 
        return 1
429
 
    return oper_func(argv[2:])
430
 
 
431
 
# Operation functions
432
 
 
433
 
def help(args):
434
 
    if args == []:
435
 
        print """Usage: python setup.py operation [args]
436
 
Operation (and args) can be:
437
 
    help [operation]
438
 
    listmake (developer use only)
439
 
    config [args]
440
 
    build
441
 
    install [--nojail] [--nosubjects] [-n|--dry]
442
 
"""
443
 
        return 1
444
 
    elif len(args) != 1:
445
 
        print """Usage: python setup.py help [operation]"""
446
 
        return 2
447
 
    else:
448
 
        operation = args[0]
449
 
 
450
 
    if operation == 'help':
451
 
        print """python setup.py help [operation]
452
 
Prints the usage message or detailed help on an operation, then exits."""
453
 
    elif operation == 'listmake':
454
 
        print """python setup.py listmake
455
 
(For developer use only)
456
 
Recurses through the source tree and builds a list of all files which should
457
 
be copied upon installation. This should be run by the developer before
458
 
cutting a distribution, and the listfile it generates should be included in
459
 
the distribution, avoiding the administrator having to run it."""
460
 
    elif operation == 'config':
461
 
        print """python setup.py config [args]
462
 
Configures IVLE with machine-specific details, most notably, various paths.
463
 
Either prompts the administrator for these details or accepts them as
464
 
command-line args. Will be interactive only if there are no arguments given.
465
 
Takes defaults from existing conf file if it exists.
466
 
 
467
 
To run IVLE out of the source directory (allowing development without having
468
 
to rebuild/install), just provide ivle_install_dir as the IVLE trunk
469
 
directory, and run build/install one time.
470
 
 
471
 
Creates lib/conf/conf.py and trampoline/conf.h.
472
 
 
473
 
Args are:"""
474
 
        for opt in config_options:
475
 
            print "    --" + opt.option_name
476
 
        print """As explained in the interactive prompt or conf.py.
477
 
"""
478
 
    elif operation == 'build':
479
 
        print """python -O setup.py build [--dry|-n]
480
 
Compiles all files and sets up a jail template in the source directory.
481
 
-O is recommended to cause compilation to be optimised.
482
 
Details:
483
 
Compiles (GCC) trampoline/trampoline.c to trampoline/trampoline.
484
 
Creates jail/.
485
 
Creates standard subdirs inside the jail, eg bin, opt, home, tmp.
486
 
Copies console/ to a location within the jail.
487
 
Copies OS programs and files to corresponding locations within the jail
488
 
  (eg. python and Python libs, ld.so, etc).
489
 
Generates .pyc or .pyo files for all the IVLE .py files.
490
 
 
491
 
--dry | -n  Print out the actions but don't do anything."""
492
 
    elif operation == 'install':
493
 
        print """sudo python setup.py install [--nojail] [--nosubjects][--dry|-n]
494
 
(Requires root)
495
 
Create target install directory ($target).
496
 
Create $target/bin.
497
 
Copy trampoline/trampoline to $target/bin.
498
 
chown and chmod the installed trampoline.
499
 
Copy www/ to $target.
500
 
Copy jail/ to jails template directory (unless --nojail specified).
501
 
Copy subjects/ to subjects directory (unless --nosubjects specified).
502
 
 
503
 
--nojail        Do not copy the jail.
504
 
--nosubjects    Do not copy the subjects and exercises directories.
505
 
--dry | -n  Print out the actions but don't do anything."""
506
 
    elif operation == 'updatejails':
507
 
        print """sudo python setup.py updatejails [--dry|-n]
508
 
(Requires root)
509
 
Copy jail/ to each subdirectory in jails directory.
510
 
 
511
 
--dry | -n  Print out the actions but don't do anything."""
512
 
    else:
513
 
        print >>sys.stderr, (
514
 
            """Invalid operation '%s'. Try python setup.py help."""
515
 
            % operation)
516
 
    return 1
517
 
 
518
 
def listmake(args):
519
 
    # We build two separate lists, by walking www and console
520
 
    list_www = build_list_py_files('www')
521
 
    list_lib = build_list_py_files('lib')
522
 
    list_subjects = build_list_py_files('subjects', no_top_level=True)
523
 
    list_exercises = build_list_py_files('exercises', no_top_level=True)
524
 
    list_scripts = [
525
 
        "scripts/python-console",
526
 
        "scripts/fileservice",
527
 
        "scripts/usrmgt-server",
528
 
        "scripts/diffservice",
529
 
    ]
530
 
    # Make sure that the files generated by conf are in the list
531
 
    # (since listmake is typically run before conf)
532
 
    if "lib/conf/conf.py" not in list_lib:
533
 
        list_lib.append("lib/conf/conf.py")
534
 
    # Write these out to a file
535
 
    cwd = os.getcwd()
536
 
    # the files that will be created/overwritten
537
 
    listfile = os.path.join(cwd, "install_list.py")
538
 
 
539
 
    try:
540
 
        file = open(listfile, "w")
541
 
 
542
 
        file.write("""# IVLE Configuration File
543
 
# install_list.py
544
 
# Provides lists of all files to be installed by `setup.py install' from
545
 
# certain directories.
546
 
# Note that any files with the given filename plus 'c' or 'o' (that is,
547
 
# compiled .pyc or .pyo files) will be copied as well.
548
 
 
549
 
# List of all installable files in www directory.
550
 
list_www = """)
551
 
        writelist_pretty(file, list_www)
552
 
        file.write("""
553
 
# List of all installable files in lib directory.
554
 
list_lib = """)
555
 
        writelist_pretty(file, list_lib)
556
 
        file.write("""
557
 
# List of all installable files in scripts directory.
558
 
list_scripts = """)
559
 
        writelist_pretty(file, list_scripts)
560
 
        file.write("""
561
 
# List of all installable files in subjects directory.
562
 
# This is to install sample subjects and material.
563
 
list_subjects = """)
564
 
        writelist_pretty(file, list_subjects)
565
 
        file.write("""
566
 
# List of all installable files in exercises directory.
567
 
# This is to install sample exercise material.
568
 
list_exercises = """)
569
 
        writelist_pretty(file, list_exercises)
570
 
 
571
 
        file.close()
572
 
    except IOError, (errno, strerror):
573
 
        print "IO error(%s): %s" % (errno, strerror)
574
 
        sys.exit(1)
575
 
 
576
 
    print "Successfully wrote install_list.py"
577
 
 
578
 
    print
579
 
    print ("You may modify the set of installable files before cutting the "
580
 
            "distribution:")
581
 
    print listfile
582
 
    print
583
 
 
584
 
    return 0
585
 
 
586
 
def build_list_py_files(dir, no_top_level=False):
587
 
    """Builds a list of all py files found in a directory and its
588
 
    subdirectories. Returns this as a list of strings.
589
 
    no_top_level=True means the file paths will not include the top-level
590
 
    directory.
591
 
    """
592
 
    pylist = []
593
 
    for (dirpath, dirnames, filenames) in os.walk(dir):
594
 
        # Exclude directories beginning with a '.' (such as '.svn')
595
 
        filter_mutate(lambda x: x[0] != '.', dirnames)
596
 
        # All *.py files are added to the list
597
 
        pylist += [os.path.join(dirpath, item) for item in filenames
598
 
            if mimetypes.guess_type(item)[0] in listmake_mimetypes]
599
 
    if no_top_level:
600
 
        for i in range(0, len(pylist)):
601
 
            _, pylist[i] = pylist[i].split(os.sep, 1)
602
 
    return pylist
603
 
 
604
 
def writelist_pretty(file, list):
605
 
    """Writes a list one element per line, to a file."""
606
 
    if list == []:
607
 
        file.write("[]\n")
608
 
    else:
609
 
        file.write('[\n')
610
 
        for elem in list:
611
 
            file.write('    %s,\n' % repr(elem))
612
 
        file.write(']\n')
613
 
 
614
 
def conf(args):
615
 
    global db_port, usrmgt_port
616
 
    # Set up some variables
617
 
 
618
 
    cwd = os.getcwd()
619
 
    # the files that will be created/overwritten
620
 
    conffile = os.path.join(cwd, "lib/conf/conf.py")
621
 
    jailconffile = os.path.join(cwd, "lib/conf/jailconf.py")
622
 
    conf_hfile = os.path.join(cwd, "trampoline/conf.h")
623
 
 
624
 
    # Get command-line arguments to avoid asking questions.
625
 
 
626
 
    optnames = []
627
 
    for opt in config_options:
628
 
        optnames.append(opt.option_name + "=")
629
 
    (opts, args) = getopt.gnu_getopt(args, "", optnames)
630
 
 
631
 
    if args != []:
632
 
        print >>sys.stderr, "Invalid arguments:", string.join(args, ' ')
633
 
        return 2
634
 
 
635
 
    if opts == []:
636
 
        # Interactive mode. Prompt the user for all the values.
637
 
 
638
 
        print """This tool will create the following files:
639
 
    %s
640
 
    %s
641
 
    %s
642
 
prompting you for details about your configuration. The file will be
643
 
overwritten if it already exists. It will *not* install or deploy IVLE.
644
 
 
645
 
Please hit Ctrl+C now if you do not wish to do this.
646
 
""" % (conffile, jailconffile, conf_hfile)
647
 
 
648
 
        # Get information from the administrator
649
 
        # If EOF is encountered at any time during the questioning, just exit
650
 
        # silently
651
 
 
652
 
        for opt in config_options:
653
 
            globals()[opt.option_name] = \
654
 
                query_user(globals()[opt.option_name], opt.prompt)
655
 
    else:
656
 
        opts = dict(opts)
657
 
        # Non-interactive mode. Parse the options.
658
 
        for opt in config_options:
659
 
            if '--' + opt.option_name in opts:
660
 
                globals()[opt.option_name] = opts['--' + opt.option_name]
661
 
 
662
 
    # Error handling on input values
663
 
    try:
664
 
        allowed_uids_list = map(int, allowed_uids.split(','))
665
 
    except ValueError:
666
 
        print >>sys.stderr, (
667
 
        "Invalid UID list (%s).\n"
668
 
        "Must be a comma-separated list of integers." % allowed_uids)
669
 
        return 1
670
 
    try:
671
 
        db_port = int(db_port)
672
 
        if db_port < 0 or db_port >= 65536: raise ValueError()
673
 
    except ValueError:
674
 
        print >>sys.stderr, (
675
 
        "Invalid DB port (%s).\n"
676
 
        "Must be an integer between 0 and 65535." % repr(db_port))
677
 
        return 1
678
 
    try:
679
 
        usrmgt_port = int(usrmgt_port)
680
 
        if usrmgt_port < 0 or usrmgt_port >= 65536: raise ValueError()
681
 
    except ValueError:
682
 
        print >>sys.stderr, (
683
 
        "Invalid user management port (%s).\n"
684
 
        "Must be an integer between 0 and 65535." % repr(usrmgt_port))
685
 
        return 1
686
 
 
687
 
    # Write lib/conf/conf.py
688
 
 
689
 
    try:
690
 
        conf = open(conffile, "w")
691
 
 
692
 
        conf.write("""# IVLE Configuration File
693
 
# conf.py
694
 
# Miscellaneous application settings
695
 
 
696
 
""")
697
 
        for opt in config_options:
698
 
            conf.write('%s\n%s = %s\n' % (opt.comment, opt.option_name,
699
 
                repr(globals()[opt.option_name])))
700
 
 
701
 
        conf.close()
702
 
    except IOError, (errno, strerror):
703
 
        print "IO error(%s): %s" % (errno, strerror)
704
 
        sys.exit(1)
705
 
 
706
 
    print "Successfully wrote lib/conf/conf.py"
707
 
 
708
 
    # Write conf/jailconf.py
709
 
 
710
 
    try:
711
 
        conf = open(jailconffile, "w")
712
 
 
713
 
        # In the "in-jail" version of conf, we don't need MOST of the details
714
 
        # (it would be a security risk to have them here).
715
 
        # So we just write root_dir, and jail_base is "/".
716
 
        # (jail_base being "/" means "jail-relative" paths are relative to "/"
717
 
        # when inside the jail.)
718
 
        conf.write("""# IVLE Configuration File
719
 
# conf.py
720
 
# Miscellaneous application settings
721
 
# (User jail version)
722
 
 
723
 
 
724
 
# In URL space, where in the site is IVLE located. (All URLs will be prefixed
725
 
# with this).
726
 
# eg. "/" or "/ivle".
727
 
root_dir = %s
728
 
 
729
 
# In the local file system, where are the student/user file spaces located.
730
 
# The user jails are expected to be located immediately in subdirectories of
731
 
# this location.
732
 
jail_base = '/'
733
 
 
734
 
# The hostname for serving publicly accessible pages
735
 
public_host = %s
736
 
""" % (repr(root_dir),repr(public_host)))
737
 
 
738
 
        conf.close()
739
 
    except IOError, (errno, strerror):
740
 
        print "IO error(%s): %s" % (errno, strerror)
741
 
        sys.exit(1)
742
 
 
743
 
    print "Successfully wrote lib/conf/jailconf.py"
744
 
 
745
 
    # Write trampoline/conf.h
746
 
 
747
 
    try:
748
 
        conf = open(conf_hfile, "w")
749
 
 
750
 
        conf.write("""/* IVLE Configuration File
751
 
 * conf.h
752
 
 * Administrator settings required by trampoline.
753
 
 * Note: trampoline will have to be rebuilt in order for changes to this file
754
 
 * to take effect.
755
 
 */
756
 
 
757
 
/* In the local file system, where are the jails located.
758
 
 * The trampoline does not allow the creation of a jail anywhere besides
759
 
 * jail_base or a subdirectory of jail_base.
760
 
 */
761
 
static const char* jail_base = "%s";
762
 
 
763
 
/* Which user IDs are allowed to run the trampoline.
764
 
 * This list should be limited to the web server user.
765
 
 * (Note that root is an implicit member of this list).
766
 
 */
767
 
static const int allowed_uids[] = { %s };
768
 
""" % (repr(jail_base)[1:-1], repr(allowed_uids_list)[1:-1]))
769
 
    # Note: The above uses PYTHON reprs, not C reprs
770
 
    # However they should be the same with the exception of the outer
771
 
    # characters, which are stripped off and replaced
772
 
 
773
 
        conf.close()
774
 
    except IOError, (errno, strerror):
775
 
        print "IO error(%s): %s" % (errno, strerror)
776
 
        sys.exit(1)
777
 
 
778
 
    print "Successfully wrote trampoline/conf.h"
779
 
 
780
 
    print
781
 
    print "You may modify the configuration at any time by editing"
782
 
    print conffile
783
 
    print jailconffile
784
 
    print conf_hfile
785
 
    print
786
 
    return 0
787
 
 
788
 
def build(args):
789
 
    # Get "dry" variable from command line
790
 
    (opts, args) = getopt.gnu_getopt(args, "n", ['dry'])
791
 
    opts = dict(opts)
792
 
    dry = '-n' in opts or '--dry' in opts
793
 
 
794
 
    if dry:
795
 
        print "Dry run (no actions will be executed\n"
796
 
 
797
 
    # Compile the trampoline
798
 
    curdir = os.getcwd()
799
 
    os.chdir('trampoline')
800
 
    action_runprog('make', [], dry)
801
 
    os.chdir(curdir)
802
 
 
803
 
    # Create the jail and its subdirectories
804
 
    # Note: Other subdirs will be made by copying files
805
 
    action_mkdir('jail', dry)
806
 
    action_mkdir('jail/home', dry)
807
 
    action_mkdir('jail/tmp', dry)
808
 
 
809
 
    # Chmod the tmp directory to world writable
810
 
    action_chmod_w('jail/tmp', dry)
811
 
 
812
 
    # Copy all console and operating system files into the jail
813
 
    action_copylist(install_list.list_scripts, 'jail/opt/ivle', dry)
814
 
    copy_os_files_jail(dry)
815
 
    # Chmod the python console
816
 
    action_chmod_x('jail/opt/ivle/scripts/python-console', dry)
817
 
    action_chmod_x('jail/opt/ivle/scripts/fileservice', dry)
818
 
    
819
 
    # Also copy the IVLE lib directory into the jail
820
 
    # This is necessary for running certain scripts
821
 
    action_copylist(install_list.list_lib, 'jail/opt/ivle', dry)
822
 
    # IMPORTANT: The file jail/opt/ivle/lib/conf/conf.py contains details
823
 
    # which could compromise security if left in the jail (such as the DB
824
 
    # password).
825
 
    # The "safe" version is in jailconf.py. Delete conf.py and replace it with
826
 
    # jailconf.py.
827
 
    action_copyfile('lib/conf/jailconf.py',
828
 
        'jail/opt/ivle/lib/conf/conf.py', dry)
829
 
 
830
 
    # Compile .py files into .pyc or .pyo files
831
 
    compileall.compile_dir('www', quiet=True)
832
 
    compileall.compile_dir('lib', quiet=True)
833
 
    compileall.compile_dir('scripts', quiet=True)
834
 
    compileall.compile_dir('jail/opt/ivle/lib', quiet=True)
835
 
 
836
 
    # Set up ivle.pth inside the jail
837
 
    # Need to set /opt/ivle/lib to be on the import path
838
 
    ivle_pth = \
839
 
        "jail/usr/lib/python%s/site-packages/ivle.pth" % PYTHON_VERSION
840
 
    f = open(ivle_pth, 'w')
841
 
    f.write('/opt/ivle/lib\n')
842
 
    f.close()
843
 
 
844
 
    return 0
845
 
 
846
 
def copy_os_files_jail(dry):
847
 
    """Copies necessary Operating System files from their usual locations
848
 
    into the jail/ directory of the cwd."""
849
 
    # Currently source paths are configured for Ubuntu.
850
 
    for filename in JAIL_FILES:
851
 
        copy_file_to_jail(filename, dry)
852
 
    for src, dst in JAIL_LINKS.items():
853
 
        action_symlink(src, dst, dry)
854
 
    for src, dst in JAIL_COPYTREES.items():
855
 
        action_copytree(src, dst, dry)
856
 
 
857
 
def copy_file_to_jail(src, dry):
858
 
    """Copies a single file from an absolute location into the same location
859
 
    within the jail. src must begin with a '/'. The jail will be located
860
 
    in a 'jail' subdirectory of the current path."""
861
 
    action_copyfile(src, 'jail' + src, dry)
862
 
 
863
 
def install(args):
864
 
    # Get "dry" and "nojail" variables from command line
865
 
    (opts, args) = getopt.gnu_getopt(args, "n",
866
 
        ['dry', 'nojail', 'nosubjects'])
867
 
    opts = dict(opts)
868
 
    dry = '-n' in opts or '--dry' in opts
869
 
    nojail = '--nojail' in opts
870
 
    nosubjects = '--nosubjects' in opts
871
 
 
872
 
    if dry:
873
 
        print "Dry run (no actions will be executed\n"
874
 
 
875
 
    if not dry and os.geteuid() != 0:
876
 
        print >>sys.stderr, "Must be root to run install"
877
 
        print >>sys.stderr, "(I need to chown some files)."
878
 
        return 1
879
 
 
880
 
    # Create the target (install) directory
881
 
    action_mkdir(ivle_install_dir, dry)
882
 
 
883
 
    # Create bin and copy the compiled files there
884
 
    action_mkdir(os.path.join(ivle_install_dir, 'bin'), dry)
885
 
    tramppath = os.path.join(ivle_install_dir, 'bin/trampoline')
886
 
    action_copyfile('trampoline/trampoline', tramppath, dry)
887
 
    # chown trampoline to root and set setuid bit
888
 
    action_chown_setuid(tramppath, dry)
889
 
 
890
 
    # Create a scripts directory to put the usrmgt-server in.
891
 
    action_mkdir(os.path.join(ivle_install_dir, 'scripts'), dry)
892
 
    usrmgtpath = os.path.join(ivle_install_dir, 'scripts/usrmgt-server')
893
 
    action_copyfile('scripts/usrmgt-server', usrmgtpath, dry)
894
 
    action_chmod_x(usrmgtpath, dry)
895
 
 
896
 
    # Copy the www and lib directories using the list
897
 
    action_copylist(install_list.list_www, ivle_install_dir, dry)
898
 
    action_copylist(install_list.list_lib, ivle_install_dir, dry)
899
 
    
900
 
    # Copy the php directory
901
 
    action_copytree('www/php/phpBB3',os.path.join(ivle_install_dir,'www/php/phpBB3'), 
902
 
    dry)
903
 
 
904
 
    if not nojail:
905
 
        # Copy the local jail directory built by the build action
906
 
        # to the jails template directory (it will be used as a template
907
 
        # for all the students' jails).
908
 
        action_copytree('jail', os.path.join(jail_base, 'template'), dry)
909
 
    if not nosubjects:
910
 
        # Copy the subjects and exercises directories across
911
 
        action_copylist(install_list.list_subjects, subjects_base, dry,
912
 
            srcdir="./subjects")
913
 
        action_copylist(install_list.list_exercises, exercises_base, dry,
914
 
            srcdir="./exercises")
915
 
 
916
 
    # Append IVLE path to ivle.pth in python site packages
917
 
    # (Unless it's already there)
918
 
    ivle_pth = os.path.join(sys.prefix,
919
 
        "lib/python%s/site-packages/ivle.pth" % PYTHON_VERSION)
920
 
    ivle_www = os.path.join(ivle_install_dir, "www")
921
 
    ivle_lib = os.path.join(ivle_install_dir, "lib")
922
 
    write_ivle_pth = True
923
 
    write_ivle_lib_pth = True
924
 
    try:
925
 
        file = open(ivle_pth, 'r')
926
 
        for line in file:
927
 
            if line.strip() == ivle_www:
928
 
                write_ivle_pth = False
929
 
            elif line.strip() == ivle_lib:
930
 
                write_ivle_lib_pth = False
931
 
        file.close()
932
 
    except (IOError, OSError):
933
 
        pass
934
 
    if write_ivle_pth:
935
 
        action_append(ivle_pth, ivle_www)
936
 
    if write_ivle_lib_pth:
937
 
        action_append(ivle_pth, ivle_lib)
938
 
 
939
 
    return 0
940
 
 
941
 
def updatejails(args):
942
 
    # Get "dry" variable from command line
943
 
    (opts, args) = getopt.gnu_getopt(args, "n", ['dry'])
944
 
    opts = dict(opts)
945
 
    dry = '-n' in opts or '--dry' in opts
946
 
 
947
 
    if dry:
948
 
        print "Dry run (no actions will be executed\n"
949
 
 
950
 
    if not dry and os.geteuid() != 0:
951
 
        print >>sys.stderr, "Must be root to run install"
952
 
        print >>sys.stderr, "(I need to chown some files)."
953
 
        return 1
954
 
 
955
 
    # Update the template jail directory in case it hasn't been installed
956
 
    # recently.
957
 
    action_copytree('jail', os.path.join(jail_base, 'template'), dry)
958
 
 
959
 
    # Re-link all the files in all students jails.
960
 
    for dir in os.listdir(jail_base):
961
 
        if dir == 'template': continue
962
 
        # First back up the student's home directory
963
 
        temp_home = os.tmpnam()
964
 
        action_rename(os.path.join(jail_base, dir, 'home'), temp_home, dry)
965
 
        # Delete the student's jail and relink the jail files
966
 
        action_linktree(os.path.join(jail_base, 'template'),
967
 
            os.path.join(jail_base, dir), dry)
968
 
        # Restore the student's home directory
969
 
        action_rename(temp_home, os.path.join(jail_base, dir, 'home'), dry)
970
 
        # Set up the user's home directory just in case they don't have a
971
 
        # directory for this yet
972
 
        action_mkdir(os.path.join(jail_base, dir, 'home', dir), dry)
973
 
 
974
 
    return 0
975
 
 
976
 
# The actions call Python os functions but print actions and handle dryness.
977
 
# May still throw os exceptions if errors occur.
978
 
 
979
 
class RunError:
980
 
    """Represents an error when running a program (nonzero return)."""
981
 
    def __init__(self, prog, retcode):
982
 
        self.prog = prog
983
 
        self.retcode = retcode
984
 
    def __str__(self):
985
 
        return str(self.prog) + " returned " + repr(self.retcode)
986
 
 
987
 
def action_runprog(prog, args, dry):
988
 
    """Runs a unix program. Searches in $PATH. Synchronous (waits for the
989
 
    program to return). Runs in the current environment. First prints the
990
 
    action as a "bash" line.
991
 
 
992
 
    Throws a RunError with a retcode of the return value of the program,
993
 
    if the program did not return 0.
994
 
 
995
 
    prog: String. Name of the program. (No path required, if in $PATH).
996
 
    args: [String]. Arguments to the program.
997
 
    dry: Bool. If True, prints but does not execute.
998
 
    """
999
 
    print prog, string.join(args, ' ')
1000
 
    if dry: return
1001
 
    ret = os.spawnvp(os.P_WAIT, prog, args)
1002
 
    if ret != 0:
1003
 
        raise RunError(prog, ret)
1004
 
 
1005
 
def action_remove(path, dry):
1006
 
    """Calls rmtree, deleting the target file if it exists."""
1007
 
    try:
1008
 
        print "rm -r", path
1009
 
        if not dry:
1010
 
            shutil.rmtree(path, True)
1011
 
    except OSError, (err, msg):
1012
 
        if err != errno.EEXIST:
1013
 
            raise
1014
 
        # Otherwise, didn't exist, so we don't care
1015
 
 
1016
 
def action_rename(src, dst, dry):
1017
 
    """Calls rename. Deletes the target if it already exists."""
1018
 
    action_remove(dst, dry)
1019
 
    print "mv ", src, dst
1020
 
    if dry: return
1021
 
    try:
1022
 
        os.rename(src, dst)
1023
 
    except OSError, (err, msg):
1024
 
        if err != errno.EEXIST:
1025
 
            raise
1026
 
 
1027
 
def action_mkdir(path, dry):
1028
 
    """Calls mkdir. Silently ignored if the directory already exists.
1029
 
    Creates all parent directories as necessary."""
1030
 
    print "mkdir -p", path
1031
 
    if dry: return
1032
 
    try:
1033
 
        os.makedirs(path)
1034
 
    except OSError, (err, msg):
1035
 
        if err != errno.EEXIST:
1036
 
            raise
1037
 
 
1038
 
def action_copytree(src, dst, dry):
1039
 
    """Copies an entire directory tree. Symlinks are seen as normal files and
1040
 
    copies of the entire file (not the link) are made. Creates all parent
1041
 
    directories as necessary.
1042
 
 
1043
 
    See shutil.copytree."""
1044
 
    # Allow copying over itself
1045
 
    if (os.path.normpath(os.path.join(os.getcwd(),src)) ==
1046
 
        os.path.normpath(os.path.join(os.getcwd(),dst))):
1047
 
        return
1048
 
    action_remove(dst, dry)
1049
 
    print "cp -r", src, dst
1050
 
    if dry: return
1051
 
    shutil.copytree(src, dst, True)
1052
 
 
1053
 
def action_linktree(src, dst, dry):
1054
 
    """Hard-links an entire directory tree. Same as copytree but the created
1055
 
    files are hard-links not actual copies. Removes the existing destination.
1056
 
    """
1057
 
    action_remove(dst, dry)
1058
 
    print "<cp with hardlinks> -r", src, dst
1059
 
    if dry: return
1060
 
    common.makeuser.linktree(src, dst)
1061
 
 
1062
 
def action_copylist(srclist, dst, dry, srcdir="."):
1063
 
    """Copies all files in a list to a new location. The files in the list
1064
 
    are read relative to the current directory, and their destinations are the
1065
 
    same paths relative to dst. Creates all parent directories as necessary.
1066
 
    srcdir is "." by default, can be overridden.
1067
 
    """
1068
 
    for srcfile in srclist:
1069
 
        dstfile = os.path.join(dst, srcfile)
1070
 
        srcfile = os.path.join(srcdir, srcfile)
1071
 
        dstdir = os.path.split(dstfile)[0]
1072
 
        if not os.path.isdir(dstdir):
1073
 
            action_mkdir(dstdir, dry)
1074
 
        print "cp -f", srcfile, dstfile
1075
 
        if not dry:
1076
 
            try:
1077
 
                shutil.copyfile(srcfile, dstfile)
1078
 
                shutil.copymode(srcfile, dstfile)
1079
 
            except shutil.Error:
1080
 
                pass
1081
 
 
1082
 
def action_copyfile(src, dst, dry):
1083
 
    """Copies one file to a new location. Creates all parent directories
1084
 
    as necessary.
1085
 
    Warn if file not found.
1086
 
    """
1087
 
    dstdir = os.path.split(dst)[0]
1088
 
    if not os.path.isdir(dstdir):
1089
 
        action_mkdir(dstdir, dry)
1090
 
    print "cp -f", src, dst
1091
 
    if not dry:
1092
 
        try:
1093
 
            shutil.copyfile(src, dst)
1094
 
            shutil.copymode(src, dst)
1095
 
        except (shutil.Error, IOError), e:
1096
 
            print "Warning: " + str(e)
1097
 
 
1098
 
def action_symlink(src, dst, dry):
1099
 
    """Creates a symlink in a given location. Creates all parent directories
1100
 
    as necessary.
1101
 
    """
1102
 
    dstdir = os.path.split(dst)[0]
1103
 
    if not os.path.isdir(dstdir):
1104
 
        action_mkdir(dstdir, dry)
1105
 
    # Delete existing file
1106
 
    if os.path.exists(dst):
1107
 
        os.remove(dst)
1108
 
    print "ln -fs", src, dst
1109
 
    if not dry:
1110
 
        os.symlink(src, dst)
1111
 
 
1112
 
def action_append(ivle_pth, ivle_www):
1113
 
    file = open(ivle_pth, 'a+')
1114
 
    file.write(ivle_www + '\n')
1115
 
    file.close()
1116
 
 
1117
 
def action_chown_setuid(file, dry):
1118
 
    """Chowns a file to root, and sets the setuid bit on the file.
1119
 
    Calling this function requires the euid to be root.
1120
 
    The actual mode of path is set to: rws--s--s
1121
 
    """
1122
 
    print "chown root:root", file
1123
 
    if not dry:
1124
 
        os.chown(file, 0, 0)
1125
 
    print "chmod a+xs", file
1126
 
    print "chmod u+rw", file
1127
 
    if not dry:
1128
 
        os.chmod(file, stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH
1129
 
            | stat.S_ISUID | stat.S_IRUSR | stat.S_IWUSR)
1130
 
 
1131
 
def action_chmod_x(file, dry):
1132
 
    """Chmod 755 a file (sets permissions to rwxr-xr-x)."""
1133
 
    print "chmod 755", file
1134
 
    if not dry:
1135
 
        os.chmod(file, stat.S_IXUSR | stat.S_IRUSR | stat.S_IWUSR
1136
 
            | stat.S_IXGRP | stat.S_IRGRP | stat.S_IXOTH | stat.S_IROTH)
1137
 
 
1138
 
 
1139
 
def action_chmod_w(file, dry):
1140
 
    """Chmod 777 a file (sets permissions to rwxrwxrwx)."""
1141
 
    print "chmod 777", file
1142
 
    if not dry:
1143
 
        os.chmod(file, stat.S_IXUSR | stat.S_IRUSR | stat.S_IWUSR
1144
 
            | stat.S_IXGRP | stat.S_IWGRP | stat.S_IRGRP | stat.S_IXOTH
1145
 
            | stat.S_IWOTH | stat.S_IROTH)
1146
 
 
1147
 
def query_user(default, prompt):
1148
 
    """Prompts the user for a string, which is read from a line of stdin.
1149
 
    Exits silently if EOF is encountered. Returns the string, with spaces
1150
 
    removed from the beginning and end.
1151
 
 
1152
 
    Returns default if a 0-length line (after spaces removed) was read.
1153
 
    """
1154
 
    sys.stdout.write('%s\n    (default: "%s")\n>' % (prompt, default))
1155
 
    try:
1156
 
        val = sys.stdin.readline()
1157
 
    except KeyboardInterrupt:
1158
 
        # Ctrl+C
1159
 
        sys.stdout.write("\n")
1160
 
        sys.exit(1)
1161
 
    sys.stdout.write("\n")
1162
 
    # If EOF, exit
1163
 
    if val == '': sys.exit(1)
1164
 
    # If empty line, return default
1165
 
    val = val.strip()
1166
 
    if val == '': return default
1167
 
    return val
1168
 
 
1169
 
def filter_mutate(function, list):
1170
 
    """Like built-in filter, but mutates the given list instead of returning a
1171
 
    new one. Returns None."""
1172
 
    i = len(list)-1
1173
 
    while i >= 0:
1174
 
        # Delete elements which do not match
1175
 
        if not function(list[i]):
1176
 
            del list[i]
1177
 
        i -= 1
 
85
        sys.exit(1)
 
86
    return oper_func
1178
87
 
1179
88
if __name__ == "__main__":
1180
89
    sys.exit(main())
 
90