def _rewrite_env_path(filename, new_env_path):
with open(filename, 'rb') as f:
lines = f.read().decode('utf-8').splitlines()
script = relative_script([new_env_path] + lines[1:])
with open(filename, 'wb') as f:
f.write('\n'.join(script).encode('utf-8'))
def _rewrite_shebang(filename, new_shebang):
"""Replace the first line of the file with the new shebang"""
with open(filename, 'rb') as f:
lines = f.read().decode('utf-8').splitlines()
script = relative_script([new_shebang] + lines[1:])
with open(filename, 'wb') as f:
f.write('\n'.join(script).encode('utf-8'))
def _rewrite_shebang(filename, new_shebang):
"""Replace the first line of the file with the new shebang"""
with open(filename, 'rb') as f:
lines = f.read().decode('utf-8').splitlines()
script = relative_script([new_shebang] + lines[1:])
with open(filename, 'wb') as f:
f.write('\n'.join(script).encode('utf-8'))
def test_activate_after_future_statements():
"""Should insert activation line after last future statement"""
script = [
'#!/usr/bin/env python', 'from __future__ import with_statement',
'from __future__ import print_function', 'print("Hello, world!")'
]
assert virtualenv.relative_script(script) == [
'#!/usr/bin/env python', 'from __future__ import with_statement',
'from __future__ import print_function', '',
"import os; activate_this=os.path.join(os.path.dirname(os.path.realpath(__file__)), 'activate_this.py'); execfile(activate_this, dict(__file__=activate_this)); del os, activate_this",
'', 'print("Hello, world!")'
]
def test_activate_after_future_statements():
"""Should insert activation line after last future statement"""
script = [
"#!/usr/bin/env python",
"from __future__ import with_statement",
"from __future__ import print_function",
'print("Hello, world!")',
]
assert virtualenv.relative_script(script) == [
"#!/usr/bin/env python",
"from __future__ import with_statement",
"from __future__ import print_function",
"",
"import os; activate_this=os.path.join(os.path.dirname(os.path.realpath(__file__)), 'activate_this.py'); exec(compile(open(activate_this).read(), activate_this, 'exec'), dict(__file__=activate_this)); del os, activate_this",
"",
'print("Hello, world!")',
]
def test_activate_after_future_statements():
"""Should insert activation line after last future statement"""
script = [
"#!/usr/bin/env python",
"from __future__ import with_statement",
"from __future__ import print_function",
'print("Hello, world!")',
]
assert virtualenv.relative_script(script) == [
"#!/usr/bin/env python",
"from __future__ import with_statement",
"from __future__ import print_function",
"",
"import os; "
"activate_this=os.path.join(os.path.dirname(os.path.realpath(__file__)), 'activate_this.py'); "
"exec(compile(open(activate_this).read(), activate_this, 'exec'), dict(__file__=activate_this)); "
"del os, activate_this",
"",
'print("Hello, world!")',
]
