U Lk7gE @s~ddlZddlmZmZmZddlmZzddlmZWn e k rXddl mZYnXzddl m Z Wne k rdZ YnXddl Z ddlmZddd d d d d ddg ZdZdZdZdZiZiZiZejZGdd d eZddZddddedddddf ddZddddedddddf ddZefdd Zdd Z edfdd Z!ddZ"d d!Z#edfd"dZ$defd#d Z%e!Z&e$Z'dS)$N) MongoClientReadPreference uri_parser)_UUID_REPRESENTATIONS) _check_name) DriverInfo)PYMONGO_VERSIONDEFAULT_CONNECTION_NAMEDEFAULT_DATABASE_NAMEConnectionFailureconnect disconnectdisconnect_allget_connectionget_dbregister_connectiondefaulttest localhostiic@seZdZdZdS)r zError raised when the database connection can't be established or when a connection with a requested alias can't be retrieved. N)__name__ __module__ __qualname____doc__rr:/tmp/pip-unpacked-wheel-n1etwkgt/mongoengine/connection.pyr -scCs,t|tstdtn|dkr(t|dS)zqCheck if a database name is valid. This functionality is copied from pymongo Database class constructor. zname must be an instance of %sz $externalN) isinstancestr TypeErrorrnamerrr_check_db_name5s r c  Ksb|p |p t|pt|pt|||||| d } t| d| d} t| trL| g} g} | D]}|dsn| drvtdd|krt |}| ||d}|r|| d<d D]}||r||| |<q|d }d |kr|d | d <d |kr|d | d<d|kr|d| d<d|krt j t jt jt jt jf}|d}t|trN|}|D].}|j|ksr|j|krR|j}qqRd|kr||dd| d<n || d<d|kr|d| d<d|krddtD}||d| d<qT| |qT| | d<| dd| dddd| D}d|krTd| krTtdtd| d <| | | S)!aGet the connection settings as a dict :param db: the name of the database to use, for compatibility with connect :param name: the name of the specific database to use :param host: the host name of the: program: `mongod` instance to connect to :param port: the port that the: program: `mongod` instance is running on :param read_preference: The read preference for the collection :param username: username to authenticate with :param password: password to authenticate with :param authentication_source: database to authenticate against :param authentication_mechanism: database authentication mechanisms. By default, use SCRAM-SHA-1 with MongoDB 3.0 and later, MONGODB-CR (MongoDB Challenge Response protocol) for older servers. :param mongo_client_class: using alternative connection client other than pymongo.MongoClient, e.g. mongomock, montydb, that provides pymongo alike interface but not necessarily for connecting to a real mongo instance. :param kwargs: ad-hoc parameters to be passed into the pymongo driver, for example maxpoolsize, tz_aware, etc. See the documentation for pymongo's `MongoClient` for a full list. ) rhostportread_preferenceusernamepasswordauthentication_sourceauthentication_mechanismauthmechanismpropertiesrr!z mongomock://Zis_mockzUse of mongomock:// URI or 'is_mock' were removed in favor of 'mongo_client_class=mongomock.MongoClient'. Check the CHANGELOG for more infoz://database)r#r$r%optionsZ replicasetZ replicaSetZ authsourcer&Z authmechanismr'ZreadpreferenceZreadpreferencetags)Ztag_setsr#r(ZuuidrepresentationcSsi|]\}}||qSrr.0kvrrr sz,_get_connection_settings..ZslavesNZis_slavecSsh|] }|qSr)lower)r,keyrrr sz+_get_connection_settings..a?No uuidRepresentation is specified! Falling back to 'pythonLegacy' which is the default for pymongo 3.x. For compatibility with other MongoDB drivers this should be specified as 'standard' or '{java,csharp}Legacy' to work with older drivers in those languages. This will be changed to 'unspecified' in a future release.Z pythonLegacyZuuidRepresentation)r DEFAULT_HOST DEFAULT_PORTr rr startswithget Exceptionr parse_uriappendrZNEARESTPRIMARYZPRIMARY_PREFERREDZ SECONDARYZSECONDARY_PREFERREDr0rmode __class__ritemspopkeyswarningswarnDeprecationWarningupdate)dbrr!r"r#r$r%r&r'r(kwargs conn_settingsZ conn_hostZresolved_hostsentityZuri_dictr)paramZ uri_optionsZread_preferencesZ read_pf_modeZ preferenceZ ReadPrefClassZREV_UUID_REPRESENTATIONSr?rrr_get_connection_settings?s"                             rIc Ks0tf||||||||| | d | } | t|<dS)a#Register the connection settings. :param alias: the name that will be used to refer to this connection throughout MongoEngine :param db: the name of the database to use, for compatibility with connect :param name: the name of the specific database to use :param host: the host name of the: program: `mongod` instance to connect to :param port: the port that the: program: `mongod` instance is running on :param read_preference: The read preference for the collection :param username: username to authenticate with :param password: password to authenticate with :param authentication_source: database to authenticate against :param authentication_mechanism: database authentication mechanisms. By default, use SCRAM-SHA-1 with MongoDB 3.0 and later, MONGODB-CR (MongoDB Challenge Response protocol) for older servers. :param mongo_client_class: using alternative connection client other than pymongo.MongoClient, e.g. mongomock, montydb, that provides pymongo alike interface but not necessarily for connecting to a real mongo instance. :param kwargs: ad-hoc parameters to be passed into the pymongo driver, for example maxpoolsize, tz_aware, etc. See the documentation for pymongo's `MongoClient` for a full list. ) rDrr!r"r#r$r%r&r'r(N)rI_connection_settings) aliasrDrr!r"r#r$r%r&r'r(rErFrrrrs#  csddlm}ddlm}t|drJtfddtDrJ|t krz||t D]}t ||r\| q\t |=|t krt |=dS)z(Close the connection with a given alias.r)Document)_get_documents_by_dbNc3s|]}|k VqdSNr)r,c connectionrr szdisconnect..) mongoenginerLZmongoengine.base.commonrM _connectionsr>allvaluesclose_dbsr issubclassZ _disconnectrJ)rKrLrMZdoc_clsrrPrr s     cCsttD] }t|q dS)zClose all registered database.N)listrTr?r )rKrrrr(sFc Cs|r t||tkrt|S|tkrB|tkr2d}nd|}t|dd}t|}||}tdk rz|dtdtj d|kr| d}nt }t |}|r|}nt f||d |}|t|<t|S) z'Return a connection with a given alias.z)You have not defined a default connectionz/Connection with alias "%s" has not been definedcsFtdkrddddddhindhdd d fd d |DS) Nrr$r%r&r'r(Z authSourceZ authMechanism)r&r'cs.i|]&\}}|kr|dk r|||qSrN)r6r+Zirrelevant_fields_setZ rename_fieldsrrr/Ts  z;get_connection.._clean_settings..)rr=Z settings_dictrr]r_clean_settingsCs  z'get_connection.._clean_settingsNZdriverZ MongoEnginemongo_client_class)rKr`)r rTrJr r copyr setdefaultrS __version__r>r_find_existing_connection_create_connection) rK reconnectmsgr_Zraw_conn_settingsrFr`Zexisting_connectionrQrrrr.s>   c KsHz |f|WStk rB}ztd|d|W5d}~XYnXdS)zk Create the new connection for this alias. Raise ConnectionFailure if it can't be established. zCannot connect to database z : N)r7r )rKr`connection_settingserrrrews recCsZddtD}dd}||}|D].\}}||}||kr&t|r&t|Sq&dS)a, Check if an existing connection could be reused Iterate over all of the connection settings and if an existing connection with the same parameters is suitable, return it :param connection_settings: the settings of the new connection :return: An existing connection or None css|]\}}||fVqdSrN)ra)r,db_aliassettingsrrrrRsz,_find_existing_connection..cSsdd|DS)NcSsi|]\}}|dkr||qSrrr+rrrr/szF_find_existing_connection.._clean_settings..)r=r^rrrr_sz2_find_existing_connection.._clean_settingsN)rJr=rTr6)rhZconnection_settings_bisr_Zcleaned_conn_settingsrjZdb_conn_settingsrrrrds  rdcCs|r t||tkrt|}t|}||d}tdkr|dr|dsT|ddkr|ddkrd|d i}|ddk r|d|d <|j|d|df||t|<t|S) Nrr[r$r%r'z MONGODB-X509r(sourcer&Z mechanism)r rXrrJrZ authenticate)rKrfconnrFrDZ auth_kwargsrrrrs6      cKsN|tkr8t|}t|f|}||krFd|}t|nt||f|t|S)alConnect to the database specified by the 'db' argument. Connection settings may be provided here as well if the database is not running on the default port on localhost. If authentication is needed, provide username and password arguments as well. Multiple databases are supported by using aliases. Provide a separate `alias` to connect to a different instance of: program: `mongod`. In order to replace a connection identified by a given alias, you'll need to call ``disconnect`` first See the docstring for `register_connection` for more details about all supported kwargs. zUA different connection with alias `{}` was already registered. Use disconnect() first)rTrJrIformatr rr)rDrKrEZprev_conn_settingZnew_conn_settingserr_msgrrrr s  )(r@ZpymongorrrZpymongo.commonrZpymongo.database_sharedr ImportErrorZpymongo.databaseZpymongo.driver_inforrSZmongoengine.pymongo_supportr__all__r r r3r4rJrTrXr:ZREAD_PREFERENCEr7r r rIrr rrrerdrr Z_get_connectionZ_get_dbrrrrs       3 I !