Mercurial > p > roundup > code

annotate roundup/install_util.py @ 6721:c6dc2e8d59a8

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Try to fix install failure for xapian on 3.11-dev and newer. distutils.sysconfig.get_config_vars('SO') is returning None on these platforms. Use: sysconfig.get_config_vars('SO') instead. This works with my 3.6 install, so hopefully will work with all of the 3.x releases we have.
author John Rouillard <rouilj@ieee.org>
date Thu, 23 Jun 2022 01:24:58 -0400
parents 7ada5d44b21d
children 050bcfc801c3
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
6068
7ada5d44b21d flake8 whitespace fixes.
John Rouillard <rouilj@ieee.org>
parents: 5427
diff changeset
34
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
35 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
36 # 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
37 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
38 # 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
39 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
40 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
41 # 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
42 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
43 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
44 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
45 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
46 # 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
47 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
48 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
49 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
50 return None
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
51
6068
7ada5d44b21d flake8 whitespace fixes.
John Rouillard <rouilj@ieee.org>
parents: 5427
diff changeset
52
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
53 def checkDigest(filename):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
54 """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
55 # 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
56 inp = open(filename, "rb")
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
57 lines = inp.readlines()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
58 inp.close()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
59
1910
d19fd344bd1f recalculate SHA on template files when installed tracker used as template
Richard Jones <richard@users.sourceforge.net>
parents: 1090
diff changeset
60 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
61 if fingerprint is None:
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
62 return 0
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
63 del lines[-1]
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
64
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
65 # 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
66 digest = sha1()
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
67 for line in lines:
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
68 digest.update(line)
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 # 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
71 return fingerprint == s2b(digest.hexdigest())
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
72
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
73
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
74 class DigestFile:
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
75 """ 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
76 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
77 """
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
78
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
79 def __init__(self, filename):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
80 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
81 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
82 self.file = open(self.filename, "wb")
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
83
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
84 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
85 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
86 # 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
87 # 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
88 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
89 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
90 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
91 self.file.write(data)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
92 self.digest.update(data)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
93
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
94 def close(self):
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
95 file, ext = os.path.splitext(self.filename)
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
96
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
97 if ext in sgml_file_types:
6068
7ada5d44b21d flake8 whitespace fixes.
John Rouillard <rouilj@ieee.org>
parents: 5427
diff changeset
98 self.file.write(s2b("<!-- SHA: %s -->\n" %
7ada5d44b21d flake8 whitespace fixes.
John Rouillard <rouilj@ieee.org>
parents: 5427
diff changeset
99 (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
100 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
101 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
102 elif ext in slast_file_types:
6068
7ada5d44b21d flake8 whitespace fixes.
John Rouillard <rouilj@ieee.org>
parents: 5427
diff changeset
103 self.file.write(s2b("/* SHA: %s */\n" %
7ada5d44b21d flake8 whitespace fixes.
John Rouillard <rouilj@ieee.org>
parents: 5427
diff changeset
104 (self.digest.hexdigest(),)))
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
105
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
106 self.file.close()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
107
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
108
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
109 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
110 """ 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
111 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
112 (like shutil.copy2).
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
113 """
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
114 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
115 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
116
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
117 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
118 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
119 if copystat:
388
2d92f93072f9 Code using copyDigestedFile() that passes unit tests
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 387
diff changeset
120 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
121 else:
388
2d92f93072f9 Code using copyDigestedFile() that passes unit tests
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 387
diff changeset
122 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
123
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
124 fsrc = None
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
125 fdst = None
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
126 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
127 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
128 fdst = DigestFile(dst)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
129 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
130 finally:
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
131 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
132 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
133
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
134 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
135
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
136
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
137 def test():
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
138 import sys
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
139
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
140 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
141
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
142 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
143 testfile = "__digest_test" + ext
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 out = DigestFile(testfile)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
146 out.write(testdata)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
147 out.close()
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
148
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
149 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
150
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
151 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
152 mod.seek(0)
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
153 mod.write('# changed!')
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
154 mod.close()
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
155
387
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
156 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
157
c066d3c91728 Copy function, and proper handling of unknown file types
Jürgen Hermann <jhermann@users.sourceforge.net>
parents: 385
diff changeset
158 os.remove(testfile)
383
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
159
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
160
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
161 if __name__ == '__main__':
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
162 test()
a213e2a90364 Added install utils (digest calculation)
Jürgen Hermann <jhermann@users.sourceforge.net>
parents:
diff changeset
163
1090
9b910e8d987d removed Log
Richard Jones <richard@users.sourceforge.net>
parents: 409
diff changeset
164 # vim: set filetype=python ts=4 sw=4 et si
Roundup Issue Tracker: http://roundup-tracker.org/