U
    _{f                     @  s  d dl mZ d dlmZ d dlZd dlmZ d dlZd dlm	Z	m
Z
mZ d dlZd dlmZmZmZ d dlmZ d dlmZ d d	lmZ d d
lmZ eddddddZedddddZed&ddddddZeddddZeddd d!Zed'ddd"d#Zd$d% ZdS )(    )annotations)contextmanagerN)Path)IOAny	Generator)
BaseBufferCompressionOptionsFilePath)PYPY)ChainedAssignmentError)
set_option)
get_handlezFilePath | BaseBufferr	   z Generator[IO[bytes], None, None])pathcompressionreturnc              	   c  s(   t | d|dd}|jV  W 5 Q R X dS )a&  
    Open a compressed file and return a file object.

    Parameters
    ----------
    path : str
        The path where the file is read from.

    compression : {'gzip', 'bz2', 'zip', 'xz', 'zstd', None}
        Name of the decompression to use

    Returns
    -------
    file object
    rbF)r   Zis_textN)r   handle)r   r   r    r   W/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/pandas/_testing/contexts.pydecompress_file   s    r   strzGenerator[(None, None, None)])tzr   c                 #  sH   ddl  dd fdd}tjd}||  z
dV  W 5 || X dS )a  
    Context manager for temporarily setting a timezone.

    Parameters
    ----------
    tz : str
        A string representing a valid timezone.

    Examples
    --------
    >>> from datetime import datetime
    >>> from dateutil.tz import tzlocal
    >>> tzlocal().tzname(datetime(2021, 1, 1))  # doctest: +SKIP
    'IST'

    >>> with set_timezone('US/Eastern'):
    ...     tzlocal().tzname(datetime(2021, 1, 1))
    ...
    'EST'
    r   NNoner   c                   sB   | d kr,zt jd= W q> tk
r(   Y q>X n| t jd<    d S )NTZ)osenvironKeyErrortzset)r   timer   r   setTZJ   s    
zset_timezone.<locals>.setTZr   )r!   r   r   get)r   r"   Zorig_tzr   r    r   set_timezone2   s    

r$   Fboolr   zGenerator[Any, None, None])return_filelikekwargsr   c              	   k  s   t t }| dkrd} tt |  } ||  }|  t|}|r\|dd t|f|}z
|V  W 5 t	|tsz|
  | r|  X dS )az  
    Gets a temporary path and agrees to remove on close.

    This implementation does not use tempfile.mkstemp to avoid having a file handle.
    If the code using the returned path wants to delete the file itself, windows
    requires that no program has a file handle to it.

    Parameters
    ----------
    filename : str (optional)
        suffix of the created file.
    return_filelike : bool (default False)
        if True, returns a file-like which is *always* cleaned. Necessary for
        savefig and other functions which want to append extensions.
    **kwargs
        Additional keywords are passed to open().

    N modezw+b)r   tempfile
gettempdirr   uuiduuid4touch
setdefaultopen
isinstancecloseis_fileunlink)filenamer&   r'   folderr   Zhandle_or_strr   r   r   ensure_clean\   s     

r7   r   c               	   c  s2   t tj} z
dV  W 5 tj  tj|  X dS )z
    Get a context manager to safely set environment variables

    All changes will be undone on close, hence environment variables set
    within this contextmanager will neither persist nor change global state.
    N)dictr   r   clearupdate)Zsaved_environr   r   r   !ensure_safe_environment_variables   s
    


r;   c              	   k  sL   ddl }dddh}| |kr"td|j| f| z
dV  W 5 ||  X dS )au  
    Context manager to temporarily register a CSV dialect for parsing CSV.

    Parameters
    ----------
    name : str
        The name of the dialect.
    kwargs : mapping
        The parameters for the dialect.

    Raises
    ------
    ValueError : the name of the dialect conflicts with a builtin one.

    See Also
    --------
    csv : Python's CSV library.
    r   Nexcelz	excel-tabunixz Cannot override builtin dialect.)csv
ValueErrorregister_dialectunregister_dialect)namer'   r>   Z_BUILTIN_DIALECTSr   r   r   with_csv_dialect   s    

rC   c              	   c  sX   ddl m} |d kr|j}|j}|j}td|  ||_z
d V  W 5 ||_td| X d S )Nr   )expressionszcompute.use_numexpr)Zpandas.core.computationrD   Z_MIN_ELEMENTSZUSE_NUMEXPRr   )ZuseZmin_elementsexprZolduseZoldminr   r   r   use_numexpr   s    

rF   c                  C  s2   t rddlm}  |  S ddlm} |tddS d S )Nr   )nullcontext)assert_produces_warningzYA value is trying to be set on a copy of a DataFrame or Series through chained assignment)match)r   
contextlibrG   Zpandas._testingrH   r   )rG   rH   r   r   r   raises_chained_assignment_error   s    rK   )NF)N) 
__future__r   rJ   r   r   pathlibr   r*   typingr   r   r   r,   Zpandas._typingr   r	   r
   Zpandas.compatr   Zpandas.errorsr   Zpandasr   Zpandas.io.commonr   r   r$   r7   r;   rC   rF   rK   r   r   r   r   <module>   s4   )   ,!