import re
p = re.compile(r'^[ \t]*((?:class|def)[ \t].*)', re.MULTILINE)
s = "test.py \n\nimport os\nclass test():\n\n def __init__(self):\n pass\n\n def add(num1, num2):\n return num1+num2"
print(p.findall(s))
aW1wb3J0IHJlCnAgPSByZS5jb21waWxlKHInXlsgXHRdKigoPzpjbGFzc3xkZWYpWyBcdF0uKiknLCByZS5NVUxUSUxJTkUpCnMgPSAidGVzdC5weSBcblxuaW1wb3J0IG9zXG5jbGFzcyB0ZXN0KCk6XG5cbiAgICBkZWYgX19pbml0X18oc2VsZik6XG4gICAgICAgIHBhc3NcblxuICAgIGRlZiBhZGQobnVtMSwgbnVtMik6XG4gICAgICAgIHJldHVybiBudW0xK251bTIiCnByaW50KHAuZmluZGFsbChzKSk=