aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/hydro_roll/models/utils.py
blob: 9e27307cf16df9c61a14fc5f3617838dc1d7aacd (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import difflib
import pickle


def find_max_similarity(input_string, string_list):
    max_similarity = 0
    max_string = ""

    for string in string_list:
        similarity = difflib.SequenceMatcher(None, input_string, string).quick_ratio()
        if similarity > max_similarity:
            max_similarity = similarity
            max_string = string

    return max_string, max_similarity