23
23
# This is a command-line application, for use by the administrator.
24
# This program is a frontend for the modules in the setup packages that
25
# build and install IVLE in separate steps.
24
# This program configures, builds and installs IVLE in three separate steps.
26
25
# It is called with at least one argument, which specifies which operation to
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.
34
# setup.py conf [args]
35
# Configures IVLE with machine-specific details, most notably, various paths.
36
# Either prompts the administrator for these details or accepts them as
38
# Creates www/conf/conf.py and trampoline/conf.h.
41
# Compiles all files and sets up a jail template in the source directory.
43
# Compiles (GCC) trampoline/trampoline.c to trampoline/trampoline.
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.
51
# setup.py install [--nojail] [--dry|n]
53
# Create target install directory ($target).
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).
71
# Import modules from the website is tricky since they're in the www
73
sys.path.append(os.path.join(os.getcwd(), 'www'))
75
import common.makeuser
77
# Operating system files to copy over into the jail.
78
# These will be copied from the given place on the OS file system into the
79
# same place within the jail.
82
'/lib/tls/i686/cmov/libc.so.6',
83
'/lib/tls/i686/cmov/libdl.so.2',
84
'/lib/tls/i686/cmov/libm.so.6',
85
'/lib/tls/i686/cmov/libpthread.so.0',
86
'/lib/tls/i686/cmov/libutil.so.1',
89
# These 2 files do not exist in Ubuntu
90
#'/etc/ld.so.preload',
91
#'/etc/ld.so.nohwcap',
98
# Needed by matplotlib
99
'/usr/lib/i686/cmov/libssl.so.0.9.8',
100
'/usr/lib/i686/cmov/libcrypto.so.0.9.8',
101
'/lib/tls/i686/cmov/libnsl.so.1',
102
'/usr/lib/libz.so.1',
103
'/usr/lib/atlas/liblapack.so.3',
104
'/usr/lib/atlas/libblas.so.3',
105
'/usr/lib/libg2c.so.0',
106
'/usr/lib/libstdc++.so.6',
107
'/usr/lib/libfreetype.so.6',
108
'/usr/lib/libpng12.so.0',
109
'/usr/lib/libBLT.2.4.so.8.4',
110
'/usr/lib/libtk8.4.so.0',
111
'/usr/lib/libtcl8.4.so.0',
112
'/usr/lib/tcl8.4/init.tcl',
113
'/usr/lib/libX11.so.6',
114
'/usr/lib/libXau.so.6',
115
'/usr/lib/libXdmcp.so.6',
116
'/lib/libgcc_s.so.1',
119
# Symlinks to make within the jail. Src mapped to dst.
121
'python2.5': 'jail/usr/bin/python',
123
# Trees to copy. Src mapped to dst (these will be passed to action_copytree).
125
'/usr/lib/python2.5': 'jail/usr/lib/python2.5',
126
'/usr/share/matplotlib': 'jail/usr/share/matplotlib',
127
'/etc/ld.so.conf.d': 'jail/etc/ld.so.conf.d',
130
# Try importing existing conf, but if we can't just set up defaults
131
# The reason for this is that these settings are used by other phases
132
# of setup besides conf, so we need to know them.
133
# Also this allows you to hit Return to accept the existing value.
135
confmodule = __import__("www/conf/conf")
137
root_dir = confmodule.root_dir
141
ivle_install_dir = confmodule.ivle_install_dir
143
ivle_install_dir = "/opt/ivle"
145
public_host = confmodule.public_host
147
public_host = "public.localhost"
149
jail_base = confmodule.jail_base
151
jail_base = "/home/informatics/jails"
153
subjects_base = confmodule.subjects_base
155
subjects_base = "/home/informatics/subjects"
157
# Just set reasonable defaults
159
ivle_install_dir = "/opt/ivle"
160
public_host = "public.localhost"
161
jail_base = "/home/informatics/jails"
162
subjects_base = "/home/informatics/subjects"
166
# Try importing install_list, but don't fail if we can't, because listmake can
167
# function without it.
173
# Mime types which will automatically be placed in the list by listmake.
174
# Note that listmake is not intended to be run by the final user (the system
175
# administrator who installs this), so the developers can customize the list
176
# as necessary, and include it in the distribution.
177
listmake_mimetypes = ['text/x-python', 'text/html',
178
'application/x-javascript', 'application/javascript',
179
'text/css', 'image/png', 'application/xml']
181
# Main function skeleton from Guido van Rossum
182
# http://www.artima.com/weblogs/viewpost.jsp?thread=4829
33
184
def main(argv=None):
56
oper_func = call_operator(operation)
57
return oper_func(argv[2:])
61
print """Usage: python setup.py operation [options]
67
For help and options for a specific operation use 'help [operation]'."""
70
oper_func = call_operator(operator)
71
oper_func(['operator','--help'])
73
def call_operator(operation):
207
# Disallow run as root unless installing
208
if (operation != 'install' and operation != 'updatejails'
209
and os.geteuid() == 0):
210
print >>sys.stderr, "I do not want to run this stage as root."
211
print >>sys.stderr, "Please run as a normal user."
74
213
# Call the requested operation's function
78
'build' : setup.build.build,
79
'install' : setup.install.install,
219
'listmake' : listmake,
221
'updatejails' : updatejails,
82
224
print >>sys.stderr, (
83
225
"""Invalid operation '%s'. Try python setup.py help."""
228
return oper_func(argv[2:])
230
# Operation functions
234
print """Usage: python setup.py operation [args]
235
Operation (and args) can be:
237
listmake (developer use only)
240
install [--nojail] [--nosubjects] [-n|--dry]
244
print """Usage: python setup.py help [operation]"""
249
if operation == 'help':
250
print """python setup.py help [operation]
251
Prints the usage message or detailed help on an operation, then exits."""
252
elif operation == 'listmake':
253
print """python setup.py listmake
254
(For developer use only)
255
Recurses through the source tree and builds a list of all files which should
256
be copied upon installation. This should be run by the developer before
257
cutting a distribution, and the listfile it generates should be included in
258
the distribution, avoiding the administrator having to run it."""
259
elif operation == 'conf':
260
print """python setup.py conf [args]
261
Configures IVLE with machine-specific details, most notably, various paths.
262
Either prompts the administrator for these details or accepts them as
263
command-line args. Will be interactive only if there are no arguments given.
264
Takes defaults from existing conf file if it exists.
266
To run IVLE out of the source directory (allowing development without having
267
to rebuild/install), just provide ivle_install_dir as the IVLE trunk
268
directory, and run build/install one time.
270
Creates www/conf/conf.py and trampoline/conf.h.
279
As explained in the interactive prompt or conf.py.
281
elif operation == 'build':
282
print """python -O setup.py build [--dry|-n]
283
Compiles all files and sets up a jail template in the source directory.
284
-O is recommended to cause compilation to be optimised.
286
Compiles (GCC) trampoline/trampoline.c to trampoline/trampoline.
288
Creates standard subdirs inside the jail, eg bin, opt, home, tmp.
289
Copies console/ to a location within the jail.
290
Copies OS programs and files to corresponding locations within the jail
291
(eg. python and Python libs, ld.so, etc).
292
Generates .pyc or .pyo files for all the IVLE .py files.
294
--dry | -n Print out the actions but don't do anything."""
295
elif operation == 'install':
296
print """sudo python setup.py install [--nojail] [--nosubjects][--dry|-n]
298
Create target install directory ($target).
300
Copy trampoline/trampoline to $target/bin.
301
chown and chmod the installed trampoline.
302
Copy www/ to $target.
303
Copy jail/ to jails template directory (unless --nojail specified).
304
Copy subjects/ to subjects directory (unless --nosubjects specified).
306
--nojail Do not copy the jail.
307
--nosubjects Do not copy the subjects.
308
--dry | -n Print out the actions but don't do anything."""
309
elif operation == 'updatejails':
310
print """sudo python setup.py updatejails [--dry|-n]
312
Copy jail/ to each subdirectory in jails directory.
314
--dry | -n Print out the actions but don't do anything."""
316
print >>sys.stderr, (
317
"""Invalid operation '%s'. Try python setup.py help."""
322
# We build two separate lists, by walking www and console
323
list_www = build_list_py_files('www')
324
list_console = build_list_py_files('console')
325
list_subjects = build_list_py_files('subjects', no_top_level=True)
326
# Make sure that the files generated by conf are in the list
327
# (since listmake is typically run before conf)
328
if "www/conf/conf.py" not in list_www:
329
list_www.append("www/conf/conf.py")
330
# Make sure that console/python-console is in the list
331
if "console/python-console" not in list_console:
332
list_console.append("console/python-console")
333
# Write these out to a file
335
# the files that will be created/overwritten
336
listfile = os.path.join(cwd, "install_list.py")
339
file = open(listfile, "w")
341
file.write("""# IVLE Configuration File
343
# Provides lists of all files to be installed by `setup.py install' from
344
# certain directories.
345
# Note that any files with the given filename plus 'c' or 'o' (that is,
346
# compiled .pyc or .pyo files) will be copied as well.
348
# List of all installable files in www directory.
350
writelist_pretty(file, list_www)
352
# List of all installable files in console directory.
354
writelist_pretty(file, list_console)
356
# List of all installable files in subjects directory.
357
# This is to install sample subjects and material.
359
writelist_pretty(file, list_subjects)
362
except IOError, (errno, strerror):
363
print "IO error(%s): %s" % (errno, strerror)
366
print "Successfully wrote install_list.py"
369
print ("You may modify the set of installable files before cutting the "
376
def build_list_py_files(dir, no_top_level=False):
377
"""Builds a list of all py files found in a directory and its
378
subdirectories. Returns this as a list of strings.
379
no_top_level=True means the file paths will not include the top-level
383
for (dirpath, dirnames, filenames) in os.walk(dir):
384
# Exclude directories beginning with a '.' (such as '.svn')
385
filter_mutate(lambda x: x[0] != '.', dirnames)
386
# All *.py files are added to the list
387
pylist += [os.path.join(dirpath, item) for item in filenames
388
if mimetypes.guess_type(item)[0] in listmake_mimetypes]
390
for i in range(0, len(pylist)):
391
_, pylist[i] = pylist[i].split(os.sep, 1)
394
def writelist_pretty(file, list):
395
"""Writes a list one element per line, to a file."""
401
file.write(' %s,\n' % repr(elem))
405
global root_dir, ivle_install_dir, jail_base, subjects_base
406
global public_host, allowed_uids
407
# Set up some variables
410
# the files that will be created/overwritten
411
conffile = os.path.join(cwd, "www/conf/conf.py")
412
conf_hfile = os.path.join(cwd, "trampoline/conf.h")
414
# Get command-line arguments to avoid asking questions.
416
(opts, args) = getopt.gnu_getopt(args, "", ['root_dir=',
417
'ivle_install_dir=', 'jail_base=', 'allowed_uids='])
420
print >>sys.stderr, "Invalid arguments:", string.join(args, ' ')
424
# Interactive mode. Prompt the user for all the values.
426
print """This tool will create the following files:
429
prompting you for details about your configuration. The file will be
430
overwritten if it already exists. It will *not* install or deploy IVLE.
432
Please hit Ctrl+C now if you do not wish to do this.
433
""" % (conffile, conf_hfile)
435
# Get information from the administrator
436
# If EOF is encountered at any time during the questioning, just exit
439
root_dir = query_user(root_dir,
440
"""Root directory where IVLE is located (in URL space):""")
441
ivle_install_dir = query_user(ivle_install_dir,
442
'Root directory where IVLE will be installed (on the local file '
444
jail_base = query_user(jail_base,
445
"""Root directory where the jails (containing user files) are stored
446
(on the local file system):""")
447
subjects_base = query_user(subjects_base,
448
"""Root directory where the subject directories (containing worksheets
449
and other per-subject files) are stored (on the local file system):""")
450
public_host = query_user(public_host,
451
"""Hostname which will cause the server to go into "public mode",
452
providing login-free access to student's published work:""")
453
allowed_uids = query_user(allowed_uids,
454
"""UID of the web server process which will run IVLE.
455
Only this user may execute the trampoline. May specify multiple users as
456
a comma-separated list.
461
# Non-interactive mode. Parse the options.
462
if '--root_dir' in opts:
463
root_dir = opts['--root_dir']
464
if '--ivle_install_dir' in opts:
465
ivle_install_dir = opts['--ivle_install_dir']
466
if '--jail_base' in opts:
467
jail_base = opts['--jail_base']
468
if '--subjects_base' in opts:
469
jail_base = opts['--subjects_base']
470
if '--public_host' in opts:
471
public_host = opts['--public_host']
472
if '--allowed_uids' in opts:
473
allowed_uids = opts['--allowed_uids']
475
# Error handling on input values
477
allowed_uids = map(int, allowed_uids.split(','))
479
print >>sys.stderr, (
480
"Invalid UID list (%s).\n"
481
"Must be a comma-separated list of integers." % allowed_uids)
484
# Write www/conf/conf.py
487
conf = open(conffile, "w")
489
conf.write("""# IVLE Configuration File
491
# Miscellaneous application settings
494
# In URL space, where in the site is IVLE located. (All URLs will be prefixed
496
# eg. "/" or "/ivle".
499
# In the local file system, where IVLE is actually installed.
500
# This directory should contain the "www" and "bin" directories.
501
ivle_install_dir = "%s"
503
# The server goes into "public mode" if the browser sends a request with this
504
# host. This is for security reasons - we only serve public student files on a
505
# separate domain to the main IVLE site.
506
# Public mode does not use cookies, and serves only public content.
507
# Private mode (normal mode) requires login, and only serves files relevant to
508
# the logged-in user.
511
# In the local file system, where are the student/user file spaces located.
512
# The user jails are expected to be located immediately in subdirectories of
516
# In the local file system, where are the per-subject file spaces located.
517
# The individual subject directories are expected to be located immediately
518
# in subdirectories of this location.
520
""" % (root_dir, ivle_install_dir, public_host, jail_base, subjects_base))
523
except IOError, (errno, strerror):
524
print "IO error(%s): %s" % (errno, strerror)
527
print "Successfully wrote www/conf/conf.py"
529
# Write trampoline/conf.h
532
conf = open(conf_hfile, "w")
534
conf.write("""/* IVLE Configuration File
536
* Administrator settings required by trampoline.
537
* Note: trampoline will have to be rebuilt in order for changes to this file
541
/* In the local file system, where are the jails located.
542
* The trampoline does not allow the creation of a jail anywhere besides
543
* jail_base or a subdirectory of jail_base.
545
static const char* jail_base = "%s";
547
/* Which user IDs are allowed to run the trampoline.
548
* This list should be limited to the web server user.
549
* (Note that root is an implicit member of this list).
551
static const int allowed_uids[] = { %s };
552
""" % (jail_base, repr(allowed_uids)[1:-1]))
555
except IOError, (errno, strerror):
556
print "IO error(%s): %s" % (errno, strerror)
559
print "Successfully wrote trampoline/conf.h"
562
print "You may modify the configuration at any time by editing"
569
# Get "dry" variable from command line
570
(opts, args) = getopt.gnu_getopt(args, "n", ['dry'])
572
dry = '-n' in opts or '--dry' in opts
575
print "Dry run (no actions will be executed\n"
577
# Compile the trampoline
578
action_runprog('gcc', ['-Wall', '-o', 'trampoline/trampoline',
579
'trampoline/trampoline.c'], dry)
581
# Create the jail and its subdirectories
582
# Note: Other subdirs will be made by copying files
583
action_mkdir('jail', dry)
584
action_mkdir('jail/home', dry)
585
action_mkdir('jail/tmp', dry)
587
# Copy all console and operating system files into the jail
588
action_copylist(install_list.list_console, 'jail/opt/ivle', dry)
589
copy_os_files_jail(dry)
590
# Chmod the python console
591
action_chmod_x('jail/opt/ivle/console/python-console', dry)
594
# Compile .py files into .pyc or .pyo files
595
compileall.compile_dir('www', quiet=True)
596
compileall.compile_dir('console', quiet=True)
600
def copy_os_files_jail(dry):
601
"""Copies necessary Operating System files from their usual locations
602
into the jail/ directory of the cwd."""
603
# Currently source paths are configured for Ubuntu.
604
for filename in JAIL_FILES:
605
copy_file_to_jail(filename, dry)
606
for src, dst in JAIL_LINKS.items():
607
action_symlink(src, dst, dry)
608
for src, dst in JAIL_COPYTREES.items():
609
action_copytree(src, dst, dry)
611
def copy_file_to_jail(src, dry):
612
"""Copies a single file from an absolute location into the same location
613
within the jail. src must begin with a '/'. The jail will be located
614
in a 'jail' subdirectory of the current path."""
615
action_copyfile(src, 'jail' + src, dry)
618
# Get "dry" and "nojail" variables from command line
619
(opts, args) = getopt.gnu_getopt(args, "n",
620
['dry', 'nojail', 'nosubjects'])
622
dry = '-n' in opts or '--dry' in opts
623
nojail = '--nojail' in opts
624
nosubjects = '--nosubjects' in opts
627
print "Dry run (no actions will be executed\n"
629
if not dry and os.geteuid() != 0:
630
print >>sys.stderr, "Must be root to run install"
631
print >>sys.stderr, "(I need to chown some files)."
634
# Create the target (install) directory
635
action_mkdir(ivle_install_dir, dry)
637
# Create bin and copy the compiled files there
638
action_mkdir(os.path.join(ivle_install_dir, 'bin'), dry)
639
tramppath = os.path.join(ivle_install_dir, 'bin/trampoline')
640
action_copyfile('trampoline/trampoline', tramppath, dry)
641
# chown trampoline to root and set setuid bit
642
action_chown_setuid(tramppath, dry)
644
# Copy the www directory using the list
645
action_copylist(install_list.list_www, ivle_install_dir, dry)
648
# Copy the local jail directory built by the build action
649
# to the jails template directory (it will be used as a template
650
# for all the students' jails).
651
action_copytree('jail', os.path.join(jail_base, 'template'), dry)
653
# Copy the subjects directory across
654
action_copylist(install_list.list_subjects, subjects_base, dry,
657
# Append IVLE path to ivle.pth in python site packages
658
# (Unless it's already there)
659
ivle_pth = os.path.join(sys.prefix,
660
"lib/python2.5/site-packages/ivle.pth")
661
ivle_www = os.path.join(ivle_install_dir, "www")
662
write_ivle_pth = True
664
file = open(ivle_pth, 'r')
666
if line.strip() == ivle_www:
667
write_ivle_pth = False
669
except (IOError, OSError):
672
action_append(ivle_pth, ivle_www)
676
def updatejails(args):
677
# Get "dry" variable from command line
678
(opts, args) = getopt.gnu_getopt(args, "n", ['dry'])
680
dry = '-n' in opts or '--dry' in opts
683
print "Dry run (no actions will be executed\n"
685
if not dry and os.geteuid() != 0:
686
print >>sys.stderr, "Must be root to run install"
687
print >>sys.stderr, "(I need to chown some files)."
690
# Update the template jail directory in case it hasn't been installed
692
action_copytree('jail', os.path.join(jail_base, 'template'), dry)
694
# Re-link all the files in all students jails.
695
for dir in os.listdir(jail_base):
696
if dir == 'template': continue
697
# First back up the student's home directory
698
temp_home = os.tmpnam()
699
action_rename(os.path.join(jail_base, dir, 'home'), temp_home, dry)
700
# Delete the student's jail and relink the jail files
701
action_linktree(os.path.join(jail_base, 'template'),
702
os.path.join(jail_base, dir), dry)
703
# Restore the student's home directory
704
action_rename(temp_home, os.path.join(jail_base, dir, 'home'), dry)
705
# Set up the user's home directory just in case they don't have a
706
# directory for this yet
707
action_mkdir(os.path.join(jail_base, dir, 'home', dir), dry)
711
# The actions call Python os functions but print actions and handle dryness.
712
# May still throw os exceptions if errors occur.
715
"""Represents an error when running a program (nonzero return)."""
716
def __init__(self, prog, retcode):
718
self.retcode = retcode
720
return str(self.prog) + " returned " + repr(self.retcode)
722
def action_runprog(prog, args, dry):
723
"""Runs a unix program. Searches in $PATH. Synchronous (waits for the
724
program to return). Runs in the current environment. First prints the
725
action as a "bash" line.
727
Throws a RunError with a retcode of the return value of the program,
728
if the program did not return 0.
730
prog: String. Name of the program. (No path required, if in $PATH).
731
args: [String]. Arguments to the program.
732
dry: Bool. If True, prints but does not execute.
734
print prog, string.join(args, ' ')
736
ret = os.spawnvp(os.P_WAIT, prog, args)
738
raise RunError(prog, ret)
740
def action_rename(src, dst, dry):
741
"""Calls rename. Deletes the target if it already exists."""
742
if os.access(dst, os.F_OK):
745
shutil.rmtree(dst, True)
746
print "mv ", src, dst
750
except OSError, (err, msg):
751
if err != errno.EEXIST:
754
def action_mkdir(path, dry):
755
"""Calls mkdir. Silently ignored if the directory already exists.
756
Creates all parent directories as necessary."""
757
print "mkdir -p", path
761
except OSError, (err, msg):
762
if err != errno.EEXIST:
765
def action_copytree(src, dst, dry):
766
"""Copies an entire directory tree. Symlinks are seen as normal files and
767
copies of the entire file (not the link) are made. Creates all parent
768
directories as necessary.
770
See shutil.copytree."""
771
if os.access(dst, os.F_OK):
774
shutil.rmtree(dst, True)
775
print "cp -r", src, dst
777
shutil.copytree(src, dst, True)
779
def action_linktree(src, dst, dry):
780
"""Hard-links an entire directory tree. Same as copytree but the created
781
files are hard-links not actual copies. Removes the existing destination.
783
if os.access(dst, os.F_OK):
786
shutil.rmtree(dst, True)
787
print "<cp with hardlinks> -r", src, dst
789
common.makeuser.linktree(src, dst)
791
def action_copylist(srclist, dst, dry, srcdir="."):
792
"""Copies all files in a list to a new location. The files in the list
793
are read relative to the current directory, and their destinations are the
794
same paths relative to dst. Creates all parent directories as necessary.
795
srcdir is "." by default, can be overridden.
797
for srcfile in srclist:
798
dstfile = os.path.join(dst, srcfile)
799
srcfile = os.path.join(srcdir, srcfile)
800
dstdir = os.path.split(dstfile)[0]
801
if not os.path.isdir(dstdir):
802
action_mkdir(dstdir, dry)
803
print "cp -f", srcfile, dstfile
806
shutil.copyfile(srcfile, dstfile)
807
shutil.copymode(srcfile, dstfile)
811
def action_copyfile(src, dst, dry):
812
"""Copies one file to a new location. Creates all parent directories
815
dstdir = os.path.split(dst)[0]
816
if not os.path.isdir(dstdir):
817
action_mkdir(dstdir, dry)
818
print "cp -f", src, dst
821
shutil.copyfile(src, dst)
822
shutil.copymode(src, dst)
826
def action_symlink(src, dst, dry):
827
"""Creates a symlink in a given location. Creates all parent directories
830
dstdir = os.path.split(dst)[0]
831
if not os.path.isdir(dstdir):
832
action_mkdir(dstdir, dry)
833
# Delete existing file
834
if os.path.exists(dst):
836
print "ln -fs", src, dst
840
def action_append(ivle_pth, ivle_www):
841
file = open(ivle_pth, 'a+')
842
file.write(ivle_www + '\n')
845
def action_chown_setuid(file, dry):
846
"""Chowns a file to root, and sets the setuid bit on the file.
847
Calling this function requires the euid to be root.
848
The actual mode of path is set to: rws--s--s
850
print "chown root:root", file
853
print "chmod a+xs", file
854
print "chmod u+rw", file
856
os.chmod(file, stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH
857
| stat.S_ISUID | stat.S_IRUSR | stat.S_IWUSR)
859
def action_chmod_x(file, dry):
860
"""Chmod +xs a file (sets execute permission)."""
861
print "chmod u+rwx", file
863
os.chmod(file, stat.S_IXUSR | stat.S_IRUSR | stat.S_IWUSR)
865
def query_user(default, prompt):
866
"""Prompts the user for a string, which is read from a line of stdin.
867
Exits silently if EOF is encountered. Returns the string, with spaces
868
removed from the beginning and end.
870
Returns default if a 0-length line (after spaces removed) was read.
872
sys.stdout.write('%s\n (default: "%s")\n>' % (prompt, default))
874
val = sys.stdin.readline()
875
except KeyboardInterrupt:
877
sys.stdout.write("\n")
879
sys.stdout.write("\n")
881
if val == '': sys.exit(1)
882
# If empty line, return default
884
if val == '': return default
887
def filter_mutate(function, list):
888
"""Like built-in filter, but mutates the given list instead of returning a
889
new one. Returns None."""
892
# Delete elements which do not match
893
if not function(list[i]):
88
897
if __name__ == "__main__":
added
removed
setup.py
