コード例 #1
0
    def test_open_calais_when_unsupported_language(self, mock_post):
        response = issue_open_calais_request(keyword_arg=unsupported_language_keyword)
        expected_result = default_not_found_result(resolution_algorithm=ResolutionAlgo.OPEN_CALAIS,
                                                   keyword=get_correct_keyword(unsupported_language_keyword))

        self.assertEqual(response, expected_result)
        pass
コード例 #2
0
    def test_open_calais_when_found(self, mock_post):
        response = issue_open_calais_request(keyword_arg=found_keyword)

        self.assertEqual(len(response), 1)

        result = response[0]
        self.assertEqual(result['perm_id'], 4295907168)
        self.assertEqual(result['confidence'], '1.0')
        self.assertEqual(result['org_name'], 'MICROSOFT CORPORATION')
        pass
コード例 #3
0
def search_tr(keyword):
    entity_match_result = issue_entity_match_request(keyword_arg=keyword)
    open_calais_result = issue_open_calais_request(keyword_arg=keyword)

    if not entity_match_result and not open_calais_result:
        return None

    resolved_entity = ResolvedItem(
        keyword=keyword,
        entity_match=entity_match_result,
        open_calais=open_calais_result,
        manual_override=__get_previous_manual_override(keyword))

    if resolved_entity is not None:
        dynamo_save_entity(resolved_entity)
        return resolved_entity.__dict__
    else:
        return None
コード例 #4
0
    def test_open_calais_when_api_error(self, mock_post):
        response = issue_open_calais_request(keyword_arg=api_error_keyword)

        self.assertEqual(response, None)
        pass
コード例 #5
0
def __issue_em_oc_request(keyword_orig):
    em_result = issue_entity_match_request(keyword_arg=keyword_orig)
    oc_result = issue_open_calais_request(keyword_arg=keyword_orig,
                                          max_retry=10)

    return em_result, oc_result
コード例 #6
0
force_ers_url = f'{base_url}/resolve/force/keyword'

tr_url = 'https://api.thomsonreuters.com/permid'
tr_api_key = "'hWRwzOxar9YGo0A0BnhqKvxF5s5CpwNS', 'hWRwzOxar9YGo0A0BnhqKvxF5s5CpwNS'"

os.environ['TR_URL'] = tr_url
os.environ['TR_API_KEY'] = tr_api_key

# Lookup
result = requests.get(url=f'{lookup_ers_url}/{quoted_keyword}')
print(f'LOOKUP: {result.status_code}, {result.text}\n')

# Force Resolve
if force_resolve:
    result = requests.post(url=f'{smart_ers_url}/{quoted_keyword}/0')
    if result.status_code is not requests.codes.ok:
        result = requests.post(url=f'{force_ers_url}/{quoted_keyword}')
        print(f'FORCE-RESOLVE: {result.status_code}, {result.text}\n')

# Smart Resolve
result = requests.post(url=f'{smart_ers_url}/{quoted_keyword}/0')
print(f'SMART-RESOLVE: {result.status_code}, {result.text}\n')

# Entity Match
result = issue_entity_match_request(keyword_arg=keyword_arg)
print(f'ENTITY-MATCH: {result}')

# Open Calais
result = issue_open_calais_request(keyword_arg=keyword_arg, max_retry=10)
print(f'OPEN-CALAIS: {result}')