summaryrefslogtreecommitdiff
path: root/support/inkscape/__main__.py
blob: 29ac74517acff2fe34186cf3a219bc794247af33 (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)