тут можно немного очевидней сделать, без re.sub