import glob

def _search(query='', file_name=''):
    with open(file_name, 'r') as f:
        for line in f.readlines():
            if query in line:
                return True
        return False

def get_file_list(path=''):
    file_list = []
    for f in glob.iglob(path):
        file_list.append(f)
    return file_list

def search_files(query='', path=''):
    results = []
    path = path + '/' if path[-1] != '/' else path
    path = path+'*.xml' if path[-1] != '*' else path
    print(f"Searching: {path}")
    for f in glob.iglob(path):
        print(f"File name: {f}")
        res = _search(query=query, file_name=f)
        if res:
            results.append(f)
    return results

def main():
    path = 'test-files/*'
    files = search_files('a',path=path)
    print(len(files))

if __name__ == '__main__':
    main()
