U
    ~fhG                     @   s   U d dl Z d dlZd dlmZ ddlmZ edZeje	d< edZ
eje	d< d	d
hZddddZddddZG dd dejZdS )    N)override   )is_dictZopenailoggerZhttpxhttpx_loggerzapi-keyauthorization)returnc                   C   s   t jddd d S )Nz?[%(asctime)s - %(name)s:%(lineno)d - %(levelname)s] %(message)sz%Y-%m-%d %H:%M:%S)formatdatefmt)loggingbasicConfig r   r   7/tmp/pip-unpacked-wheel-rjiviquo/openai/_utils/_logs.py_basic_config   s    r   c                  C   s^   t jd} | dkr4t  ttj ttj n&| dkrZt  ttj	 ttj	 d S )NZ
OPENAI_LOGdebuginfo)
osenvirongetr   r   setLevelr   DEBUGr   INFO)envr   r   r   setup_logging   s    r   c                   @   s"   e Zd ZeejedddZdS )SensitiveHeadersFilter)recordr   c                 C   s^   t |jrZd|jkrZt |jd rZ|jd  }|jd< |D ]}t| tkr<d||< q<dS )Nheadersz
<redacted>T)r   argsstrlowerSENSITIVE_HEADERS)selfr   r   headerr   r   r   filter#   s    "
zSensitiveHeadersFilter.filterN)__name__
__module____qualname__r   r   	LogRecordboolr#   r   r   r   r   r   "   s   r   )r   r   Ztyping_extensionsr   _utilsr   	getLoggerr   Logger__annotations__r   r    r   r   Filterr   r   r   r   r   <module>   s    