U ڲgQ@sLdZddlmZddlmZddlmZddlmZGdddZeZ d S) z psycopg capabilities objects ) annotations)pq)_cmodule)NotSupportedErrorc@seZdZdZddddZd"dddd d Zd#dddd d Zd$dddd dZd%dddddZd&dddddZ d'dddddZ d(dddddZ dddddddZ ddddddZ dddd Zd!S)) Capabilitiesz\ An object to check if a feature is supported by the libpq available on the client. None)returncCs i|_dS)N)_cache)selfr 9/tmp/pip-unpacked-wheel-b_ea6rx_/psycopg/_capabilities.py__init__szCapabilities.__init__Fbool)checkr cCs|jdd|dS)z~Check if the `PGconn.encrypt_password()` method is implemented. The feature requires libpq 10.0 and greater. zpq.PGconn.encrypt_password()ir _has_featurer rr r r has_encrypt_passwordsz!Capabilities.has_encrypt_passwordcCs|jdd|dS)zCheck if the `ConnectionInfo.hostaddr` attribute is implemented. The feature requires libpq 12.0 and greater. zConnection.info.hostaddrirrrr r r has_hostaddrszCapabilities.has_hostaddrcCs|jdd|dS)z~Check if the :ref:`pipeline mode ` is supported. The feature requires libpq 14.0 and greater. zConnection.pipeline()"rrrr r r has_pipeline$szCapabilities.has_pipelinecCs|jdd|dS)zCheck if the `pq.PGconn.set_trace_flags()` method is implemented. The feature requires libpq 14.0 and greater. zPGconn.set_trace_flags()rrrrr r r has_set_trace_flags+sz Capabilities.has_set_trace_flagscCs|jdd|dS)z}Check if the `Connection.cancel_safe()` method is implemented. The feature requires libpq 17.0 and greater. zConnection.cancel_safe()rrrr r r has_cancel_safe2szCapabilities.has_cancel_safecCs|jdd|dS)zCheck if `Cursor.stream()` can handle a `size` parameter value greater than 1 to retrieve results by chunks. The feature requires libpq 17.0 and greater. z4Cursor.stream() with 'size' parameter greater than 1rrrrr r r has_stream_chunked9s zCapabilities.has_stream_chunkedcCs|jdd|dS)zCheck if the `pq.PGconn.send_closed_prepared()` method is implemented. The feature requires libpq 17.0 and greater. zPGconn.send_close_prepared()rrrrr r r has_send_close_preparedCsz$Capabilities.has_send_close_preparedstrint)feature want_versionrr cCsJ||jkr|j|}n|||}||j|<|s4dS|rBt|ndSdS)z Check is a version is supported. If `check` is true, raise an exception with an explicative message explaining why the feature is not supported. The expletive messages, are left to the user. TFN)r _get_unsupported_messager)r r r!rmsgr r r rJs     zCapabilities._has_feature)r r!r c Cs~t|kr>d|d|dttdt|d Stj|krvd|dtjdttjdt|d SdSd S) z Return a descriptinve message to describe why a feature is unsupported. Return an empty string if the feature is supported. z the feature 'z<' is not available: the client libpq version (imported from z) is z%; the feature requires libpq version z or newerz,' is not available: you are using a psycopg[z)] libpq wrapper built with libpq version N)rversion _libpq_sourceZversion_prettyZ__build_version____impl__)r r r!r r r r"`s 0 ,z%Capabilities._get_unsupported_messagecCs&tjdkrtjpd}d|SdSdS)z5Return a string reporting where the libpq comes from.binaryunknownz$the psycopg[binary] package version zsystem librariesN)rr'r __version__)r r%r r r r&zs   zCapabilities._libpq_sourceN)F)F)F)F)F)F)F)__name__ __module__ __qualname____doc__rrrrrrrrrr"r&r r r r rs rN) r. __future__rr$rrerrorsrrZ capabilitiesr r r r s     v