1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
|
#! /usr/bin/env python
# encoding: utf-8
import os
all_modifs={}
def modif(dir,name,fun):
if name=='*':
lst=[]
for y in'. Tools 3rdparty'.split():
for x in os.listdir(os.path.join(dir,y)):
if x.endswith('.py'):
lst.append(y+os.sep+x)
for x in lst:
modif(dir,x,fun)
return
filename=os.path.join(dir,name)
f=open(filename,'r')
txt=f.read()
f.close()
txt=fun(txt)
f=open(filename,'w')
f.write(txt)
f.close()
def subst(filename):
def do_subst(fun):
global all_modifs
try:
all_modifs[filename]+=fun
except KeyError:
all_modifs[filename]=[fun]
return fun
return do_subst
def r1(code):
code=code.replace("'iluvcuteoverload'","b'iluvcuteoverload'")
code=code.replace("ABI=7","ABI=37")
return code
def r2(code):
code=code.replace("p.stdin.write('\\n')","p.stdin.write(b'\\n')")
code=code.replace('p.communicate()[0]','p.communicate()[0].decode("utf-8")')
return code
def r3(code):
code=code.replace("m.update(str(lst))","m.update(str(lst).encode())")
code=code.replace('p.communicate()[0]','p.communicate()[0].decode("utf-8")')
return code
def r33(code):
code=code.replace('unicode','str')
return code
def r4(code):
code=code.replace("up(self.__class__.__name__)","up(self.__class__.__name__.encode())")
code=code.replace("up(self.env.variant())","up(self.env.variant().encode())")
code=code.replace("up(x.parent.abspath())","up(x.parent.abspath().encode())")
code=code.replace("up(x.name)","up(x.name.encode())")
code=code.replace('class TaskBase(object):\n\t__metaclass__=store_task_type','import binascii\n\nclass TaskBase(object, metaclass=store_task_type):')
code=code.replace('keys=self.cstr_groups.keys()','keys=list(self.cstr_groups.keys())')
code=code.replace("sig.encode('hex')",'binascii.hexlify(sig)')
return code
def r5(code):
code=code.replace("cPickle.dump(data,file,-1)","cPickle.dump(data,file)")
code=code.replace('for node in src_dir_node.childs.values():','for node in list(src_dir_node.childs.values()):')
return code
def r6(code):
code=code.replace('xrange','range')
code=code.replace('iteritems','items')
code=code.replace('maxint','maxsize')
code=code.replace('iterkeys','keys')
code=code.replace('Error,e:','Error as e:')
code=code.replace('Exception,e:','Exception as e:')
return code
def r7(code):
code=code.replace('class task_gen(object):\n\t__metaclass__=register_obj','class task_gen(object, metaclass=register_obj):')
return code
def r8(code):
code=code.replace('p.communicate()[0]','p.communicate()[0].decode("utf-8")')
return code
def r9(code):
code=code.replace('f.write(c)','f.write(c.encode("utf-8"))')
return code
def fixdir(dir):
global all_modifs
for k in all_modifs:
for v in all_modifs[k]:
modif(os.path.join(dir,'wafadmin'),k,v)
subst('Constants.py')(r1)
subst('Tools/ccroot.py')(r2)
subst('Utils.py')(r3)
subst('ansiterm.py')(r33)
subst('Task.py')(r4)
subst('Build.py')(r5)
subst('*')(r6)
subst('TaskGen.py')(r7)
subst('Tools/config_c.py')(r8)
subst('Tools/python.py')(r8)
subst('Tools/glib2.py')(r9)
|