请问用Python怎么做

编写程序，让用户输入一段字符串，判断其中是否包含字符串"python"，若包含则替换为"PYTHON"后输出新字符串，否则输出原字符串。

import re

target = re.compile('python')
input_chart = input('please input something：')
m = target.search(input_chart)
if m:
    print('PYTHON')
else:
    print(input_chart)