def make_me_magic(write):
""" Main script.
Here we find templates, replace old-style url tags and add future import where necessary.
"""
# Search for files with appropriate extensions.
log.info('Files needing modification:')
found = False
for file_path in find_files(os.walk(CURRENT_PATH)):
with open(file_path, 'r+') as t_file:
file_content = t_file.read()
# Checking for presence of old-style tags and absence of load url from future
if has_deprecated_tag(file_content):
found = True
log.info(file_path.replace(CURRENT_PATH + '/', ''))
new_content = parse_file(file_content)
if write:
t_file.seek(0)
t_file.write(new_content)
log.info(' File updated')
if not found:
log.info('All files are up to date. Congrats!')
else:
if not write:
log.info('No actual changes made. Run future_url --write to fix files right now.')
def make_me_magic(write):
""" Main script.
Here we find templates, replace old-style url tags and add future import where necessary.
"""
# Search for files with appropriate extensions.
log.info('Files needing modification:')
found = False
for file_path in find_files(os.walk(CURRENT_PATH)):
with open(file_path, 'r+') as t_file:
file_content = t_file.read()
# Checking for presence of old-style tags and absence of load url from future
if has_deprecated_tag(file_content):
found = True
log.info(file_path.replace(CURRENT_PATH + '/', ''))
new_content = parse_file(file_content)
if write:
t_file.seek(0)
t_file.write(new_content)
log.info(' File updated')
if not found:
log.info('All files are up to date. Congrats!')
else:
if not write:
log.info(
'No actual changes made. Run future_url --write to fix files right now.'
)
def assertTemplateFixed(self, template_name):
base = os.path.abspath(os.path.dirname(__file__))
input_filepath = os.path.join(base, 'examples', template_name + '.html')
expected_filepath = os.path.join(base, 'expected', template_name + '.fixed.html')
output = parse_file(codecs.open(input_filepath, 'r', 'utf-8').read())
expected_output = codecs.open(expected_filepath, 'r', 'utf-8').read()
self.assertEqual(output, expected_output)
def assertTemplateFixed(self, template_name):
base = os.path.abspath(os.path.dirname(__file__))
input_filepath = os.path.join(base, 'examples',
template_name + '.html')
expected_filepath = os.path.join(base, 'expected',
template_name + '.fixed.html')
output = parse_file(codecs.open(input_filepath, 'r', 'utf-8').read())
expected_output = codecs.open(expected_filepath, 'r', 'utf-8').read()
self.assertEqual(output, expected_output)