From 73a22f0500ce713d46fce864e95402c34a45afd9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20v=2E=20L=C3=B6wis?= <martin@v.loewis.de>
Date: Sat, 22 Mar 2008 00:35:10 +0000
Subject: Add build_py_2to3.

---
 Lib/distutils/command/build_py.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'Lib/distutils/command/build_py.py')

diff --git a/Lib/distutils/command/build_py.py b/Lib/distutils/command/build_py.py
index 63ced4b470..f39ffb878e 100644
--- a/Lib/distutils/command/build_py.py
+++ b/Lib/distutils/command/build_py.py
@@ -383,3 +383,27 @@ class build_py (Command):
         if self.optimize > 0:
             byte_compile(files, optimize=self.optimize,
                          force=self.force, prefix=prefix, dry_run=self.dry_run)
+
+class build_py_2to3(build_py):
+    def run(self):
+        from lib2to3.refactor import RefactoringTool
+        self.updated_files = []
+        build_py.run(self)
+        class Options:
+            pass
+        o = Options()
+        o.doctests_only = False
+        o.fix = []
+        o.list_fixes = []
+        o.print_function = False
+        o.verbose = False
+        o.write = True
+        r = RefactoringTool(o)
+        r.refactor_args(self.updated_files)
+
+    def build_module(self, module, module_file, package):
+        res = build_py.build_module(self, module, module_file, package)
+        if res[1]:
+            # file was copied
+            self.updated_files.append(res[0])
+        return res
-- 
cgit v1.2.1