U
    f/e	                     @   sL   d Z ddlmZ ddlZddlmZ ddlmZmZ dddZ	dd
dZ
dS )z io on the clipboard     )StringION)ABCDataFrame)
get_optionoption_context\s+c           	      K   s  | dd}|dk	r0| dddkr0tddd	lm} dd
lm} | }z||	dpdt
d}W n tk
r~   Y nX |dd ddd dd }dd |D }t|dkrt|dkr|  dkrd} t|d t|d d }|dkr|dtt| | dkr8|	ddkr8d} t| dkr`|	ddkr`d|d< n(t| dkr|	ddkrtd |t|fd| i|S )aj  
    Read text from clipboard and pass to read_csv.

    Parameters
    ----------
    sep : str, default '\s+'
        A string or regex delimiter. The default of '\s+' denotes
        one or more whitespace characters.

    **kwargs
        See read_csv for the full argument list.

    Returns
    -------
    DataFrame
        A parsed DataFrame object.
    encodingutf-8N- utf8z3reading from clipboard only supports utf-8 encodingr   )clipboard_get)read_csvzdisplay.encodingi'  

   c                 S   s   h | ]}| d dqS ) 	)lstripcount).0x r   8/tmp/pip-unpacked-wheel-tiezk1ph/pandas/io/clipboards.py	<setcomp>=   s     z!read_clipboard.<locals>.<setcomp>   r   z 	Z	index_colZdelim_whitespacer   ZenginepythonczHread_clipboard with regex separator does not work properly with c enginesep)poplowerreplaceNotImplementedErrorpandas.io.clipboardr   Zpandas.io.parsersr   decodegetr   AttributeErrorsplitlenr   
setdefaultlistrangewarningswarnr   )	r   kwargsr   r   r   textlinescountsZindex_lengthr   r   r   read_clipboard   s4    "*

r1   Tc           	   	   K   s  | dd}|dk	r0| dddkr0tddd	lm} |dkrHd
}|rz@|dkrZd}t }| j|f|dd| | }|| W dS  t	k
r   t
d Y qX n|dk	rt
d t| trtdd | jf |}W 5 Q R X nt| }|| dS )a  
    Attempt to write text representation of object to the system clipboard
    The clipboard can be then pasted into Excel for example.

    Parameters
    ----------
    obj : the object to write to the clipboard
    excel : bool, defaults to True
            if True, use the provided separator, writing in a csv
            format for allowing easy pasting into excel.
            if False, write a string representation of the object
            to the clipboard
    sep : optional, defaults to tab
    other keywords are passed to to_csv

    Notes
    -----
    Requirements for your platform
      - Linux: xclip, or xsel (with PyQt4 modules)
      - Windows:
      - OS X:
    r   r   Nr	   r
   r   z&clipboard only supports utf-8 encodingr   )clipboard_setTr   )r   r   zAto_clipboard in excel mode requires a single character separator.z6to_clipboard with excel=False ignores the sep argumentzdisplay.max_colwidth)r   r   r    
ValueErrorr"   r2   r   Zto_csvgetvalue	TypeErrorr+   r,   
isinstancer   r   Z	to_stringstr)	objexcelr   r-   r   r2   bufr.   Zobjstrr   r   r   to_clipboardV   s4    

r;   )r   )TN)__doc__ior   r+   Zpandas.core.dtypes.genericr   Zpandasr   r   r1   r;   r   r   r   r   <module>   s   
I