def main(argv):
input_dir = argv[0]
output_dir = argv[1]
try:
windows_encoding = argv[2] is not None
except IndexError:
windows_encoding = False
print("Input directory: " + input_dir)
print("Output directory: " + output_dir)
helpers.copy_files(input_dir, output_dir)
print("Copied files to output directory.")
try:
for f in helpers.get_html_files(output_dir):
print("Editing file: " + os.path.basename(f))
helpers.edit_src_and_href_paths(f)
helpers.add_header_markup(f)
helpers.remove_title_tag(f)
helpers.update_file_name(f) # Update the file name last
except Exception as e:
print(e)
print(
'Batch process failed. Deleting the contents of the output directory.'
)
for filename in os.listdir(output_dir):
filepath = os.path.join(output_dir, filename)
try:
shutil.rmtree(filepath)
except OSError:
os.remove(filepath)
if windows_encoding:
print("Encoding to Windows-1252...")
for filename in helpers.get_html_files(output_dir):
with io.open(filename, mode="r", encoding="utf8") as fd:
content = fd.read()
with io.open(filename,
mode="w",
encoding="cp1252",
errors='replace') as fd:
fd.write(content)
print('Done.')
exit()
def test_add_header_markup():
clear_test_target()
helpers.copy_files(input_dir, output_dir)
for f in helpers.get_html_files(output_dir):
helpers.add_header_markup(f)
def test_get_html_files():
clear_test_target()
helpers.copy_files(input_dir, output_dir)
print(helpers.get_html_files(output_dir))
def test_edit_src_and_href_paths():
clear_test_target()
helpers.copy_files(input_dir, output_dir)
for f in helpers.get_html_files(output_dir):
helpers.edit_src_and_href_paths(f)
def test_update_file_names():
clear_test_target()
helpers.copy_files(input_dir, output_dir)
for f in helpers.get_html_files(output_dir):
helpers.update_file_name(f)