summaryrefslogtreecommitdiff
path: root/plugin/mesh_plugin.py
blob: 2208297d8b060126b8dc867388c9163ddf53134b (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
from os import path
import subprocess
import sys

import wx

import pcbnew

def check_requirements(*packages):
    missing = []
    for pkg in packages:
        try:
            __import__(pkg)
        except ImportError:
            missing.append(pkg)

    if missing:
        if 'win' not in sys.platform:
            wx.MessageDialog(None, "Error: Missing python dependencies {}.".format(', '.join(missing)),
                    "Missing Dependencies").ShowModal()
            return False
        
        else:
            msg = 'The following python dependencies are missing:\n\n' + '\n'.join(missing) +\
                  '\n\nShould we go ahead and install these missing dependencies into the plugin directory?'
            dialog = wx.MessageDialog(None, msg, caption='Error: Missing dependencies', style=wx.YES_NO | wx.NO_DEFAULT)
            dialog.SetYesNoLabels("Install missing dependencies", "Cancel")
            if dialog.ShowModal() == wx.ID_YES:
                for dep in packages:
                    proc = subprocess.Popen(
                            "pip install --target deps {} --no-use-pep517 --only-binary :all: --platform win_amd64"\
                                    .format(dep).split(),
                            cwd=path.dirname(__file__),
                            stdout=subprocess.PIPE,
                            stderr=subprocess.PIPE)
                    (stdout, stderr) = proc.communicate()
                    if proc.returncode != 0:
                        wx.MessageDialog(None, "Error installing dependencies:\n\n{}\n{}".format(stdout, stderr),
                                "Installation Error").ShowModal()
                        return False

                sys.path.append(path.abspath(path.join(path.dirname(__file__), 'deps')))

    else:
        return True

class MeshPlugin(pcbnew.ActionPlugin):
    def defaults(self):
        self.name = 'Mesh generator'
        self.category = 'Modify PCB'
        self.description = 'Creates security mesh traces on a PCB'
        self.icon_file_name = path.join(path.dirname(__file__), 'mesh_plugin_icon.png')
        self.show_toolbar_button = True

    def Run(self):
        if not check_requirements('pyclipper'):
            return

        from .mesh_dialog import show_dialog
        show_dialog(pcbnew.GetBoard())