annotate roundup/install_util.py @ 5639:f576957cbb1f

Add support for prev/next/self links when returning paginated results. To do this: 1) change "data" envelope from an array to a dict 2) move the "data" array to the "collection" property, which is an array of elements in the collection. 3) add @links dict keyed by link relation: self, next, prev. Each relation is an array of dicts with uri and rel keys. In this case there is only one element, but there is nothing preventing a relation from having multiple url's. So this follows the formatting needed for the general case. Relations are present only if it makes sense. So first page has no prev and last page has no next. 4) add @total_size with number of element selected if they were not paginated. Replicates data in X-Count-Total header. Changed index to start at 1. So the first page is page_index 1 and not page_index 0. (So I am no longer surprised when I set page_index to 1 and am missing a bunch of records 8-)). Also a small fixup, json response ends with a newline so printing the data, or using curl makes sure that anything printing after the json output (like shell prompts) is on a new line. Tests added for all cases.
author John Rouillard <rouilj@ieee.org>
date Sat, 09 Mar 2019 11:06:10 -0500
parents 88cf5614e0f4
children 7ada5d44b21d
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 #
406
bdc2ea127ae9 Added module docstrings to all modules.
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 388
diff changeset
18
2005
fc52d57c6c3e documentation cleanup
Richard Jones <richard@users.sourceforge.net>
parents: 1910
diff changeset
19 """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
20 """
2005
fc52d57c6c3e documentation cleanup
Richard Jones <richard@users.sourceforge.net>
parents: 1910
diff changeset
21 __docformat__ = 'restructuredtext'
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
22
4089
eddb82d0964c Add compatibility package to allow us to deal with Python versions 2.3..2.6.
Richard Jones <richard@users.sourceforge.net>
parents: 3496
diff changeset
23 import os, shutil
4982
9ba03348f923 Remove roundup/anypy/hashlib_.py
John Kristensen <john@jerrykan.com>
parents: 4570
diff changeset
24 from hashlib import sha1
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
25
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
26 from roundup.anypy.strings import s2b
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
27
3496
4ee255777c25 clean up digested_file_types [SF#1268303]
Richard Jones <richard@users.sourceforge.net>
parents: 2005
diff changeset
28 sgml_file_types = [".xml", ".ent", ".html"]
4ee255777c25 clean up digested_file_types [SF#1268303]
Richard Jones <richard@users.sourceforge.net>
parents: 2005
diff changeset
29 hash_file_types = [".py", ".sh", ".conf", ".cgi"]
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
30 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
31
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
32 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
33
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
34 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
35 # get fingerprint from last line
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
36 if lines[-1].startswith(b"#SHA: "):
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
37 # 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
38 return lines[-1][6:].strip()
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
39 elif lines[-1].startswith(b"<!-- SHA: "):
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
40 # 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
41 fingerprint = lines[-1][10:]
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
42 fingerprint = fingerprint.replace(b'-->', b'')
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
43 return fingerprint.strip()
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
44 elif lines[-1].startswith(b"/* SHA: "):
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
45 # 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
46 fingerprint = lines[-1][8:]
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
47 fingerprint = fingerprint.replace(b'*/', b'')
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
48 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
49 return None
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
50
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
51 def checkDigest(filename):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
52 """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
53 # open and read file
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
54 inp = open(filename, "rb")
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
55 lines = inp.readlines()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
56 inp.close()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
57
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
58 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
59 if fingerprint is None:
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
60 return 0
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
61 del lines[-1]
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
62
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
63 # calculate current digest
4089
eddb82d0964c Add compatibility package to allow us to deal with Python versions 2.3..2.6.
Richard Jones <richard@users.sourceforge.net>
parents: 3496
diff changeset
64 digest = sha1()
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
65 for line in lines:
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
66 digest.update(line)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
67
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
68 # compare current to stored digest
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
69 return fingerprint == s2b(digest.hexdigest())
383
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
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
72 class DigestFile:
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
73 """ 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
74 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
75 """
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
76
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
77 def __init__(self, filename):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
78 self.filename = filename
4089
eddb82d0964c Add compatibility package to allow us to deal with Python versions 2.3..2.6.
Richard Jones <richard@users.sourceforge.net>
parents: 3496
diff changeset
79 self.digest = sha1()
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
80 self.file = open(self.filename, "wb")
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
81
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
82 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
83 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
84 # 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
85 # 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
86 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
87 if fingerprint is not None:
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
88 data = b'\n'.join(lines[:-1]) + b'\n'
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
89 self.file.write(data)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
90 self.digest.update(data)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
91
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
92 def close(self):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
93 file, ext = os.path.splitext(self.filename)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
94
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
95 if ext in sgml_file_types:
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
96 self.file.write(s2b("<!-- 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
97 elif ext in hash_file_types:
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
98 self.file.write(s2b("#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
99 elif ext in slast_file_types:
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
100 self.file.write(s2b("/* SHA: %s */\n" % (self.digest.hexdigest(),)))
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
101
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
102 self.file.close()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
103
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
104
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
105 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
106 """ 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
107 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
108 (like shutil.copy2).
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
109 """
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
110 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
111 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
112
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
113 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
114 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
115 if copystat:
388
2d92f93072f9 Code using copyDigestedFile() that passes unit tests
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 387
diff changeset
116 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
117 else:
388
2d92f93072f9 Code using copyDigestedFile() that passes unit tests
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 387
diff changeset
118 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
119
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
120 fsrc = None
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
121 fdst = None
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
122 try:
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
123 fsrc = open(src, 'rb')
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
124 fdst = DigestFile(dst)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
125 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
126 finally:
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
127 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
128 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
129
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
130 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
131
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
132
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
133 def test():
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
134 import sys
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
135
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
136 testdata = open(sys.argv[0], 'rb').read()
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
137
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
138 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
139 testfile = "__digest_test" + ext
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
140
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
141 out = DigestFile(testfile)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
142 out.write(testdata)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
143 out.close()
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
144
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
145 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
146
5427
88cf5614e0f4 Python 3 preparation: use byte strings and binary I/O in roundup/install_util.py.
Joseph Myers <jsm@polyomino.org.uk>
parents: 4982
diff changeset
147 mod = open(testfile, 'r+b')
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
148 mod.seek(0)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
149 mod.write('# changed!')
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
150 mod.close()
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
151
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
152 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
153
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
154 os.remove(testfile)
383
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
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
157 if __name__ == '__main__':
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
158 test()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
159
1090
9b910e8d987d removed Log
Richard Jones <richard@users.sourceforge.net>
parents: 409
diff changeset
160 # vim: set filetype=python ts=4 sw=4 et si

Roundup Issue Tracker: http://roundup-tracker.org/