#!/usr/bin/env python3
import distutils
from os import path
DISABLE_BYTECODE = "import sys\nsys.dont_write_bytecode = True\n"
[docs]def ensure_file_contains(target, content):
if path.exists(target):
with open(target) as fp:
if content not in fp.read():
print('{target!r} has unexpected content'.format(target=target))
print('please open the file and add the following:')
print(content)
print("# end")
else:
with open(target, 'w') as fp:
fp.write(content)
if __name__ == '__main__':
try:
site_packages = distutils.sysconfig_get_python_lib()
except AttributeError:
import site
site_packages = site.getsitepackages()[0]
print(site_packages)
target = path.join(site_packages, 'sitecustomize.py')
ensure_file_contains(target, content=DISABLE_BYTECODE)