Mercurial > p > roundup > code
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 |
| 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 | 160 # vim: set filetype=python ts=4 sw=4 et si |
