OLD | NEW |
(Empty) | |
| 1 # -*- coding: utf-8 -*- |
| 2 """ |
| 3 jinja2._markupsafe._bundle |
| 4 ~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| 5 |
| 6 This script pulls in markupsafe from a source folder and |
| 7 bundles it with Jinja2. It does not pull in the speedups |
| 8 module though. |
| 9 |
| 10 :copyright: Copyright 2010 by the Jinja team, see AUTHORS. |
| 11 :license: BSD, see LICENSE for details. |
| 12 """ |
| 13 import sys |
| 14 import os |
| 15 import re |
| 16 |
| 17 |
| 18 def rewrite_imports(lines): |
| 19 for idx, line in enumerate(lines): |
| 20 new_line = re.sub(r'(import|from)\s+markupsafe\b', |
| 21 r'\1 jinja2._markupsafe', line) |
| 22 if new_line != line: |
| 23 lines[idx] = new_line |
| 24 |
| 25 |
| 26 def main(): |
| 27 if len(sys.argv) != 2: |
| 28 print 'error: only argument is path to markupsafe' |
| 29 sys.exit(1) |
| 30 basedir = os.path.dirname(__file__) |
| 31 markupdir = sys.argv[1] |
| 32 for filename in os.listdir(markupdir): |
| 33 if filename.endswith('.py'): |
| 34 f = open(os.path.join(markupdir, filename)) |
| 35 try: |
| 36 lines = list(f) |
| 37 finally: |
| 38 f.close() |
| 39 rewrite_imports(lines) |
| 40 f = open(os.path.join(basedir, filename), 'w') |
| 41 try: |
| 42 for line in lines: |
| 43 f.write(line) |
| 44 finally: |
| 45 f.close() |
| 46 |
| 47 |
| 48 if __name__ == '__main__': |
| 49 main() |
OLD | NEW |