B
    n\                 @   s  d dl mZmZmZ d dlZd dlZd dlZdZdZde	ej
dd  ZdZde Zed	e ejZed
e ejZdd Zddde gfddZdd ZejfddZedkre \ZZedkrejZn
eedZeeeegZeeZee\ZZeeeee dS )    )divisionabsolute_importprint_functionNa  This 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<py_ver>.lib and output.def defaults to stdout

Author: Robert Kern <kernr@mail.ncifcrf.gov>
Last Update: April 30, 1999
z0.1az%d%d   znm -CszrLIBRARY         python%s.dll
;CODE           PRELOAD MOVEABLE DISCARDABLE
;DATA           PRELOAD SINGLE

EXPORTS
z^(.*) in python%s\.dllz^_imp__(.*) in python%s\.dllc              C   s  t tjdkrtjd dd dkrNtjd dd dkrNtjdd \} }nPtjd dd dkrtjd dd dkrtjdd \}} ntd td	 njt tjdkrtjd dd dkrtjd }d
t } n$tjd dd dkrd}tjd } nd
t } d}| |fS )zBParses the command-line arguments.

libfile, deffile = parse_cmd()      Nz.libr   z.defz4I'm assuming that your first argument is the libraryzand the second is the DEF file.zpython%s.lib)lensysargvprintpy_ver)libfiledeffile r   6lib/python3.7/site-packages/numpy/distutils/lib2def.py	parse_cmd(   s"    ,,


r   Znmz-Cszpython%s.libc             C   s,   t j| dt jdd}|j }|j  |S )zVReturns the output of nm_cmd via a pipe.

nm_output = getnam(nm_cmd = 'nm -Cs py_lib')T)shellstdoutZuniversal_newlines)
subprocessPopenPIPEr   readclose)nm_cmdf	nm_outputr   r   r   getnm@   s    

r   c             C   s   t | }t| }g }xN|D ]F}||kr|dd dksZ|dd dksZ|dd dkr|| qW g }x>|D ]6}||krr|dd dks|dd dkrr|| qrW |  |  ||fS )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)Nr   ZPyr   Z_Py   Zinit)DATA_REfindallFUNC_REappendsort)r   datafuncflistZsymdlistr   r   r   parse_nmI   s    


8
(r(   c             C   sJ   x| D ]}|d|  }qW |d }x|D ]}|d|  }q(W | | dS )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&   headerfileZdata_symZfunc_symr   r   r   
output_def_   s    

r-   __main__w) Z
__future__r   r   r   rer
   r   __doc____version__tupleversion_infor   Z
DEFAULT_NMZ
DEF_HEADERcompile	MULTILINEr!   r   r   r   r(   r   r-   __name__r   r   openstrr   r   r'   r&   r   r   r   r   <module>   s.   	


