Python variable in re.match

Question

Я пытаюсь написать функцию, которая принимает ключ (между прочим) и возвращает слово после этого ключа в файле. Код ниже работает, но только в том случае, если ключ является первой фразой в файле. Может ли кто-нибудь указать, где я ошибаюсь?

def findmatch(key, split_by, tempsl, filename, temp): 
    rx=r''+key+'(.*)' 
    f = open(tempsl + filename, 'r', encoding='windows-1252') 
    for eachline in f: 
     string=re.match(rx, eachline) 
     if string: 
      return (string.group().split(' ')[split_by]) 
     else: 
      return "didn't work" 

Answer 1

Вы закончите цикл после первой итерации, поскольку

if string: 
    return (string.group().split(' ')[split_by]) 
else: 
    return "didn't work" 

всегда разорвать петлю. Он вернет некоторый результат, только если у вас есть ключевое слово в первой строке. Поэтому я предлагаю следующее:

for eachline in f: 
    string=re.match(rx, eachline) 
    if string: 
     return (string.group().split(' ')[split_by]) 
else: # else statemant is a part of for loop (moved to the left) 
    return "didn't work" 

и попробовать это:

m = re.search('(?<=' + key + ')\w+', eachline) 
m.group(0)