Server : Apache/2.4.6 (CentOS) OpenSSL/1.0.2k-fips PHP/7.4.20 System : Linux st2.domain.com 3.10.0-1127.10.1.el7.x86_64 #1 SMP Wed Jun 3 14:28:03 UTC 2020 x86_64 User : apache ( 48) PHP Version : 7.4.20 Disable Function : NONE Directory : /usr/lib/python2.7/site-packages/libpasteurize/fixes/ |
u"""
Fixer for os.getcwd() -> os.getcwdu().
Also warns about "from os import getcwd", suggesting the above form.
"""
from lib2to3 import fixer_base
from lib2to3.fixer_util import Name
class FixGetcwd(fixer_base.BaseFix):
PATTERN = u"""
power< 'os' trailer< dot='.' name='getcwd' > any* >
|
import_from< 'from' 'os' 'import' bad='getcwd' >
"""
def transform(self, node, results):
if u"name" in results:
name = results[u"name"]
name.replace(Name(u"getcwdu", prefix=name.prefix))
elif u"bad" in results:
# Can't convert to getcwdu and then expect to catch every use.
self.cannot_convert(node, u"import os, use os.getcwd() instead.")
return
else:
raise ValueError(u"For some reason, the pattern matcher failed.")