#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
import os
import re
import sys
def replace_line(line, resize_method, n):
# Extract Width and Height values with type 'integer'.
width_data = re.search(r'(\bwidth\b\s*=\s*")(\d+)(pt")',line).groups()
height_data = re.search(r'(\bheight\b\s*=\s*")(\d+)(pt")',line).groups()
original_width, original_height = int(width_data[1]), int(height_data[1])
if resize_method == "multiply":
final_width, final_height = original_width*n, original_height*n
elif resize_method == "1000x1000":
final_width, final_height = 1000, 1000
else:
print "Invalid resize method specified\nReturning line as is"
print '------------------------------'
return line
# Replace width
line = line.replace("".join(width_data), width_data[0]+str(final_width)+width_data[2])
# Replace height
line = line.replace("".join(height_data), height_data[0]+str(final_height)+height_data[2])
return line
def process_files(lig_dir, output_dir, resize_method, n):
try:
os.mkdir(output_dir)
except:
pass
dirlist = os.listdir(lig_dir)
svglist = []
for filename in dirlist:
if re.search(r'.svg$',filename):
svglist.append(filename)
print '------------------------------'
print 'List of files to process:\n', svglist
print '------------------------------'
for svgfile in svglist:
input_file = open(lig_dir+os.sep+svgfile,"r")
svglines = input_file.readlines()
for i in range(len(svglines)):
if re.search(r'<svg.+>',svglines[i]):
svglines[i] = replace_line(svglines[i],resize_method,n)
output_file = open(output_dir+os.sep+svgfile, "w")
output_file.writelines(svglines)
output_file.close()
print 'Generated', output_dir+os.sep+svgfile
return True
if __name__=='__main__':
if len(sys.argv) < 4 or len(sys.argv) > 5:
print 'Usage: python {0} directory_containing_SVGs output_directory resize_method n_(optional)'.format(sys.argv[0])
else:
lig_dir = sys.argv[1]
output_dir = sys.argv[2]
resize_method = sys.argv[3]
try:
n = sys.argv[4]
except IndexError:
print 'Warning: n is not specified. Using default value of n=3.'
print '(Will not be used if resize_method is not multiply)'
n = 3
# n from sys.argv is type 'string'. Must be converted to type 'integer' first.
try:
n = int(n)
except ValueError:
print 'Error: n must be a number'
sys.exit(1)
print 'n =',n
result = process_files(lig_dir, output_dir, resize_method, n)
if result:
print '=============================='
print "Finished processing files."
print '=============================='
