Python:搜索文本文件

我有一个以下格式的文本文件:

XXXX Testing123 
    YYYY hellow 
    ZZZZ worldd 
AAAA Testing456 
    BBBB heyyy 
    YYYY YoYo 
AAAA Testing789 
    CCCC hezzz 
    YYYY hellow 

这是我的文本搜索代码:

for line in open('test.txt') 
     a = "XXXX" 
     b = "\t" + "YYYY" 
     if a in line: 
        print line 
     if b in line: 
        print line 

以上打印出来:

XXXX Testing123 
   YYYY YoYo 

但我想要的是:

XXXX Testing123 
  YYYY hellow 

知道如何在 Python 中搜索它吗？

基本上我需要先搜索XXXX，后面的字段YYYY应该在这个部分下。如果在 XXXX 下找不到 YYYY，则返回 not found 而不是继续查找。

请您参考如下方法:

您可以制作一个标志，显示您是否在“XXXX”部分。如果在，则使flag=1，否则使flag=0。此外，您需要区分两种线。

a = "XXXX" 
b = "\t" + "YYYY" 
flag = 0 
for line in open('test.txt'): 
    if line[0] == '\t': 
        if flag and b in line: 
            print line, 
    else:         
        if a in line: 
            flag = 1 
            print line, 
        else: 
            flag = 0