aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/hydro_roll/models/utils.py
blob: 73e7ba0b05f355cb481b596038baaa1afabf450b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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