summaryrefslogtreecommitdiff
path: root/support/inkscape/__main__.py
blob: 78ffb55884398603a9fa346b1afd1504a0c14591 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import os, shutil
from lib import util
from . import effect, inkscape


def _unfuck_svg_document(temp_svg_path):
	"""
	Unfucks an SVG document so is can be processed by the better_dxf_export plugin (or what's left of it).
	"""
	
	command_line = inkscape.InkscapeCommandLine(temp_svg_path)
	layers = command_line.layers
	
	command_line.apply_to_document('LayerUnlockAll', 'LayerShowAll')
	
	layer_copies = []
	
	for i in layers:
		layer_copy = command_line.duplicate_layer(i)
		layer_copies.append(layer_copy)
		
		command_line.apply_to_layer_content(layer_copy, 'ObjectToPath')
		command_line.apply_to_layer_content(layer_copy, 'SelectionUnGroup')
		
		if not i.use_paths:
			command_line.apply_to_layer_content(layer_copy, 'StrokeToPath')
			command_line.apply_to_layer_content(layer_copy, 'SelectionUnion')
	
	for original, copy in zip(layers, layer_copies):
		command_line.clear_layer(original)
		command_line.move_content(copy, original)
		command_line.delete_layer(copy)
	
	command_line.apply_to_document('FileSave', 'FileClose', 'FileQuit')
	
	command_line.run()


@util.main
def main(in_path, out_path):
	try:
		_, out_suffix = os.path.splitext(out_path)
		
		effect.ExportEffect.check_document_units(in_path)
		
		with util.TemporaryDirectory() as temp_dir:
			temp_svg_path = os.path.join(temp_dir, os.path.basename(in_path))
			
			shutil.copyfile(in_path, temp_svg_path)
			
			_unfuck_svg_document(temp_svg_path)
			
			export_effect = effect.ExportEffect()
			export_effect.affect(args = [temp_svg_path], output = False)
			
		with open(out_path, 'w') as file:
			if out_suffix == '.dxf':
				export_effect.write_dxf(file)
			elif out_suffix == '.asy':
				export_effect.write_asy(file)
			else:
				raise Exception('Unknown file type: {}'.format(out_suffix))
	except util.UserError as e:
		raise util.UserError('While processing {}: {}', in_path, e)