Mercurial > p > roundup > code

annotate roundup/install_util.py @ 2119:cc4667ef3f12

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Added the ability to toggle where error messages go. They either go to the user (default, for backwards compatibility), the dispatcher, or both. These are able to be toggled via settings in config.py. Please refer to upgrading.txt for more details. (And Richard, let me know if I've done anything wrong with this checkin. :))
author Eddie Parker <eparker@users.sourceforge.net>
date Thu, 25 Mar 2004 19:27:15 +0000
parents fc52d57c6c3e
children 4ee255777c25 897ac2dab754
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
1 #
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
2 # Copyright (c) 2001 Bizar Software Pty Ltd (http://www.bizarsoftware.com.au/)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
3 # This module is free software, and you may redistribute it and/or modify
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
4 # under the same terms as Python, so long as this copyright message and
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
5 # disclaimer are retained in their original form.
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
6 #
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
7 # IN NO EVENT SHALL BIZAR SOFTWARE PTY LTD BE LIABLE TO ANY PARTY FOR
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
8 # DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
9 # OUT OF THE USE OF THIS CODE, EVEN IF THE AUTHOR HAS BEEN ADVISED OF THE
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
10 # POSSIBILITY OF SUCH DAMAGE.
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
11 #
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
12 # BIZAR SOFTWARE PTY LTD SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING,
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
13 # BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
14 # FOR A PARTICULAR PURPOSE. THE CODE PROVIDED HEREUNDER IS ON AN "AS IS"
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
15 # BASIS, AND THERE IS NO OBLIGATION WHATSOEVER TO PROVIDE MAINTENANCE,
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
16 # SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
17 #
2005
fc52d57c6c3e documentation cleanup
Richard Jones <richard@users.sourceforge.net>
parents: 1910
diff changeset
18 # $Id: install_util.py,v 1.10 2004-02-11 23:55:08 richard Exp $
406
bdc2ea127ae9 Added module docstrings to all modules.
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 388
diff changeset
19
2005
fc52d57c6c3e documentation cleanup
Richard Jones <richard@users.sourceforge.net>
parents: 1910
diff changeset
20 """Support module to generate and check fingerprints of installed files.
406
bdc2ea127ae9 Added module docstrings to all modules.
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 388
diff changeset
21 """
2005
fc52d57c6c3e documentation cleanup
Richard Jones <richard@users.sourceforge.net>
parents: 1910
diff changeset
22 __docformat__ = 'restructuredtext'
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
23
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
24 import os, sha, shutil
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
25
409
fc461bf67c99 Added .newitem extension
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 406
diff changeset
26 # ".filter", ".index", ".item", ".newitem" are roundup-specific
fc461bf67c99 Added .newitem extension
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 406
diff changeset
27 sgml_file_types = [".xml", ".ent", ".html", ".filter", ".index", ".item", ".newitem"]
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
28 hash_file_types = [".py", ".sh", ".conf", ".cgi", '']
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
29 slast_file_types = [".css"]
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
30
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
31 digested_file_types = sgml_file_types + hash_file_types + slast_file_types
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
32
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
33 def extractFingerprint(lines):
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
34 # get fingerprint from last line
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
35 if lines[-1].startswith("#SHA: "):
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
36 # handle .py/.sh comment
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
37 return lines[-1][6:].strip()
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
38 elif lines[-1].startswith("<!-- SHA: "):
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
39 # handle xml/html files
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
40 fingerprint = lines[-1][10:]
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
41 fingerprint = fingerprint.replace('-->', '')
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
42 return fingerprint.strip()
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
43 elif lines[-1].startswith("/* SHA: "):
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
44 # handle css files
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
45 fingerprint = lines[-1][8:]
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
46 fingerprint = fingerprint.replace('*/', '')
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
47 return fingerprint.strip()
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
48 return None
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
49
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
50 def checkDigest(filename):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
51 """Read file, check for valid fingerprint, return TRUE if ok"""
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
52 # open and read file
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
53 inp = open(filename, "r")
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
54 lines = inp.readlines()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
55 inp.close()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
56
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
57 fingerprint = extractFingerprint(lines)
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
58 if fingerprint is None:
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
59 return 0
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
60 del lines[-1]
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
61
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
62 # calculate current digest
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
63 digest = sha.new()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
64 for line in lines:
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
65 digest.update(line)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
66
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
67 # compare current to stored digest
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
68 return fingerprint == digest.hexdigest()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
69
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
70
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
71 class DigestFile:
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
72 """ A class that you can use like open() and that calculates
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
73 and writes a SHA digest to the target file.
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
74 """
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
75
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
76 def __init__(self, filename):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
77 self.filename = filename
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
78 self.digest = sha.new()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
79 self.file = open(self.filename, "w")
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
80
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
81 def write(self, data):
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
82 lines = data.splitlines()
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
83 # if the file is coming from an installed tracker being used as a
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
84 # template, then we will want to re-calculate the SHA
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
85 fingerprint = extractFingerprint(lines)
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
86 if fingerprint is not None:
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
87 data = '\n'.join(lines[:-1]) + '\n'
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
88 self.file.write(data)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
89 self.digest.update(data)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
90
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
91 def close(self):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
92 file, ext = os.path.splitext(self.filename)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
93
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
94 if ext in sgml_file_types:
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
95 self.file.write("<!-- SHA: %s -->\n" % (self.digest.hexdigest(),))
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
96 elif ext in hash_file_types:
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
97 self.file.write("#SHA: %s\n" % (self.digest.hexdigest(),))
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
98 elif ext in slast_file_types:
385
66e18fd9f5a2 bleah typo
Richard Jones <richard@users.sourceforge.net>
parents: 384
diff changeset
99 self.file.write("/* SHA: %s */\n" % (self.digest.hexdigest(),))
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
100
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
101 self.file.close()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
102
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
103
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
104 def copyDigestedFile(src, dst, copystat=1):
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
105 """ Copy data from `src` to `dst`, adding a fingerprint to `dst`.
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
106 If `copystat` is true, the file status is copied, too
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
107 (like shutil.copy2).
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
108 """
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
109 if os.path.isdir(dst):
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
110 dst = os.path.join(dst, os.path.basename(src))
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
111
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
112 dummy, ext = os.path.splitext(src)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
113 if ext not in digested_file_types:
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
114 if copystat:
388
2d92f93072f9 Code using copyDigestedFile() that passes unit tests
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 387
diff changeset
115 return shutil.copy2(src, dst)
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
116 else:
388
2d92f93072f9 Code using copyDigestedFile() that passes unit tests
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 387
diff changeset
117 return shutil.copyfile(src, dst)
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
118
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
119 fsrc = None
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
120 fdst = None
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
121 try:
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
122 fsrc = open(src, 'r')
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
123 fdst = DigestFile(dst)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
124 shutil.copyfileobj(fsrc, fdst)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
125 finally:
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
126 if fdst: fdst.close()
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
127 if fsrc: fsrc.close()
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
128
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
129 if copystat: shutil.copystat(src, dst)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
130
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
131
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
132 def test():
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
133 import sys
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
134
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
135 testdata = open(sys.argv[0], 'r').read()
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
136
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
137 for ext in digested_file_types:
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
138 testfile = "__digest_test" + ext
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
139
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
140 out = DigestFile(testfile)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
141 out.write(testdata)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
142 out.close()
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
143
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
144 assert checkDigest(testfile), "digest ok w/o modification"
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
145
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
146 mod = open(testfile, 'r+')
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
147 mod.seek(0)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
148 mod.write('# changed!')
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
149 mod.close()
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
150
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
151 assert not checkDigest(testfile), "digest fails after modification"
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
152
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
153 os.remove(testfile)
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
154
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
155
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
156 if __name__ == '__main__':
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
157 test()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
158
1090
9b910e8d987d removed Log
Richard Jones <richard@users.sourceforge.net>
parents: 409
diff changeset
159 # vim: set filetype=python ts=4 sw=4 et si
Roundup Issue Tracker: http://roundup-tracker.org/