

class DocParserTestHelper:

    def __init__(self, tester):
        self.tester = tester
    doc_parser_url = "/doc_parser/docx/parse"

    def docx_parse_helper(self, institute_id, questions_docx, answers_docx=None, skip_images=False):
        request = {'context_institute_id': institute_id,
                   'questions_docx': questions_docx}

        if answers_docx:
            request['answers_docx'] = answers_docx

        if skip_images:
            request['skip_images'] = skip_images

        print("request--->", request)
        (response_data, status_code, error_message) = self.tester.post_with_token(self.doc_parser_url,
                                                                                  request)

        return response_data, status_code, error_message
