def getClassifies(__messages): try: __json = json.loads(__messages) __idList = list() __msgList = list() for x in __json['messages']: __idList.append(x['id']) __msgList.append(x['message']) __response = list() __predict = predictNew(__clf, __msgList, __groups) logger.debug("Predictions on new comments: ") logger.debug("--------------------------------------------------------------") for i in range(len(__msgList)): __response.append({'id': __idList[i], 'result': __predict[i]}) __model = Data(__idList[i], __msgList[i], __predict[i]) session.add(__model) logger.debug("Input Messages: %s ==> %s", __msgList[i], __predict[i]) logger.debug("--------------------------------------------------------------") session.commit() __data = { 'results': __response } return __data except (ValueError, KeyError, TypeError): logger.error("Exception: %s, %s, %s", ValueError, KeyError, TypeError) session.rollback()
def getClassify(__id, __message): try: __msgList = [__message] __predict = predictNew(__clf, __msgList, __groups) logger.debug("Predictions on new comments: ") logger.debug("--------------------------------------------------------------") logger.debug("Input Messages: %s ==> %s", __msgList[0], __predict[0]) logger.debug("--------------------------------------------------------------") __model = Data(__id, __msgList[0], __predict[0]) session.add(__model) session.commit() __data = { 'id': __id, 'result': __predict[0] } return __data except (ValueError, KeyError, TypeError): logger.error("Exception: %s, %s, %s", ValueError, KeyError, TypeError) session.rollback()