Update cqi package

This commit is contained in:
Patrick Jentsch
2020-03-25 15:39:32 +01:00
parent 6b24184206
commit 0db99e33fe
6 changed files with 222 additions and 179 deletions

View File

@ -421,13 +421,8 @@ class APIClient:
self.port = port
self.socket = socket.socket()
def setup(self):
self.socket.connect((self.host, self.port))
def teardown(self):
self.socket.close()
def ctrl_connect(self, username, password):
self.socket.connect((self.host, self.port))
# INPUT: (STRING username, STRING password)
# OUTPUT: CQI_STATUS_CONNECT_OK, CQI_ERROR_CONNECT_REFUSED
self.__send_WORD(CTRL_CONNECT)
@ -439,7 +434,9 @@ class APIClient:
# INPUT: ()
# OUTPUT: CQI_STATUS_BYE_OK
self.__send_WORD(CTRL_BYE)
return self.__recv_response()
response = self.__recv_response()
self.socket.close()
return response
def ctrl_user_abort(self):
# INPUT: ()
@ -575,7 +572,7 @@ class APIClient:
# INPUT: (STRING attribute)
# OUTPUT: CQI_STATUS_OK
# unload attribute from memory
self.__send_WORD(CL_LEXICON_SIZE)
self.__send_WORD(CL_DROP_ATTRIBUTE)
self.__send_STRING(attribute)
return self.__recv_response()
@ -589,7 +586,7 @@ class APIClient:
# OUTPUT: CQI_DATA_INT_LIST
# returns -1 for every string in <strings> that is not found in the
# lexicon
self.__send_WORD(CL_LEXICON_SIZE)
self.__send_WORD(CL_STR2ID)
self.__send_STRING(attribute)
self.__send_STRING_LIST(strings)
return self.__recv_response()