packmail.py 2.58 KB
Newer Older
1 2 3 4
# Module 'packmail' -- create a self-unpacking shell archive.

# This module works on UNIX and on the Mac; the archives can unpack
# themselves only on UNIX.
Guido van Rossum's avatar
Guido van Rossum committed
5

Guido van Rossum's avatar
Guido van Rossum committed
6
import os
Guido van Rossum's avatar
Guido van Rossum committed
7 8
from stat import ST_MTIME

9 10 11 12 13 14 15 16 17 18
# Print help
def help():
	print 'All fns have a file open for writing as first parameter'
	print 'pack(f, fullname, name): pack fullname as name'
	print 'packsome(f, directory, namelist): selected files from directory'
	print 'packall(f, directory): pack all files from directory'
	print 'packnotolder(f, directory, name): pack all files from directory'
	print '                        that are not older than a file there'
	print 'packtree(f, directory): pack entire directory tree'

Guido van Rossum's avatar
Guido van Rossum committed
19 20 21
# Pack one file
def pack(outfp, file, name):
	fp = open(file, 'r')
22
	outfp.write('echo ' + name + '\n')
23
	outfp.write('sed "s/^X//" >"' + name + '" <<"!"\n')
Guido van Rossum's avatar
Guido van Rossum committed
24 25 26
	while 1:
		line = fp.readline()
		if not line: break
27
		if line[-1:] != '\n':
Guido van Rossum's avatar
Guido van Rossum committed
28 29 30
			line = line + '\n'
		outfp.write('X' + line)
	outfp.write('!\n')
31
	fp.close()
Guido van Rossum's avatar
Guido van Rossum committed
32 33 34 35 36

# Pack some files from a directory
def packsome(outfp, dirname, names):
	for name in names:
		print name
Guido van Rossum's avatar
Guido van Rossum committed
37
		file = os.path.join(dirname, name)
Guido van Rossum's avatar
Guido van Rossum committed
38 39 40 41
		pack(outfp, file, name)

# Pack all files from a directory
def packall(outfp, dirname):
Guido van Rossum's avatar
Guido van Rossum committed
42
	names = os.listdir(dirname)
43 44 45 46 47 48 49 50
	try:
	    names.remove('.')
	except:
	    pass
	try:
	    names.remove('..')
	except:
	    pass
Guido van Rossum's avatar
Guido van Rossum committed
51 52 53 54 55
	names.sort()
	packsome(outfp, dirname, names)

# Pack all files from a directory that are not older than a give one
def packnotolder(outfp, dirname, oldest):
Guido van Rossum's avatar
Guido van Rossum committed
56
	names = os.listdir(dirname)
57 58 59 60 61 62 63 64
	try:
	    names.remove('.')
	except:
	    pass
	try:
	    names.remove('..')
	except:
	    pass
Guido van Rossum's avatar
Guido van Rossum committed
65 66
	oldest = os.path.join(dirname, oldest)
	st = os.stat(oldest)
Guido van Rossum's avatar
Guido van Rossum committed
67 68 69 70
	mtime = st[ST_MTIME]
	todo = []
	for name in names:
		print name, '...',
Guido van Rossum's avatar
Guido van Rossum committed
71
		st = os.stat(os.path.join(dirname, name))
Guido van Rossum's avatar
Guido van Rossum committed
72 73 74 75 76 77 78
		if st[ST_MTIME] >= mtime:
			print 'Yes.'
			todo.append(name)
		else:
			print 'No.'
	todo.sort()
	packsome(outfp, dirname, todo)
79 80 81 82 83

# Pack a whole tree (no exceptions)
def packtree(outfp, dirname):
	print 'packtree', dirname
	outfp.write('mkdir ' + unixfix(dirname) + '\n')
Guido van Rossum's avatar
Guido van Rossum committed
84
	names = os.listdir(dirname)
85 86 87 88 89 90 91 92
	try:
	    names.remove('.')
	except:
	    pass
	try:
	    names.remove('..')
	except:
	    pass
93 94
	subdirs = []
	for name in names:
Guido van Rossum's avatar
Guido van Rossum committed
95 96
		fullname = os.path.join(dirname, name)
		if os.path.isdir(fullname):
97 98 99 100 101 102 103 104
			subdirs.append(fullname)
		else:
			print 'pack', fullname
			pack(outfp, fullname, unixfix(fullname))
	for subdirname in subdirs:
		packtree(outfp, subdirname)

def unixfix(name):
Eric S. Raymond's avatar
Eric S. Raymond committed
105
	comps = name.split(os.sep)
106 107 108 109 110 111
	res = ''
	for comp in comps:
		if comp:
			if res: res = res + '/'
			res = res + comp
	return res