B
    us^~                 @   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dgZd	e Zed
e ejZede ejZdd Zddde gd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g Zeedd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   nmz-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   c/home/fristb/BIRCH/lib-linux-x86_64/python/lib64/python3.7/site-packages/numpy/distutils/lib2def.py	parse_cmd(   s"    ,,


r   zpython%s.libTc             C   sH   t j| |t jt jdd}| \}}|jdkrDtdd| |f |S )zUReturns the output of nm_cmd via a pipe.

nm_output = getnm(nm_cmd = 'nm -Cs py_lib')T)shellstdoutstderruniversal_newlinesr   zfailed to run "%s": "%s" )
subprocessPopenPIPEcommunicate
returncodeRuntimeErrorjoin)nm_cmdr   p	nm_outputZnm_errr   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   init)DATA_REfindallFUNC_REappendsort)r"   datafuncflistsymdlistr   r   r   parse_nmL   s    


8
(r0   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_defb   s    

r5   __main__wF)r   ) 
__future__r   r   r   rer   r   __doc____version__tupleversion_infor   
DEFAULT_NM
DEF_HEADERcompile	MULTILINEr(   r&   r   r#   r0   r   r5   __name__r   r   openstrr    r"   r/   r-   r   r   r   r   <module>   s.   


