JezK
Edit File: general.cpython-38.pyc
U �C�e� � @ sz d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z dZ dZ dd� Z dd� Z d d � Zdd d �Zdd� ZdS )� N��Pathz0.0.1� c G s� dt jkrt jd }nHt�� }|s*td��t�� }|dkrPt j�|dddd�}nt j�|d�}zt|�j d d � W n t k r� Y nX | D ]}t j�||�}q�|S ) z�Returns the MySQL Shell's user directory The directory will be created if it does not exist yet Args: *argv: The list of directories to be added to the path Returns: The plugin directory path as string ZMYSQLSH_USER_CONFIG_HOMEzNo home directory setZWindowsZAppDataZRoamingZMySQL�mysqlshz.mysqlshT��parents) �os�environr �home� Exception�platform�system�path�join�mkdir�FileExistsError)�argvZ shell_dirZhome_dir�os_name�arg� r �B/usr/lib/mysqlsh/python-packages/mysqlsh/plugin_manager/general.py�get_shell_user_dir& s, �r c G sN t d�}zt|�jdd� W n tk r0 Y nX | D ]}tj�||�}q6|S )z�Returns the MySQL Shell's temp directory The directory will be created if it does not exist yet Args: *argv: The list of directories to be added to the path Returns: The temp directory path as string ZtempTr )r r r r r r r )r Ztemp_dirr r r r �get_shell_temp_dirK s r c G sb ddl m} ddl}td�}z||�jdd� W n tk rD Y nX | D ]}|j�||�}qJ|S )z�Returns the MySQL Shell's plugin directory The directory will be created if it does not exist yet Args: *argv: The list of directories to be added to the path Returns: The plugin directory path as string r r NZpluginsTr )�pathlibr Zos.pathr r r r r )r r r Z plugins_dirr r r r �get_plugins_dird s r Fc C s| z�t �� }|dkr4|dkr4t| |dtjdd�d�W S tjj| t|d��}|� � }W 5 Q R X |sj|� d�W S t |d ��}|� |� W 5 Q R X W dS W n� tj jk r� } z8|r�� |jd kr�td | � d t|�� �� W Y � dS � W 5 d}~X Y n� tj jk �rv } zht|j�} d | k�rF|dk�rFt� } t| ||tj| d�d� W Y �&S td | � d t|�� �� |�rf� W 5 d}~X Y nX dS )aj Downloads a file from a given URL Args: url (str): The URL to the text file to download file_path (str): The file to store the downloaded file raise_exceptions (bool): Whether exceptions are raised Returns: The contents of the file decoded to utf-8 if no file is given or True on success or None on failure NZDarwinTz/private/etc/ssl/cert.pem)�cafile)� file_path�raise_exceptions�ctx)Ztimeout�contextzutf-8�wbi� z Could not download file from z ERROR: ZCERTIFICATE_VERIFY_FAILED)r r � download_file�sslZcreate_default_context�urllibZrequestZurlopen�URL_OPEN_TIMEOUT_SECONDS�read�decode�open�write�errorZ HTTPError�code�print�strZURLError�reason�install_ssl_certificates) Zurlr r r r Zresponse�dataZout_file�er- r r r r r! � sF � �r! c C s� t jt jB t jB t jB t jB t jB t jB t jB } t j � t � d�}t d� t j �t�� �}t d� zt �|� W n tk r� Y nX t d� t �||� t d� t �|| � t d� |S )Nzcertifi-ca.pemz Installing SSL certificate...z& -- removing any existing file or linkz2 -- creating symlink to certifi certificate bundlez -- setting permissionsz -- update complete)�stat�S_IRUSR�S_IWUSR�S_IXUSR�S_IRGRP�S_IWGRP�S_IXGRP�S_IROTH�S_IXOTHr r r r r+ �relpath�certifi�where�remove�FileNotFoundError�symlink�chmod)Z STAT_0o775Z shell_cafileZrelpath_to_certifi_cafiler r r r. � s< �������� r. )NFN)r; r r Z urllib.errorr# Zurllib.requestr1 r" r r �VERSIONr$ r r r r! r. r r r r �<module> s % :