| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 """ | |
| 3 Helper script to rebuild virtualenv.py from virtualenv_support | |
| 4 """ | |
| 5 | |
| 6 import re | |
| 7 import os | |
| 8 import sys | |
| 9 | |
| 10 here = os.path.dirname(__file__) | |
| 11 script = os.path.join(here, '..', 'virtualenv.py') | |
| 12 | |
| 13 file_regex = re.compile( | |
| 14 r'##file (.*?)\n([a-zA-Z][a-zA-Z0-9_]+)\s*=\s*convert\("""(.*?)"""\)', | |
| 15 re.S) | |
| 16 file_template = '##file %(filename)s\n%(varname)s = convert("""\n%(data)s""")' | |
| 17 | |
| 18 def rebuild(): | |
| 19 f = open(script, 'rb') | |
| 20 content = f.read() | |
| 21 f.close() | |
| 22 parts = [] | |
| 23 last_pos = 0 | |
| 24 match = None | |
| 25 for match in file_regex.finditer(content): | |
| 26 parts.append(content[last_pos:match.start()]) | |
| 27 last_pos = match.end() | |
| 28 filename = match.group(1) | |
| 29 varname = match.group(2) | |
| 30 data = match.group(3) | |
| 31 print('Found reference to file %s' % filename) | |
| 32 pathname = os.path.join(here, '..', 'virtualenv_embedded', filename) | |
| 33 f = open(pathname, 'rb') | |
| 34 c = f.read() | |
| 35 f.close() | |
| 36 new_data = c.encode('zlib').encode('base64') | |
| 37 if new_data == data: | |
| 38 print(' Reference up to date (%s bytes)' % len(c)) | |
| 39 parts.append(match.group(0)) | |
| 40 continue | |
| 41 print(' Content changed (%s bytes -> %s bytes)' % ( | |
| 42 zipped_len(data), len(c))) | |
| 43 new_match = file_template % dict( | |
| 44 filename=filename, | |
| 45 varname=varname, | |
| 46 data=new_data) | |
| 47 parts.append(new_match) | |
| 48 parts.append(content[last_pos:]) | |
| 49 new_content = ''.join(parts) | |
| 50 if new_content != content: | |
| 51 sys.stdout.write('Content updated; overwriting... ') | |
| 52 f = open(script, 'wb') | |
| 53 f.write(new_content) | |
| 54 f.close() | |
| 55 print('done.') | |
| 56 else: | |
| 57 print('No changes in content') | |
| 58 if match is None: | |
| 59 print('No variables were matched/found') | |
| 60 | |
| 61 def zipped_len(data): | |
| 62 if not data: | |
| 63 return 'no data' | |
| 64 try: | |
| 65 return len(data.decode('base64').decode('zlib')) | |
| 66 except: | |
| 67 return 'unknown' | |
| 68 | |
| 69 if __name__ == '__main__': | |
| 70 rebuild() | |
| 71 | |
| OLD | NEW |