fork download
  1. import re
  2. p = re.compile(r'^[ \t]*((?:class|def)[ \t].*)', re.MULTILINE)
  3. 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"
  4. print(p.findall(s))
Success #stdin #stdout 0.01s 9016KB
stdin
Standard input is empty
stdout
['class test():', 'def __init__(self):', 'def add(num1, num2):']