3 h<@sddlZddlZddlZdZdZdeejddZddgZdeZ ej d eej Z ej d eej Z d d Zddd egdfddZddZejfddZedkre\ZZedkrejZn eedZeeegZeeddZee\ZZeeee edS)NaThis module generates a DEF file from the symbols in an MSVC-compiled DLL import library. It correctly discriminates between data and functions. The data is collected from the output of the program nm(1). Usage: python lib2def.py [libname.lib] [output.def] or python lib2def.py [libname.lib] > output.def libname.lib defaults to python.lib and output.def defaults to stdout Author: Robert Kern Last Update: April 30, 1999 z0.1az%d%dnmz-CszrLIBRARY python%s.dll ;CODE PRELOAD MOVEABLE DISCARDABLE ;DATA PRELOAD SINGLE EXPORTS z^(.*) in python%s\.dllz^_imp__(.*) in python%s\.dllcCsttjdkrtjdd ddkrNtjdd ddkrNtjdd\}}nPtjdd ddkrtjddddkrtjdd\}}ntdtd nnttjdkrtjddddkrtjd}d t}n&tjddddkrd}tjd}n d t}d}||fS)zBParses the command-line arguments. libfile, deffile = parse_cmd()Nz.librz.defz4I'm assuming that your first argument is the libraryzand the second is the DEF file.z python%s.librrrrr)lensysargvprintpy_ver)libfiledeffiler8/tmp/pip-build-fibhr3ey/numpy/numpy/distutils/lib2def.py parse_cmd&s",,    rz python%s.libTcCsHtj||tjtjdd}|j\}}|jdkrDtddj||f|S)zUReturns the output of nm_cmd via a pipe. nm_output = getnm(nm_cmd = 'nm -Cs py_lib')T)shellstdoutstderruniversal_newlinesrzfailed to run "%s": "%s" ) subprocessPopenPIPE communicate returncode RuntimeErrorjoin)nm_cmdrp nm_outputZnm_errrrrgetnm>s    r!cCstj|}tj|}g}xN|D]F}||kr|dddksZ|dddksZ|dddkr|j|qWg}x>|D]6}||krr|dddks|dddkrr|j|qrW|j|j||fS)zReturns a tuple of lists: dlist for the list of data symbols and flist for the list of function symbols. dlist, flist = parse_nm(nm_output)NrZPyrZ_Pyrinit)DATA_REfindallFUNC_REappendsort)r datafuncflistsymdlistrrrparse_nmJs   8 (r-cCsJx|D]}|d|}qW|d}x|D]}|d|}q(W|j|dS)zoOutputs the final DEF file to a file defaulting to stdout. output_def(dlist, flist, header, file = sys.stdout)z %s DATA  z %s N)write)r,r*headerfileZdata_symZfunc_symrrr output_def`s   r2__main__wF)r)rer r__doc__ __version__tuple version_infor Z DEFAULT_NMZ DEF_HEADERcompile MULTILINEr%r#rr!r-rr2__name__r ropenstrrr r,r*rrrrs,