î 5dÝVœã@s2dZddlZddlmZmZddlmZmZmZejdd…dkrlddl Z n+yddl Z Wne k r–dZ YnXdZ dZ d Zd ZGd d „d eƒZdd dddddd„Zdd ddd„Zdd ddd„Zdddd„Zddd„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éééréÇiÚißc@sseZdZdZddd„Zdd„Zdd„Zd d „Zej dkrcd d „Z n dd „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 (`!simplejson` for Python < 2.6; `~psycopg2.extensions.ISQLQuote.getquoted()` will raise `!ImportError` if the module is not available). NcCsI||_|dk r!||_n$tdk r<tj|_n d|_dS)N)ÚadaptedÚ_dumpsÚjsonÚdumps)Úselfr r©rú2/tmp/pip-build-0jahl3lb/psycopg2/psycopg2/_json.pyÚ__init__Bs     z Json.__init__cCs|tkr|SdS)N)r)rÚprotorrrÚ __conform__Ls zJson.__conform__cCs/|j}|dk r||ƒStdƒ‚dS)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. Nz>json module not available: you should provide a dumps function)r Ú ImportError)rÚobjrrrrrPs    z Json.dumpscCs"|j|jƒ}t|ƒjƒS)N)rr rÚ getquoted)rÚsrrrr_szJson.getquotedécCs |jƒS)N)r)rrrrÚ__str__dsz Json.__str__cCs|jƒjddƒS)NÚasciiÚreplace)rÚdecode)rrrrrgs)r) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrÚsysÚ version_inforrrrrr 6s    r FrcCs˜|dkr$t||ƒ\}}nt||d|d|jƒƒ\}}t|| r^|padƒ|dk rŽt|| r„|p‡dƒn||fS)aCreate 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Ú JSONARRAYrrrÚ register_jsonls $ r0c Cs%td|d|d|dtdtƒ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-)r0ÚJSON_OIDÚ JSONARRAY_OID)r*r+r%rrrÚregister_default_json’s r3c Cs+td|d|d|dtdtddƒ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()`. r*r+r%r,r-r&Zjsonb)r0Ú JSONB_OIDÚJSONBARRAY_OID)r*r+r%rrrÚregister_default_jsonbžs r6r.cs’ˆdkr3tdkr'tdƒ‚q3tj‰n‡fdd†}t|f||ƒ}|dk r‚t|fd||ƒ}nd}||fS)z&Create typecasters for json data type.Nzno json module availablecs|dkrdSˆ|ƒS)Nr)rÚcur)r%rrÚ typecast_json²s z/_create_json_typecasters..typecast_jsonz%sARRAY)rrr%rr)r,r-r%r&r8r.r/r)r%rr(ªs    r(c Cs¼ddlm}ddlm}||ƒ\}}|j}|jdkrPdpSd}|jd||fƒ|jƒ}||krœ|j rœ|j ƒn|s¸|j d|ƒ‚n|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.extensionsr9Zpsycopg2.extrasr:ÚstatusZserver_versionÚexecuteZfetchoneZ autocommitÚrollbackZProgrammingError) r*r&r9r:ÚconnZcursZ conn_statusr;Úrrrrr'¿s   r')rr)r"r#Zpsycopg2._psycopgrrrrrr$rZ simplejsonrr1r2r4r5Úobjectr r0r3r6r(r'rrrrÚs(   6 %