
    5i                     .    d dl mZ d dlZ G d de      Zy)    )print_functionNc                   (    e Zd ZefdZd Zd Zd Zy)	VizClientc                 .    || _         || _        || _        y )N)url
project_iderror_handler)selfr   r   r	   s       /home/marpiech/ifpan-abm-pgxpred/analysis/marpiech-gwas-test/venv/lib/python3.12/site-packages/dxpy/bindings/apollo/vizclient.py__init__zVizClient.__init__   s    $*    c                 D   	 t        j                  ||d      }d|v rR|d   d   dk(  rd|d   d   z   }n!|d   d   dk(  rd	}n|d   d   d
k(  rd	}n|d   }| j                  t        |             |S # t        $ r$}| j                  t        |             Y d }~y d }~ww xY w)NF)resourcedataprepend_srverrortypeInvalidInputz4Insufficient permissions due to the project policy.
messageQueryTimeOutzmPlease consider using --sql option to generate the SQL query and execute query via a private compute cluster.DataTooLarge)dxpyDXHTTPRequestr	   str	Exception)r
   payloadresource_urlresponseerr_messagedetailss         r   _get_responsezVizClient._get_response
   s    	-))%GH ("G$V,>O"7+I67   g&v..@ #RKg&v..@ #RK"*7"3K""3{#34O 	-s7|,,	-s   A/A2 2	B;BBc                 ^    dj                  | j                  |      }| j                  ||      S )Nz{}/data/3.0/{}/rawformatr   r!   r
   r   	record_idr   s       r   get_datazVizClient.get_data    s+    +22488YG!!'<88r   c                 ^    dj                  | j                  |      }| j                  ||      S )Nz{}/viz-query/3.0/{}/raw-queryr#   r%   s       r   get_raw_sqlzVizClient.get_raw_sql$   s+    6==dhh	R!!'<88r   N)__name__
__module____qualname__printr   r!   r'   r)    r   r   r   r      s    6; +
-,99r   r   )
__future__r   r   objectr   r.   r   r   <module>r1      s    % "9 "9r   