예제 #1
0
파일: net.py 프로젝트: guyskk/curethinkdb
 def __init__(
     self,
     host,
     port,
     user,
     password,
     ssl,
     json_encoder,
     json_decoder,
     timeout,
     buffer_size=16 * 1024,
 ):
     self.host = host
     self.port = port
     self.ssl = ssl
     self._handshake = HandshakeV1_0(
         json_decoder=json_decoder,
         json_encoder=json_encoder,
         host=host,
         port=port,
         username=user,
         password=password,
     )
     self.json_encoder = json_encoder
     self.json_decoder = json_decoder
     self.timeout = timeout
     self.buffer_size = buffer_size
     self._socket = None
     self._read_buffer = b''
예제 #2
0
 def _get_handshake(self):
     return HandshakeV1_0(json_encoder=self.encoder,
                          json_decoder=self.decoder,
                          host='localhost',
                          port=28015,
                          username='******',
                          password='')
예제 #3
0
 def _get_handshake(self):
     return HandshakeV1_0(
         json_encoder=self.encoder,
         json_decoder=self.decoder,
         host="localhost",
         port=28015,
         username="******",
         password="",
     )
예제 #4
0
def handshake():
    """
    Fixture returning a Handshake object for testing.
    """

    return HandshakeV1_0(
        host="localhost",
        port=28015,
        username=b"admin",
        password=b"",
    )
예제 #5
0
    def __init__(self, conn_type, host, port, db, auth_key, user, password,
                 timeout, ssl, _handshake_version, **kwargs):
        self.db = db

        self.host = host
        try:
            self.port = int(port)
        except ValueError:
            raise ReqlDriverError("Could not convert port %r to an integer." %
                                  port)

        self.connect_timeout = timeout

        self.ssl = ssl

        self._conn_type = conn_type
        self._child_kwargs = kwargs
        self._instance = None
        self._next_token = 0

        if "json_encoder" in kwargs:
            self._json_encoder = kwargs.pop("json_encoder")
        if "json_decoder" in kwargs:
            self._json_decoder = kwargs.pop("json_decoder")

        if auth_key is None and password is None:
            auth_key = password = ""
        elif auth_key is None and password is not None:
            auth_key = password
        elif auth_key is not None and password is None:
            password = auth_key
        else:
            # auth_key is not None and password is not None
            raise ReqlDriverError("`auth_key` and `password` are both set.")

        if _handshake_version == 4:
            raise NotImplementedError("The v0.4 handshake was removed.")

        self.handshake = HandshakeV1_0(
            self._json_decoder(),
            self._json_encoder(),
            self.host,
            self.port,
            user,
            password,
        )