仅检查字符串的var的前XX行数

Question

嘿，我得到了以下代码：

import urllib

source = urllib.urlopen('WEBPAGE').read()
if 'STRING TO LOOK FOR' in source:
    print 'MESSAGE'
else:
    print 'else MESSAGE'

检查特定字符串的网页源代码，有没有一种方法只有第一个让代码的20行代码被搜索到字符串？如果字符串不在那里我得到别消息？

Answer 1

将if 'STRING TO LOOK FOR' in source:更改为

if 'STRING TO LOOK FOR' in '\n'.join(source.split('\n', 20)[:20]):

将源分成线，然后重新组合前20行。