+++ /dev/null
-"""
-Convert modules from the old 'from wxPython.wx import *' style to
-the new 'import wx' style.
-
-Author: dfh@forestfield.co.uk (David Hughes)
-
-
-This should probably go in the tools package and have a startup script
-like the others, but I don't think it's ready for prime-time yet. So
-just put it in distrib for now so it doesn't get lost...
-
---Robin
-"""
-
-
-import os, sys, time
-import string
-t0 = time.time()
-
-
-import wx
-print 'length = ', len(wx._newnames)
-
-for key in ['true', 'True', 'false', 'False']: # use native Python not wx.<bool>
- if key in wx._newnames:
- del wx._newnames[key]
-
-import wx.calendar, wx.gizmos, wx.glcanvas, wx.grid, wx.help, wx.html, \
- wx.htmlhelp, wx.iewin, wx.ogl, wx.stc, wx.wizard, wx.xrc
-import wx.lib
-##import wx.lib.activexwrapper, wx.lib.analogclock, wx.lib.anchors, \
-import wx.lib.activexwrapper, wx.lib.anchors, \
- wx.lib.buttons, wx.lib.calendar, wx.lib.ClickableHtmlWindow, \
- wx.lib.colourdb, wx.lib.colourselect, wx.lib.dialogs, \
- wx.lib.ErrorDialogs, wx.lib.evtmgr, wx.lib.fancytext, \
- wx.lib.filebrowsebutton, wx.lib.gridmovers, wx.lib.grids, \
- wx.lib.imagebrowser, wx.lib.imageutils, wx.lib.infoframe, \
- wx.lib.intctrl, wx.lib.layoutf, wx.lib.multisash, \
- wx.lib.popupctl, wx.lib.printout, wx.lib.rcsizer, \
- wx.lib.rightalign, wx.lib.sheet, wx.lib.stattext, \
- wx.lib.throbber, wx.lib.timectrl, wx.lib.wxPlotCanvas, \
- wx.lib.wxpTag
-import wx.lib.mixins, wx.lib.editor, wx.lib.colourchooser
-import wx.lib.mixins.grid, wx.lib.mixins.imagelist, \
- wx.lib.mixins.listctrl
-## wx.lib.mixins.listctrl, wx.lib.mixins.rubberband
-import wx.lib.editor.editor, wx.lib.editor.images, wx.lib.editor.selection
-import wx.lib.colourchooser.canvas, wx.lib.colourchooser.intl, wx.lib.colourchooser.pycolourbox, \
- wx.lib.colourchooser.pycolourchooser, wx.lib.colourchooser.pycolourslider, wx.lib.colourchooser.pypalette
-
-wxlist = []
-for kd in wx._newnames.items():
- wxlist.append(kd)
-wxlist.sort()
-
-n = 0
-for item in wxlist:
- n += 1
-## print n, item
-print 'length = ', len(wx._newnames)
-print 'imports completed in ', time.time()-t0, 'secs'
-
-base_path = 'G:/wxConvert/Test1'
-exclude_dir = ( 'wx', 'wx_cc', 'reportlab', 'sqlite') # MUST exclude wx if it is below base_dir, others optional
-
-other_subs = { 'true': 'True',
- 'false': 'False'
- }
-
-punct = string.punctuation.replace('_','') # exclude single underscore
-punct = punct.replace('.','') # and period
-punct = punct.replace('*','') # and asterisk, all allowed in names
-punctable = string.maketrans(punct, ' '*len(punct)) # map punctuation characters to spaces
-numfiles = 0
-
-#----------------------------------------------------------------------------
-
-class AFile:
- " file object - collection of properties relating to current instance"
- def __init__(self, subsdict):
- self.subsdict = subsdict.copy() # dictionary of universal and local substitutions to make
- self.importing = [] # list of wx modules being imported
-
-#----------------------------------------------------------------------------
-
-def visit(noargs, thispath, contentlist):
- """ Function is called by os walk for every directory in base_path,
- including base_path itself. Contentlist is a list of files/dirs in thispath.
- Wx conversion function is called for every qualifying file in list
- """
- path = thispath
- base = 'something'
- while base: # check if thispath or a parent is excluded
- path, base = os.path.split(path)
- if base in exclude_dir:
- print 'Excluded:', thispath
- return
-
- for item in contentlist:
- pathname = os.path.join(thispath, item)
- if os.path.isfile(pathname) and pathname[-3:].lower() == '.py':
- wxconvert(pathname)
-
-def wxconvert(pathname):
- """ Scan each line of text in pathname. Replace each occurrence of any key in wx._newnames
- dictionary with the content stored under that key
- """
- global numfiles
- afile = AFile(other_subs)
- infile = open(pathname, 'r')
- linelist = infile.readlines()
- infile.close()
-
- lnum = 0
- outlist = []
- for line in linelist:
- lnum += 1
- tokenlist = line.replace('.__', ' ').translate(punctable).split() # split on white space and punctuation
- line, status = checkimports(afile, line, tokenlist) # return line with 'import' modifications
- if not status:
- print 'Unable to convert line %d in %s' % (lnum, pathname)
- break
- else:
- for key in afile.subsdict: # do other changes first
- if line.find(key) >= 0:
- line = line.replace(key, afile.subsdict[key])
- for token in tokenlist: # change wx names
- if token in wx._newnames:
- candidate = wx._newnames[token]
- module = candidate[:candidate.rfind('.')]
- if module in afile.importing:
- line = line.replace(token, candidate)
- outlist.append(line)
- else:
- outfile = open(pathname, 'w')
- outfile.writelines(outlist)
- numfiles += 1
- print 'Converted:', pathname
- outfile.close()
-
-def checkimports(afile, line, tlist):
- """ Scan tokenlist for wxPython import statement. Add to afile.subsdict any
- name changes that are necessary for the rest of the current source file.
- Add to afile.importing any new import modules
- Return a tuple (status, newstring) -
- line, possibly modified if an import statmeny
- status: 0: unable to handle complexity, 1: OK
- """
- aline = line
- if len(tlist) == 0: return (aline, 1)
- indent = ''
- for ch in line:
- if ch in string.whitespace:
- indent += ' '
- else:
- break
-
- if tlist[0] == 'import': # import module [as name] [module [as name]...]
- skip = False
- for t in tlist[1:]:
- if skip:
- skip = False
- elif t == 'as':
- skip = True
- elif t.startswith('wx'):
- aline = aline.replace(t, rename_module(afile, t))
- elif (tlist[0] == 'from' and
- tlist[1] == 'wxPython' and
- tlist[2] == 'import'): # from wxPython import module
- if len(tlist) > 4: # ...[as name] [module [as name]...]
- return ('', 0) # too complicated
- module = rename_module(afile,tlist[-1])
- aline = indent = 'import ' + module + '\n'
- elif (tlist[0] == 'from' and
- tlist[1].startswith('wxPython') and
- tlist[2] == 'import'): # from module import ....
- if tlist[-1] <> '*': # ...name [as name] [name [as name]...]
- aline = aline.replace(tlist[1], rename_module(afile, tlist[1]))
- skip = False
- for t in tlist[3:]:
- if skip:
- skip = False
- elif t == 'as':
- skip = True
- else:
- elem = t.split('.')
- if elem[-1].startswith('wx'): # remove wx prefix from last element of name
- elem[-1] = elem[-1][2:]
- afile.subsdict[t] = '.'.join(elem) # and apply to each occurrence in source
- else: # from module import *
- module = rename_module(afile,tlist[1])
- aline = indent = 'import ' + module + '\n'
- return (aline, 1)
-
-def rename_module(afile, t, type='A'):
- """ Substitute wx for wxPython.wx or wx, and wx.foo.bar for wxPython.foo.bar in token
- foo.bar => wx.foo.bar is also permitted (from wxPython import foo.bar)
- """
- if t in ['wx', 'wxPython.wx']:
- module = 'wx'
- elif t.startswith('wxPython'):
- module = t.replace('wxPython', 'wx')
- elif t.startswith('wx'):
- module = t
- else:
- module = 'wx.' + t
- if module not in afile.importing:
- afile.importing.append(module)
- return module
-
-def main():
- " Convert every file in base_dir and all subdirectories except in exclude_dir list"
-
- os.path.walk(base_path, visit, None)
- print '%d files converted in %.2f seconds' % (numfiles, time.time() - t0)
-
-#----------------------------------------------------------------------------
-
-if __name__ == '__main__':
- main()