관리-도구
편집 파일: _json.cpython-37.opt-1.pyc
B *��]Z � @ s� d Z ddlZddlmZmZ ddlmZmZmZ ddlm Z dZ dZ dZ d Z G d d � d e�Zddd�Zddd�Zddd�Zd dd�Zd dd�ZdS ) z�Implementation of the JSON adaptation objects This module exists to avoid a circular import problem: pyscopg2.extras depends on psycopg2.extension, so I can't create the default JSON typecasters in extensions importing register_json from extras. � N)� ISQLQuote� QuotedString)�new_type�new_array_type� register_type)�PY2�r �� i� i� c @ sP e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Ze rDd d� Z ndd� Z dS )�Jsona An `~psycopg2.extensions.ISQLQuote` wrapper to adapt a Python object to :sql:`json` data type. `!Json` can be used to wrap any object supported by the provided *dumps* function. If none is provided, the standard :py:func:`json.dumps()` is used. Nc C s || _ d | _|ptj| _d S )N)�adapted�_conn�json�dumps�_dumps)�selfr r � r �A/opt/alt/python37/lib64/python3.7/site-packages/psycopg2/_json.py�__init__8 s z Json.__init__c C s |t kr | S d S )N)r )r �protor r r � __conform__= s zJson.__conform__c C s | � |�S )z�Serialize *obj* in JSON format. The default is to call `!json.dumps()` or the *dumps* function provided in the constructor. You can override this method to create a customized JSON wrapper. )r )r �objr r r r A s z Json.dumpsc C s || _ d S )N)r )r �connr r r �prepareJ s z Json.preparec C s2 | � | j�}t|�}| jd k r*|�| j� |�� S )N)r r r r r � getquoted)r �sZqsr r r r M s zJson.getquotedc C s | � � S )N)r )r r r r �__str__U s z Json.__str__c C s | � � �dd�S )N�ascii�replace)r �decode)r r r r r X s )N) �__name__� __module__� __qualname__�__doc__r r r r r r r r r r r r . s r Fr c C sf |dkrt | |�\}}t||||�� d�\}}t|| r<| p>d� |dk r^t|| rX| pZd� ||fS )a Create and register typecasters converting :sql:`json` type to Python objects. :param conn_or_curs: a connection or cursor used to find the :sql:`json` and :sql:`json[]` oids; the typecasters are registered in a scope limited to this object, unless *globally* is set to `!True`. It can be `!None` if the oids are provided :param globally: if `!False` register the typecasters only on *conn_or_curs*, otherwise register them globally :param loads: the function used to parse the data into a Python object. If `!None` use `!json.loads()`, where `!json` is the module chosen according to the Python version (see above) :param oid: the OID of the :sql:`json` type if known; If not, it will be queried on *conn_or_curs* :param array_oid: the OID of the :sql:`json[]` array type if known; if not, it will be queried on *conn_or_curs* :param name: the name of the data type to look for in *conn_or_curs* The connection or cursor passed to the function will be used to query the database and look for the OID of the :sql:`json` type (or an alternative type if *name* if provided). No query is performed if *oid* and *array_oid* are provided. Raise `~psycopg2.ProgrammingError` if the type is not found. N)�loads�name)�_get_json_oids�_create_json_typecasters�upperr )� conn_or_curs�globallyr# �oid� array_oidr$ �JSON� JSONARRAYr r r � register_json] s r. c C s t | ||ttd�S )a{ Create and register :sql:`json` typecasters for PostgreSQL 9.2 and following. Since PostgreSQL 9.2 :sql:`json` is a builtin type, hence its oid is known and fixed. This function allows specifying a customized *loads* function for the default :sql:`json` type without querying the database. All the parameters have the same meaning of `register_json()`. )r( r) r# r* r+ )r. �JSON_OID� JSONARRAY_OID)r( r) r# r r r �register_default_json� s r1 c C s t | ||ttdd�S )a^ Create and register :sql:`jsonb` typecasters for PostgreSQL 9.4 and following. As in `register_default_json()`, the function allows to register a customized *loads* function for the :sql:`jsonb` type at its known oid for PostgreSQL 9.4 and following versions. All the parameters have the same meaning of `register_json()`. Zjsonb)r( r) r# r* r+ r$ )r. � JSONB_OID�JSONBARRAY_OID)r( r) r# r r r �register_default_jsonb� s r4 r, c sP � dkrt j� � fdd�}t| f||�}|dk rDt|fd| |�}nd}||fS )z&Create typecasters for json data type.Nc s | d kr d S � | �S )Nr )r Zcur)r# r r � typecast_json� s z/_create_json_typecasters.<locals>.typecast_jsonz%sARRAY)r r# r r )r* r+ r# r$ r5 r, r- r )r# r r& � s r& c C s� ddl m} ddlm} || �\}}|j}|jjdkr:dp<d}|�d| |f� |�� }||krn|j sn|� � |s�|� d| ��|S ) Nr )�STATUS_IN_TRANSACTION)�_solve_conn_cursi�9 �typarrayZNULLz6SELECT t.oid, %s FROM pg_type t WHERE t.typname = %%s;z%s data type not found) Zpsycopg2.extensionsr6 Zpsycopg2.extrasr7 �status�infoZserver_version�executeZfetchoneZ autocommitZrollbackZProgrammingError) r( r$ r6 r7 r ZcursZ conn_statusr8 �rr r r r% � s r% )NFNNNr )NFN)NFN)Nr, )r )r" r Zpsycopg2._psycopgr r r r r Zpsycopg2.compatr r/ r0 r2 r3 �objectr r. r1 r4 r& r% r r r r �<module> s / &