-
Notifications
You must be signed in to change notification settings - Fork 0
/
FootprintDownload.py
206 lines (176 loc) · 8.65 KB
/
FootprintDownload.py
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
import json
import socket
import pcbnew
import os
from .FootprintDownloadDialog import FootprintDownloadDialogFrame
import threading
import time
from http.server import BaseHTTPRequestHandler, HTTPServer
import wx
from .FootprintDownloadServer import FootprintDownloadServer
HOSTNAME = "localhost"
PORT = 2222
def silence_serverlogs(*args, **kwargs):
# HTTPServer logs get redirected and prompts a textbox window thats not
# required
pass
class FootprintDownloadDialog(FootprintDownloadDialogFrame):
def __init__(self, parent):
super().__init__(parent)
def threadfunc():
while True:
print("threading hard")
time.sleep(1)
def is_server_running():
# this is a brittle and hacky way to persist information
# about the server thread running because we want to keep thread
# running when pcbnew is closed (for instance for working
# in eeschema or for downloading footprints while pcbnew closed)
try:
if pcbnew.footprintdownload_server != None:
print("server is running")
return True
else:
print("Server is not running")
return False
except:
print("server is not running")
return False
class FootprintDownload(pcbnew.ActionPlugin):
def defaults(self):
self.name = "footprint_download"
self.category = "import"
self.show_toolbar_button = "true"
self.icon_file_name = os.path.join(os.path.dirname(__file__), 'logo.png')
self.description = "A tool for downloading footprints and symbols from distributor sites"
def read_config(self):
pcb_filename = pcbnew.GetBoard().GetFileName()
config_filename = os.path.dirname(pcb_filename) + "/.footprintdownloadrc"
if not os.path.exists(config_filename):
self.config = {
"symbol_lib_filename":"",
"footprint_lib_directory": "",
}
else:
with open(config_filename, "r") as f:
self.config = json.loads(f.read())
def save_config(self):
pcb_filename = pcbnew.GetBoard().GetFileName()
config_filename = os.path.dirname(pcb_filename) + "/.footprintdownloadrc"
print("writing config : ", self.config)
with open(config_filename, "w") as f:
f.write(json.dumps(self.config))
def create_default_config(self):
pcb_dir = os.path.dirname(pcbnew.GetBoard().GetFileName())
# symbol file : check exist before create
symbol_lib_filename = pcb_dir + "/footprint_download.kicad_sym"
footprint_lib_directory = pcb_dir + "/footprint_download.pretty"
if os.path.exists(symbol_lib_filename):
wx.MessageBox(f"{symbol_lib_filename} already exists. Doing nothing")
else:
symbol_lib_template = """(kicad_symbol_lib (version 20211014) (generator FootprintDownloader))"""
with open(symbol_lib_filename, "w") as f:
f.write(symbol_lib_template)
if os.path.exists(footprint_lib_directory):
wx.MessageBox(f"{footprint_lib_directory} already exists. Doing nothing")
else:
os.makedirs(footprint_lib_directory)
# put only the relative paths in config
self.config['symbol_lib_filename'] = "/footprint_download.kicad_sym"
self.config['footprint_lib_directory'] = "/footprint_download.pretty"
def Run(self):
dialog = FootprintDownloadDialog(None)
self.read_config()
dialog.symbolLibraryTextbox.SetValue(self.config["symbol_lib_filename"])
dialog.footprintLibraryTextbox.SetValue(self.config["footprint_lib_directory"])
def start_server_handler(_):
if is_server_running():
print("Server is already running. This is an error.")
return
pcbnew.footprintdownload_server = HTTPServer((HOSTNAME, PORT), FootprintDownloadServer)
pcbnew.footprintdownload_server.log_messages = silence_serverlogs
def server_threadfunc():
print("server started")
pcbnew.footprintdownload_server.serve_forever()
print("server stopped")
threading.Thread(target=server_threadfunc).start()
time.sleep(1) # give 1s for the server to start before updating UI accordingly
pcbnew.footprintdownload_server.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
# update config before serving data
# we want full path here not relative
pcbnew.footprintserver_config = {
"symbol_lib_filename": os.path.dirname(pcbnew.GetBoard().GetFileName()) + self.config['symbol_lib_filename'],
"footprint_lib_directory": os.path.dirname(pcbnew.GetBoard().GetFileName()) + self.config['footprint_lib_directory'],
}
if is_server_running():
dialog.serverStatusLabel.SetLabel("Server is running")
dialog.startServerButton.Enable(False)
dialog.stopServerbutton.Enable(True)
# disable configuration changes here
dialog.chooseFootprintLibraryButton.Enable(False)
dialog.chooseSymbolLibraryButton.Enable(False)
dialog.symbolLibraryTextbox.Enable(False)
dialog.footprintLibraryTextbox.Enable(False)
dialog.saveConfigButton.Enable(False)
dialog.createDefaultConfigButton.Enable(False)
def stop_server_handler(_):
if not is_server_running():
dialog.serverStatusLabel.SetLabel("Server is not running. This is an error")
return
pcbnew.footprintdownload_server.shutdown()
#
# wipe the data to show server no longer running
pcbnew.footprintdownload_server = None
dialog.serverStatusLabel.SetLabel("Server stopped.")
dialog.startServerButton.Enable(True)
dialog.stopServerbutton.Enable(False)
# disable configuration changes here
dialog.chooseFootprintLibraryButton.Enable(True)
dialog.chooseSymbolLibraryButton.Enable(True)
dialog.symbolLibraryTextbox.Enable(True)
dialog.footprintLibraryTextbox.Enable(True)
dialog.saveConfigButton.Enable(True)
dialog.createDefaultConfigButton.Enable(True)
def save_config_handler(_):
self.save_config()
wx.MessageBox("Config saved")
def create_default_config_handler(_):
self.create_default_config()
dialog.symbolLibraryTextbox.SetValue(self.config["symbol_lib_filename"])
dialog.footprintLibraryTextbox.SetValue(self.config["footprint_lib_directory"])
wx.MessageBox("Default config created. Don't forget to save")
def choose_symbol_library_handler(_):
wx.MessageBox("Not implemented")
def choose_footprint_library_handler(_):
wx.MessageBox("Not implemented")
dialog.startServerButton.Bind(wx.EVT_BUTTON, start_server_handler)
dialog.stopServerbutton.Bind(wx.EVT_BUTTON, stop_server_handler)
dialog.saveConfigButton.Bind(wx.EVT_BUTTON, save_config_handler)
dialog.createDefaultConfigButton.Bind(wx.EVT_BUTTON, create_default_config_handler)
dialog.chooseFootprintLibraryButton.Bind(wx.EVT_BUTTON, choose_footprint_library_handler)
dialog.chooseSymbolLibraryButton.Bind(wx.EVT_BUTTON, choose_symbol_library_handler)
if is_server_running():
dialog.serverStatusLabel.SetLabel("Server is running.")
dialog.startServerButton.Enable(False)
dialog.stopServerbutton.Enable(True)
# disable configuration changes here
dialog.chooseFootprintLibraryButton.Enable(False)
dialog.chooseSymbolLibraryButton.Enable(False)
dialog.symbolLibraryTextbox.Enable(False)
dialog.footprintLibraryTextbox.Enable(False)
dialog.saveConfigButton.Enable(False)
dialog.createDefaultConfigButton.Enable(False)
else:
dialog.serverStatusLabel.SetLabel("Server is not running")
dialog.startServerButton.Enable(True)
dialog.stopServerbutton.Enable(False)
# disable configuration changes here
dialog.chooseFootprintLibraryButton.Enable(True)
dialog.chooseSymbolLibraryButton.Enable(True)
dialog.symbolLibraryTextbox.Enable(True)
dialog.footprintLibraryTextbox.Enable(True)
dialog.saveConfigButton.Enable(True)
dialog.createDefaultConfigButton.Enable(True)
dialog.Show()
print("registering plugin")
FootprintDownload().register()