meerk40t 0.9.3001__py2.py3-none-any.whl → 0.9.7020__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (446) hide show
  1. meerk40t/__init__.py +1 -1
  2. meerk40t/balormk/balor_params.py +167 -167
  3. meerk40t/balormk/clone_loader.py +457 -457
  4. meerk40t/balormk/controller.py +1566 -1512
  5. meerk40t/balormk/cylindermod.py +64 -0
  6. meerk40t/balormk/device.py +966 -1959
  7. meerk40t/balormk/driver.py +778 -591
  8. meerk40t/balormk/galvo_commands.py +1194 -0
  9. meerk40t/balormk/gui/balorconfig.py +237 -111
  10. meerk40t/balormk/gui/balorcontroller.py +191 -184
  11. meerk40t/balormk/gui/baloroperationproperties.py +116 -115
  12. meerk40t/balormk/gui/corscene.py +845 -0
  13. meerk40t/balormk/gui/gui.py +179 -147
  14. meerk40t/balormk/livelightjob.py +466 -382
  15. meerk40t/balormk/mock_connection.py +131 -109
  16. meerk40t/balormk/plugin.py +133 -135
  17. meerk40t/balormk/usb_connection.py +306 -301
  18. meerk40t/camera/__init__.py +1 -1
  19. meerk40t/camera/camera.py +514 -397
  20. meerk40t/camera/gui/camerapanel.py +1241 -1095
  21. meerk40t/camera/gui/gui.py +58 -58
  22. meerk40t/camera/plugin.py +441 -399
  23. meerk40t/ch341/__init__.py +27 -27
  24. meerk40t/ch341/ch341device.py +628 -628
  25. meerk40t/ch341/libusb.py +595 -589
  26. meerk40t/ch341/mock.py +171 -171
  27. meerk40t/ch341/windriver.py +157 -157
  28. meerk40t/constants.py +13 -0
  29. meerk40t/core/__init__.py +1 -1
  30. meerk40t/core/bindalias.py +550 -539
  31. meerk40t/core/core.py +47 -47
  32. meerk40t/core/cutcode/cubiccut.py +73 -73
  33. meerk40t/core/cutcode/cutcode.py +315 -312
  34. meerk40t/core/cutcode/cutgroup.py +141 -137
  35. meerk40t/core/cutcode/cutobject.py +192 -185
  36. meerk40t/core/cutcode/dwellcut.py +37 -37
  37. meerk40t/core/cutcode/gotocut.py +29 -29
  38. meerk40t/core/cutcode/homecut.py +29 -29
  39. meerk40t/core/cutcode/inputcut.py +34 -34
  40. meerk40t/core/cutcode/linecut.py +33 -33
  41. meerk40t/core/cutcode/outputcut.py +34 -34
  42. meerk40t/core/cutcode/plotcut.py +335 -335
  43. meerk40t/core/cutcode/quadcut.py +61 -61
  44. meerk40t/core/cutcode/rastercut.py +168 -148
  45. meerk40t/core/cutcode/waitcut.py +34 -34
  46. meerk40t/core/cutplan.py +1843 -1316
  47. meerk40t/core/drivers.py +330 -329
  48. meerk40t/core/elements/align.py +801 -669
  49. meerk40t/core/elements/branches.py +1858 -1507
  50. meerk40t/core/elements/clipboard.py +229 -219
  51. meerk40t/core/elements/element_treeops.py +4595 -2837
  52. meerk40t/core/elements/element_types.py +125 -105
  53. meerk40t/core/elements/elements.py +4315 -3617
  54. meerk40t/core/elements/files.py +117 -64
  55. meerk40t/core/elements/geometry.py +473 -224
  56. meerk40t/core/elements/grid.py +467 -316
  57. meerk40t/core/elements/materials.py +158 -94
  58. meerk40t/core/elements/notes.py +50 -38
  59. meerk40t/core/elements/offset_clpr.py +934 -912
  60. meerk40t/core/elements/offset_mk.py +963 -955
  61. meerk40t/core/elements/penbox.py +339 -267
  62. meerk40t/core/elements/placements.py +300 -83
  63. meerk40t/core/elements/render.py +785 -687
  64. meerk40t/core/elements/shapes.py +2618 -2092
  65. meerk40t/core/elements/testcases.py +105 -0
  66. meerk40t/core/elements/trace.py +651 -563
  67. meerk40t/core/elements/tree_commands.py +415 -409
  68. meerk40t/core/elements/undo_redo.py +116 -58
  69. meerk40t/core/elements/wordlist.py +319 -200
  70. meerk40t/core/exceptions.py +9 -9
  71. meerk40t/core/laserjob.py +220 -220
  72. meerk40t/core/logging.py +63 -63
  73. meerk40t/core/node/blobnode.py +83 -86
  74. meerk40t/core/node/bootstrap.py +105 -103
  75. meerk40t/core/node/branch_elems.py +40 -31
  76. meerk40t/core/node/branch_ops.py +45 -38
  77. meerk40t/core/node/branch_regmark.py +48 -41
  78. meerk40t/core/node/cutnode.py +29 -32
  79. meerk40t/core/node/effect_hatch.py +375 -257
  80. meerk40t/core/node/effect_warp.py +398 -0
  81. meerk40t/core/node/effect_wobble.py +441 -309
  82. meerk40t/core/node/elem_ellipse.py +404 -309
  83. meerk40t/core/node/elem_image.py +1082 -801
  84. meerk40t/core/node/elem_line.py +358 -292
  85. meerk40t/core/node/elem_path.py +259 -201
  86. meerk40t/core/node/elem_point.py +129 -102
  87. meerk40t/core/node/elem_polyline.py +310 -246
  88. meerk40t/core/node/elem_rect.py +376 -286
  89. meerk40t/core/node/elem_text.py +445 -418
  90. meerk40t/core/node/filenode.py +59 -40
  91. meerk40t/core/node/groupnode.py +138 -74
  92. meerk40t/core/node/image_processed.py +777 -766
  93. meerk40t/core/node/image_raster.py +156 -113
  94. meerk40t/core/node/layernode.py +31 -31
  95. meerk40t/core/node/mixins.py +135 -107
  96. meerk40t/core/node/node.py +1427 -1304
  97. meerk40t/core/node/nutils.py +117 -114
  98. meerk40t/core/node/op_cut.py +463 -335
  99. meerk40t/core/node/op_dots.py +296 -251
  100. meerk40t/core/node/op_engrave.py +414 -311
  101. meerk40t/core/node/op_image.py +755 -369
  102. meerk40t/core/node/op_raster.py +787 -522
  103. meerk40t/core/node/place_current.py +37 -40
  104. meerk40t/core/node/place_point.py +329 -126
  105. meerk40t/core/node/refnode.py +58 -47
  106. meerk40t/core/node/rootnode.py +225 -219
  107. meerk40t/core/node/util_console.py +48 -48
  108. meerk40t/core/node/util_goto.py +84 -65
  109. meerk40t/core/node/util_home.py +61 -61
  110. meerk40t/core/node/util_input.py +102 -102
  111. meerk40t/core/node/util_output.py +102 -102
  112. meerk40t/core/node/util_wait.py +65 -65
  113. meerk40t/core/parameters.py +709 -707
  114. meerk40t/core/planner.py +875 -785
  115. meerk40t/core/plotplanner.py +656 -652
  116. meerk40t/core/space.py +120 -113
  117. meerk40t/core/spoolers.py +706 -705
  118. meerk40t/core/svg_io.py +1836 -1549
  119. meerk40t/core/treeop.py +534 -445
  120. meerk40t/core/undos.py +278 -124
  121. meerk40t/core/units.py +784 -680
  122. meerk40t/core/view.py +393 -322
  123. meerk40t/core/webhelp.py +62 -62
  124. meerk40t/core/wordlist.py +513 -504
  125. meerk40t/cylinder/cylinder.py +247 -0
  126. meerk40t/cylinder/gui/cylindersettings.py +41 -0
  127. meerk40t/cylinder/gui/gui.py +24 -0
  128. meerk40t/device/__init__.py +1 -1
  129. meerk40t/device/basedevice.py +322 -123
  130. meerk40t/device/devicechoices.py +50 -0
  131. meerk40t/device/dummydevice.py +163 -128
  132. meerk40t/device/gui/defaultactions.py +618 -602
  133. meerk40t/device/gui/effectspanel.py +114 -0
  134. meerk40t/device/gui/formatterpanel.py +253 -290
  135. meerk40t/device/gui/warningpanel.py +337 -260
  136. meerk40t/device/mixins.py +13 -13
  137. meerk40t/dxf/__init__.py +1 -1
  138. meerk40t/dxf/dxf_io.py +766 -554
  139. meerk40t/dxf/plugin.py +47 -35
  140. meerk40t/external_plugins.py +79 -79
  141. meerk40t/external_plugins_build.py +28 -28
  142. meerk40t/extra/cag.py +112 -116
  143. meerk40t/extra/coolant.py +403 -0
  144. meerk40t/extra/encode_detect.py +204 -0
  145. meerk40t/extra/ezd.py +1165 -1165
  146. meerk40t/extra/hershey.py +834 -340
  147. meerk40t/extra/imageactions.py +322 -316
  148. meerk40t/extra/inkscape.py +628 -622
  149. meerk40t/extra/lbrn.py +424 -424
  150. meerk40t/extra/outerworld.py +283 -0
  151. meerk40t/extra/param_functions.py +1542 -1556
  152. meerk40t/extra/potrace.py +257 -253
  153. meerk40t/extra/serial_exchange.py +118 -0
  154. meerk40t/extra/updater.py +602 -453
  155. meerk40t/extra/vectrace.py +147 -146
  156. meerk40t/extra/winsleep.py +83 -83
  157. meerk40t/extra/xcs_reader.py +597 -0
  158. meerk40t/fill/fills.py +781 -335
  159. meerk40t/fill/patternfill.py +1061 -1061
  160. meerk40t/fill/patterns.py +614 -567
  161. meerk40t/grbl/control.py +87 -87
  162. meerk40t/grbl/controller.py +990 -903
  163. meerk40t/grbl/device.py +1084 -768
  164. meerk40t/grbl/driver.py +989 -771
  165. meerk40t/grbl/emulator.py +532 -497
  166. meerk40t/grbl/gcodejob.py +783 -767
  167. meerk40t/grbl/gui/grblconfiguration.py +373 -298
  168. meerk40t/grbl/gui/grblcontroller.py +485 -271
  169. meerk40t/grbl/gui/grblhardwareconfig.py +269 -153
  170. meerk40t/grbl/gui/grbloperationconfig.py +105 -0
  171. meerk40t/grbl/gui/gui.py +147 -116
  172. meerk40t/grbl/interpreter.py +44 -44
  173. meerk40t/grbl/loader.py +22 -22
  174. meerk40t/grbl/mock_connection.py +56 -56
  175. meerk40t/grbl/plugin.py +294 -264
  176. meerk40t/grbl/serial_connection.py +93 -88
  177. meerk40t/grbl/tcp_connection.py +81 -79
  178. meerk40t/grbl/ws_connection.py +112 -0
  179. meerk40t/gui/__init__.py +1 -1
  180. meerk40t/gui/about.py +2042 -296
  181. meerk40t/gui/alignment.py +1644 -1608
  182. meerk40t/gui/autoexec.py +199 -0
  183. meerk40t/gui/basicops.py +791 -670
  184. meerk40t/gui/bufferview.py +77 -71
  185. meerk40t/gui/busy.py +232 -133
  186. meerk40t/gui/choicepropertypanel.py +1662 -1469
  187. meerk40t/gui/consolepanel.py +706 -542
  188. meerk40t/gui/devicepanel.py +687 -581
  189. meerk40t/gui/dialogoptions.py +110 -107
  190. meerk40t/gui/executejob.py +316 -306
  191. meerk40t/gui/fonts.py +90 -90
  192. meerk40t/gui/functionwrapper.py +252 -0
  193. meerk40t/gui/gui_mixins.py +729 -0
  194. meerk40t/gui/guicolors.py +205 -182
  195. meerk40t/gui/help_assets/help_assets.py +218 -201
  196. meerk40t/gui/helper.py +154 -0
  197. meerk40t/gui/hersheymanager.py +1440 -846
  198. meerk40t/gui/icons.py +3422 -2747
  199. meerk40t/gui/imagesplitter.py +555 -508
  200. meerk40t/gui/keymap.py +354 -344
  201. meerk40t/gui/laserpanel.py +897 -806
  202. meerk40t/gui/laserrender.py +1470 -1232
  203. meerk40t/gui/lasertoolpanel.py +805 -793
  204. meerk40t/gui/magnetoptions.py +436 -0
  205. meerk40t/gui/materialmanager.py +2944 -0
  206. meerk40t/gui/materialtest.py +1722 -1694
  207. meerk40t/gui/mkdebug.py +646 -359
  208. meerk40t/gui/mwindow.py +163 -140
  209. meerk40t/gui/navigationpanels.py +2605 -2467
  210. meerk40t/gui/notes.py +143 -142
  211. meerk40t/gui/opassignment.py +414 -410
  212. meerk40t/gui/operation_info.py +310 -299
  213. meerk40t/gui/plugin.py +500 -328
  214. meerk40t/gui/position.py +714 -669
  215. meerk40t/gui/preferences.py +901 -650
  216. meerk40t/gui/propertypanels/attributes.py +1461 -1131
  217. meerk40t/gui/propertypanels/blobproperty.py +117 -114
  218. meerk40t/gui/propertypanels/consoleproperty.py +83 -80
  219. meerk40t/gui/propertypanels/gotoproperty.py +77 -0
  220. meerk40t/gui/propertypanels/groupproperties.py +223 -217
  221. meerk40t/gui/propertypanels/hatchproperty.py +489 -469
  222. meerk40t/gui/propertypanels/imageproperty.py +2244 -1384
  223. meerk40t/gui/propertypanels/inputproperty.py +59 -58
  224. meerk40t/gui/propertypanels/opbranchproperties.py +82 -80
  225. meerk40t/gui/propertypanels/operationpropertymain.py +1890 -1638
  226. meerk40t/gui/propertypanels/outputproperty.py +59 -58
  227. meerk40t/gui/propertypanels/pathproperty.py +389 -380
  228. meerk40t/gui/propertypanels/placementproperty.py +1214 -383
  229. meerk40t/gui/propertypanels/pointproperty.py +140 -136
  230. meerk40t/gui/propertypanels/propertywindow.py +313 -181
  231. meerk40t/gui/propertypanels/rasterwizardpanels.py +996 -912
  232. meerk40t/gui/propertypanels/regbranchproperties.py +76 -0
  233. meerk40t/gui/propertypanels/textproperty.py +770 -755
  234. meerk40t/gui/propertypanels/waitproperty.py +56 -55
  235. meerk40t/gui/propertypanels/warpproperty.py +121 -0
  236. meerk40t/gui/propertypanels/wobbleproperty.py +255 -204
  237. meerk40t/gui/ribbon.py +2471 -2210
  238. meerk40t/gui/scene/scene.py +1100 -1051
  239. meerk40t/gui/scene/sceneconst.py +22 -22
  240. meerk40t/gui/scene/scenepanel.py +439 -349
  241. meerk40t/gui/scene/scenespacewidget.py +365 -365
  242. meerk40t/gui/scene/widget.py +518 -505
  243. meerk40t/gui/scenewidgets/affinemover.py +215 -215
  244. meerk40t/gui/scenewidgets/attractionwidget.py +315 -309
  245. meerk40t/gui/scenewidgets/bedwidget.py +120 -97
  246. meerk40t/gui/scenewidgets/elementswidget.py +137 -107
  247. meerk40t/gui/scenewidgets/gridwidget.py +785 -745
  248. meerk40t/gui/scenewidgets/guidewidget.py +765 -765
  249. meerk40t/gui/scenewidgets/laserpathwidget.py +66 -66
  250. meerk40t/gui/scenewidgets/machineoriginwidget.py +86 -86
  251. meerk40t/gui/scenewidgets/nodeselector.py +28 -28
  252. meerk40t/gui/scenewidgets/rectselectwidget.py +592 -346
  253. meerk40t/gui/scenewidgets/relocatewidget.py +33 -33
  254. meerk40t/gui/scenewidgets/reticlewidget.py +83 -83
  255. meerk40t/gui/scenewidgets/selectionwidget.py +2958 -2756
  256. meerk40t/gui/simpleui.py +362 -333
  257. meerk40t/gui/simulation.py +2451 -2094
  258. meerk40t/gui/snapoptions.py +208 -203
  259. meerk40t/gui/spoolerpanel.py +1227 -1180
  260. meerk40t/gui/statusbarwidgets/defaultoperations.py +480 -353
  261. meerk40t/gui/statusbarwidgets/infowidget.py +520 -483
  262. meerk40t/gui/statusbarwidgets/opassignwidget.py +356 -355
  263. meerk40t/gui/statusbarwidgets/selectionwidget.py +172 -171
  264. meerk40t/gui/statusbarwidgets/shapepropwidget.py +754 -236
  265. meerk40t/gui/statusbarwidgets/statusbar.py +272 -260
  266. meerk40t/gui/statusbarwidgets/statusbarwidget.py +268 -270
  267. meerk40t/gui/statusbarwidgets/strokewidget.py +267 -251
  268. meerk40t/gui/themes.py +200 -78
  269. meerk40t/gui/tips.py +590 -0
  270. meerk40t/gui/toolwidgets/circlebrush.py +35 -35
  271. meerk40t/gui/toolwidgets/toolcircle.py +248 -242
  272. meerk40t/gui/toolwidgets/toolcontainer.py +82 -77
  273. meerk40t/gui/toolwidgets/tooldraw.py +97 -90
  274. meerk40t/gui/toolwidgets/toolellipse.py +219 -212
  275. meerk40t/gui/toolwidgets/toolimagecut.py +25 -132
  276. meerk40t/gui/toolwidgets/toolline.py +39 -144
  277. meerk40t/gui/toolwidgets/toollinetext.py +79 -236
  278. meerk40t/gui/toolwidgets/toollinetext_inline.py +296 -0
  279. meerk40t/gui/toolwidgets/toolmeasure.py +163 -216
  280. meerk40t/gui/toolwidgets/toolnodeedit.py +2088 -2074
  281. meerk40t/gui/toolwidgets/toolnodemove.py +92 -94
  282. meerk40t/gui/toolwidgets/toolparameter.py +754 -668
  283. meerk40t/gui/toolwidgets/toolplacement.py +108 -108
  284. meerk40t/gui/toolwidgets/toolpoint.py +68 -59
  285. meerk40t/gui/toolwidgets/toolpointlistbuilder.py +294 -0
  286. meerk40t/gui/toolwidgets/toolpointmove.py +183 -0
  287. meerk40t/gui/toolwidgets/toolpolygon.py +288 -403
  288. meerk40t/gui/toolwidgets/toolpolyline.py +38 -196
  289. meerk40t/gui/toolwidgets/toolrect.py +211 -207
  290. meerk40t/gui/toolwidgets/toolrelocate.py +72 -72
  291. meerk40t/gui/toolwidgets/toolribbon.py +598 -113
  292. meerk40t/gui/toolwidgets/tooltabedit.py +546 -0
  293. meerk40t/gui/toolwidgets/tooltext.py +98 -89
  294. meerk40t/gui/toolwidgets/toolvector.py +213 -204
  295. meerk40t/gui/toolwidgets/toolwidget.py +39 -39
  296. meerk40t/gui/usbconnect.py +98 -91
  297. meerk40t/gui/utilitywidgets/buttonwidget.py +18 -18
  298. meerk40t/gui/utilitywidgets/checkboxwidget.py +90 -90
  299. meerk40t/gui/utilitywidgets/controlwidget.py +14 -14
  300. meerk40t/gui/utilitywidgets/cyclocycloidwidget.py +343 -340
  301. meerk40t/gui/utilitywidgets/debugwidgets.py +148 -0
  302. meerk40t/gui/utilitywidgets/handlewidget.py +27 -27
  303. meerk40t/gui/utilitywidgets/harmonograph.py +450 -447
  304. meerk40t/gui/utilitywidgets/openclosewidget.py +40 -40
  305. meerk40t/gui/utilitywidgets/rotationwidget.py +54 -54
  306. meerk40t/gui/utilitywidgets/scalewidget.py +75 -75
  307. meerk40t/gui/utilitywidgets/seekbarwidget.py +183 -183
  308. meerk40t/gui/utilitywidgets/togglewidget.py +142 -142
  309. meerk40t/gui/utilitywidgets/toolbarwidget.py +8 -8
  310. meerk40t/gui/wordlisteditor.py +985 -931
  311. meerk40t/gui/wxmeerk40t.py +1447 -1169
  312. meerk40t/gui/wxmmain.py +5644 -4112
  313. meerk40t/gui/wxmribbon.py +1591 -1076
  314. meerk40t/gui/wxmscene.py +1631 -1453
  315. meerk40t/gui/wxmtree.py +2416 -2089
  316. meerk40t/gui/wxutils.py +1769 -1099
  317. meerk40t/gui/zmatrix.py +102 -102
  318. meerk40t/image/__init__.py +1 -1
  319. meerk40t/image/dither.py +429 -0
  320. meerk40t/image/imagetools.py +2793 -2269
  321. meerk40t/internal_plugins.py +150 -130
  322. meerk40t/kernel/__init__.py +63 -12
  323. meerk40t/kernel/channel.py +259 -212
  324. meerk40t/kernel/context.py +538 -538
  325. meerk40t/kernel/exceptions.py +41 -41
  326. meerk40t/kernel/functions.py +463 -414
  327. meerk40t/kernel/jobs.py +100 -100
  328. meerk40t/kernel/kernel.py +3828 -3571
  329. meerk40t/kernel/lifecycles.py +71 -71
  330. meerk40t/kernel/module.py +49 -49
  331. meerk40t/kernel/service.py +147 -147
  332. meerk40t/kernel/settings.py +383 -343
  333. meerk40t/lihuiyu/controller.py +883 -876
  334. meerk40t/lihuiyu/device.py +1181 -1069
  335. meerk40t/lihuiyu/driver.py +1466 -1372
  336. meerk40t/lihuiyu/gui/gui.py +127 -106
  337. meerk40t/lihuiyu/gui/lhyaccelgui.py +377 -363
  338. meerk40t/lihuiyu/gui/lhycontrollergui.py +741 -651
  339. meerk40t/lihuiyu/gui/lhydrivergui.py +470 -446
  340. meerk40t/lihuiyu/gui/lhyoperationproperties.py +238 -237
  341. meerk40t/lihuiyu/gui/tcpcontroller.py +226 -190
  342. meerk40t/lihuiyu/interpreter.py +53 -53
  343. meerk40t/lihuiyu/laserspeed.py +450 -450
  344. meerk40t/lihuiyu/loader.py +90 -90
  345. meerk40t/lihuiyu/parser.py +404 -404
  346. meerk40t/lihuiyu/plugin.py +101 -102
  347. meerk40t/lihuiyu/tcp_connection.py +111 -109
  348. meerk40t/main.py +231 -165
  349. meerk40t/moshi/builder.py +788 -781
  350. meerk40t/moshi/controller.py +505 -499
  351. meerk40t/moshi/device.py +495 -442
  352. meerk40t/moshi/driver.py +862 -696
  353. meerk40t/moshi/gui/gui.py +78 -76
  354. meerk40t/moshi/gui/moshicontrollergui.py +538 -522
  355. meerk40t/moshi/gui/moshidrivergui.py +87 -75
  356. meerk40t/moshi/plugin.py +43 -43
  357. meerk40t/network/console_server.py +140 -57
  358. meerk40t/network/kernelserver.py +10 -9
  359. meerk40t/network/tcp_server.py +142 -140
  360. meerk40t/network/udp_server.py +103 -77
  361. meerk40t/network/web_server.py +404 -0
  362. meerk40t/newly/controller.py +1158 -1144
  363. meerk40t/newly/device.py +874 -732
  364. meerk40t/newly/driver.py +540 -412
  365. meerk40t/newly/gui/gui.py +219 -188
  366. meerk40t/newly/gui/newlyconfig.py +116 -101
  367. meerk40t/newly/gui/newlycontroller.py +193 -186
  368. meerk40t/newly/gui/operationproperties.py +51 -51
  369. meerk40t/newly/mock_connection.py +82 -82
  370. meerk40t/newly/newly_params.py +56 -56
  371. meerk40t/newly/plugin.py +1214 -1246
  372. meerk40t/newly/usb_connection.py +322 -322
  373. meerk40t/rotary/gui/gui.py +52 -46
  374. meerk40t/rotary/gui/rotarysettings.py +240 -232
  375. meerk40t/rotary/rotary.py +202 -98
  376. meerk40t/ruida/control.py +291 -91
  377. meerk40t/ruida/controller.py +138 -1088
  378. meerk40t/ruida/device.py +676 -231
  379. meerk40t/ruida/driver.py +534 -472
  380. meerk40t/ruida/emulator.py +1494 -1491
  381. meerk40t/ruida/exceptions.py +4 -4
  382. meerk40t/ruida/gui/gui.py +71 -76
  383. meerk40t/ruida/gui/ruidaconfig.py +239 -72
  384. meerk40t/ruida/gui/ruidacontroller.py +187 -184
  385. meerk40t/ruida/gui/ruidaoperationproperties.py +48 -47
  386. meerk40t/ruida/loader.py +54 -52
  387. meerk40t/ruida/mock_connection.py +57 -109
  388. meerk40t/ruida/plugin.py +124 -87
  389. meerk40t/ruida/rdjob.py +2084 -945
  390. meerk40t/ruida/serial_connection.py +116 -0
  391. meerk40t/ruida/tcp_connection.py +146 -0
  392. meerk40t/ruida/udp_connection.py +73 -0
  393. meerk40t/svgelements.py +9671 -9669
  394. meerk40t/tools/driver_to_path.py +584 -579
  395. meerk40t/tools/geomstr.py +5583 -4680
  396. meerk40t/tools/jhfparser.py +357 -292
  397. meerk40t/tools/kerftest.py +904 -890
  398. meerk40t/tools/livinghinges.py +1168 -1033
  399. meerk40t/tools/pathtools.py +987 -949
  400. meerk40t/tools/pmatrix.py +234 -0
  401. meerk40t/tools/pointfinder.py +942 -942
  402. meerk40t/tools/polybool.py +941 -940
  403. meerk40t/tools/rasterplotter.py +1660 -547
  404. meerk40t/tools/shxparser.py +1047 -901
  405. meerk40t/tools/ttfparser.py +726 -446
  406. meerk40t/tools/zinglplotter.py +595 -593
  407. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7020.dist-info}/LICENSE +21 -21
  408. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7020.dist-info}/METADATA +150 -139
  409. meerk40t-0.9.7020.dist-info/RECORD +446 -0
  410. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7020.dist-info}/WHEEL +1 -1
  411. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7020.dist-info}/top_level.txt +0 -1
  412. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7020.dist-info}/zip-safe +1 -1
  413. meerk40t/balormk/elementlightjob.py +0 -159
  414. meerk40t-0.9.3001.dist-info/RECORD +0 -437
  415. test/bootstrap.py +0 -63
  416. test/test_cli.py +0 -12
  417. test/test_core_cutcode.py +0 -418
  418. test/test_core_elements.py +0 -144
  419. test/test_core_plotplanner.py +0 -397
  420. test/test_core_viewports.py +0 -312
  421. test/test_drivers_grbl.py +0 -108
  422. test/test_drivers_lihuiyu.py +0 -443
  423. test/test_drivers_newly.py +0 -113
  424. test/test_element_degenerate_points.py +0 -43
  425. test/test_elements_classify.py +0 -97
  426. test/test_elements_penbox.py +0 -22
  427. test/test_file_svg.py +0 -176
  428. test/test_fill.py +0 -155
  429. test/test_geomstr.py +0 -1523
  430. test/test_geomstr_nodes.py +0 -18
  431. test/test_imagetools_actualize.py +0 -306
  432. test/test_imagetools_wizard.py +0 -258
  433. test/test_kernel.py +0 -200
  434. test/test_laser_speeds.py +0 -3303
  435. test/test_length.py +0 -57
  436. test/test_lifecycle.py +0 -66
  437. test/test_operations.py +0 -251
  438. test/test_operations_hatch.py +0 -57
  439. test/test_ruida.py +0 -19
  440. test/test_spooler.py +0 -22
  441. test/test_tools_rasterplotter.py +0 -29
  442. test/test_wobble.py +0 -133
  443. test/test_zingl.py +0 -124
  444. {test → meerk40t/cylinder}/__init__.py +0 -0
  445. /meerk40t/{core/element_commands.py → cylinder/gui/__init__.py} +0 -0
  446. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7020.dist-info}/entry_points.txt +0 -0
@@ -1,931 +1,985 @@
1
- import os
2
- import re
3
-
4
- import wx
5
- from wx import aui
6
-
7
- from ..kernel import signal_listener
8
- from .icons import (
9
- STD_ICON_SIZE,
10
- get_default_icon_size,
11
- icon_add_new,
12
- icon_edit,
13
- icon_trash,
14
- icons8_circled_left,
15
- icons8_circled_right,
16
- icons8_curly_brackets,
17
- icons8_paste,
18
- )
19
- from .mwindow import MWindow
20
- from .wxutils import StaticBoxSizer, dip_size
21
-
22
- _ = wx.GetTranslation
23
-
24
-
25
- def register_panel_wordlist(window, context):
26
- pane = (
27
- aui.AuiPaneInfo()
28
- .Left()
29
- .MinSize(150, 25)
30
- .FloatingSize(150, 35)
31
- .Caption(_("Wordlist"))
32
- .CaptionVisible(not context.pane_lock)
33
- .Name("wordlist")
34
- .Hide()
35
- )
36
- pane.dock_proportion = 225
37
- pane.control = WordlistMiniPanel(window, wx.ID_ANY, context=context)
38
- pane.submenu = "_50_" + _("Tools")
39
- window.on_pane_create(pane)
40
- context.register("pane/wordlist", pane)
41
-
42
-
43
- class WordlistMiniPanel(wx.Panel):
44
- def __init__(self, *args, context=None, **kwds):
45
- kwds["style"] = kwds.get("style", 0) | wx.TAB_TRAVERSAL
46
- wx.Panel.__init__(self, *args, **kwds)
47
- self.context = context
48
- main_sizer = wx.BoxSizer(wx.HORIZONTAL)
49
- self.button_edit = wx.Button(self, wx.ID_ANY, _("Edit"))
50
- self.button_edit.SetBitmap(
51
- icons8_curly_brackets.GetBitmap(resize=0.5 * get_default_icon_size())
52
- )
53
- self.button_edit.SetToolTip(_("Manages Wordlist-Entries"))
54
-
55
- self.button_next = wx.Button(self, wx.ID_ANY, _("Next"))
56
- self.button_next.SetBitmap(
57
- icons8_circled_right.GetBitmap(resize=0.5 * get_default_icon_size())
58
- )
59
- self.button_next.SetToolTip(
60
- _("Wordlist: go to next page (right-click to next entry)")
61
- )
62
-
63
- self.button_prev = wx.Button(self, wx.ID_ANY, _("Prev"))
64
- self.button_prev.SetBitmap(
65
- icons8_circled_left.GetBitmap(resize=0.5 * get_default_icon_size())
66
- )
67
- self.button_prev.SetToolTip(
68
- _("Wordlist: go to previous page (right-click to previous entry)")
69
- )
70
-
71
- main_sizer.Add(self.button_prev, 1, wx.EXPAND, 0)
72
- main_sizer.Add(self.button_edit, 1, wx.EXPAND, 0)
73
- main_sizer.Add(self.button_next, 1, wx.EXPAND, 0)
74
-
75
- self.button_next.Bind(wx.EVT_BUTTON, self.on_button_next_page)
76
- self.button_prev.Bind(wx.EVT_BUTTON, self.on_button_prev_page)
77
- self.button_next.Bind(wx.EVT_RIGHT_DOWN, self.on_button_next)
78
- self.button_prev.Bind(wx.EVT_RIGHT_DOWN, self.on_button_prev)
79
- self.button_edit.Bind(wx.EVT_BUTTON, self.on_button_edit)
80
-
81
- self.SetSizer(main_sizer)
82
- self.Layout()
83
-
84
- def establish_max_delta(self):
85
- # try to establish the needed delta to satisfy all variables...
86
- deltamin = 0
87
- deltamax = 0
88
- for node in self.context.elements.elems():
89
- sample = ""
90
- if node.type == "elem text":
91
- if node.text is not None:
92
- sample = node.text
93
- elif node.type == "elem path" and hasattr(node, "mktext"):
94
- if node.mktext is not None:
95
- sample = node.mktext
96
- if sample == "":
97
- continue
98
- # we can be rather agnostic on the individual variable, we are interested in the highest {variable#+offset} pattern
99
- brackets = re.compile(r"\{[^}]+\}")
100
- for bracketed_key in brackets.findall(sample):
101
- # print(f"Key found: {bracketed_key}")
102
- key = bracketed_key[1:-1].lower().strip()
103
- relative = 0
104
- pos = key.find("#")
105
- if pos > 0: # Needs to be after first character
106
- # Process offset modification.
107
- index_string = key[pos + 1 :]
108
- key = key[:pos].strip()
109
- if index_string.startswith("+") or index_string.startswith("-"):
110
- try:
111
- # This covers +x, -x, x
112
- relative = int(index_string)
113
- except ValueError:
114
- relative = 0
115
- deltamin = min(relative, deltamin)
116
- deltamax = max(relative, deltamax)
117
-
118
- return deltamax
119
-
120
- def on_button_edit(self, event):
121
- self.context("window toggle Wordlist\n")
122
-
123
- def on_button_prev(self, event):
124
- self.context.elements.wordlist_advance(-1)
125
-
126
- def on_button_prev_page(self, event):
127
- delta = self.establish_max_delta()
128
- self.context.elements.wordlist_advance(-1 - delta)
129
-
130
- def on_button_next(self, event):
131
- self.context.elements.wordlist_advance(+1)
132
-
133
- def on_button_next_page(self, event):
134
- delta = self.establish_max_delta()
135
- self.context.elements.wordlist_advance(+1 + delta)
136
-
137
-
138
- class WordlistPanel(wx.Panel):
139
- def __init__(self, *args, context=None, **kwds):
140
- kwds["style"] = kwds.get("style", 0) | wx.TAB_TRAVERSAL
141
- wx.Panel.__init__(self, *args, **kwds)
142
- self.context = context
143
- self.parent_panel = None
144
- self.wlist = self.context.elements.mywordlist
145
- self.current_entry = None
146
- # For Grid editing
147
- self.cur_skey = None
148
- self.cur_index = None
149
- self.to_save_content = None
150
- self.to_save_wordlist = None
151
-
152
- self.context.setting(bool, "wordlist_autosave", False)
153
-
154
- sizer_main = wx.BoxSizer(wx.VERTICAL)
155
- sizer_grids = wx.BoxSizer(wx.HORIZONTAL)
156
- sizer_grid_left = wx.BoxSizer(wx.VERTICAL)
157
- sizer_grid_right = wx.BoxSizer(wx.VERTICAL)
158
- sizer_grids.Add(sizer_grid_left, 1, wx.EXPAND, 0)
159
- sizer_grids.Add(sizer_grid_right, 1, wx.EXPAND, 0)
160
-
161
- sizer_index_left = wx.BoxSizer(wx.HORIZONTAL)
162
- sizer_grid_left.Add(sizer_index_left, 0, wx.EXPAND, 0)
163
-
164
- label_2 = wx.StaticText(self, wx.ID_ANY, _("Start Index for CSV-based data:"))
165
- sizer_index_left.Add(label_2, 0, wx.ALIGN_CENTER_VERTICAL, 0)
166
- self.cbo_Index = wx.ComboBox(
167
- self, wx.ID_ANY, choices=[], style=wx.CB_DROPDOWN | wx.CB_READONLY
168
- )
169
- sizer_index_left.Add(self.cbo_Index, 0, wx.ALIGN_CENTER_VERTICAL, 0)
170
-
171
- self.grid_wordlist = wx.ListCtrl(
172
- self,
173
- wx.ID_ANY,
174
- style=wx.LC_HRULES
175
- | wx.LC_REPORT
176
- | wx.LC_VRULES
177
- | wx.LC_SINGLE_SEL
178
- | wx.LC_EDIT_LABELS,
179
- )
180
- sizer_grid_left.Add(self.grid_wordlist, 1, wx.EXPAND, 0)
181
-
182
- self.grid_content = wx.ListCtrl(
183
- self,
184
- wx.ID_ANY,
185
- style=wx.LC_HRULES
186
- | wx.LC_REPORT
187
- | wx.LC_VRULES
188
- | wx.LC_SINGLE_SEL
189
- | wx.LC_EDIT_LABELS,
190
- )
191
-
192
- sizer_edit_wordlist_buttons = wx.BoxSizer(wx.HORIZONTAL)
193
- sizer_edit_content_buttons = wx.BoxSizer(wx.HORIZONTAL)
194
-
195
- dummylabel = wx.StaticText(self, wx.ID_ANY, " ")
196
- sizer_index_left.Add(dummylabel, 1, wx.ALIGN_CENTER_VERTICAL, 0)
197
- sizer_index_left.Add(
198
- sizer_edit_wordlist_buttons, 0, wx.ALIGN_CENTER_VERTICAL, 0
199
- )
200
- testsize = dip_size(self, 22, 22)
201
- icon_size = testsize[0]
202
-
203
- self.btn_edit_wordlist_del = wx.StaticBitmap(
204
- self, wx.ID_ANY, size=dip_size(self, 25, 25)
205
- )
206
- self.btn_edit_wordlist_edit = wx.StaticBitmap(
207
- self, wx.ID_ANY, size=dip_size(self, 25, 25)
208
- )
209
- self.btn_edit_content_add = wx.StaticBitmap(
210
- self, wx.ID_ANY, size=dip_size(self, 25, 25)
211
- )
212
- self.btn_edit_content_del = wx.StaticBitmap(
213
- self, wx.ID_ANY, size=dip_size(self, 25, 25)
214
- )
215
- self.btn_edit_content_edit = wx.StaticBitmap(
216
- self, wx.ID_ANY, size=dip_size(self, 25, 25)
217
- )
218
- self.btn_edit_content_paste = wx.StaticBitmap(
219
- self, wx.ID_ANY, size=dip_size(self, 25, 25)
220
- )
221
- self.btn_edit_wordlist_del.SetBitmap(
222
- icon_trash.GetBitmap(resize=icon_size, buffer=1)
223
- )
224
- self.btn_edit_wordlist_edit.SetBitmap(
225
- icon_edit.GetBitmap(resize=icon_size, buffer=1)
226
- )
227
- self.btn_edit_content_add.SetBitmap(
228
- icon_add_new.GetBitmap(resize=icon_size, buffer=1)
229
- )
230
- self.btn_edit_content_del.SetBitmap(
231
- icon_trash.GetBitmap(resize=icon_size, buffer=1)
232
- )
233
- self.btn_edit_content_edit.SetBitmap(
234
- icon_edit.GetBitmap(resize=icon_size, buffer=1)
235
- )
236
- self.btn_edit_content_paste.SetBitmap(
237
- icons8_paste.GetBitmap(resize=icon_size, buffer=1)
238
- )
239
-
240
- self.btn_edit_wordlist_del.SetToolTip(_("Delete the current variable"))
241
- self.btn_edit_wordlist_edit.SetToolTip(
242
- _("Edit the name of the active variable")
243
- )
244
- self.btn_edit_content_add.SetToolTip(
245
- _("Add a new entry for the active variable")
246
- )
247
- self.btn_edit_content_del.SetToolTip(
248
- _("Delete the current entry for the active variable")
249
- )
250
- self.btn_edit_content_edit.SetToolTip(
251
- _("Edit the current entry for the active variable")
252
- )
253
- self.btn_edit_content_paste.SetToolTip(
254
- _(
255
- "Paste the clipboard as new entries for the active variable, any line as new entry"
256
- )
257
- )
258
- minsize = 23
259
- self.btn_edit_wordlist_del.SetMinSize(dip_size(self, minsize, minsize))
260
- self.btn_edit_wordlist_edit.SetMinSize(dip_size(self, minsize, minsize))
261
- self.btn_edit_content_add.SetMinSize(dip_size(self, minsize, minsize))
262
- self.btn_edit_content_del.SetMinSize(dip_size(self, minsize, minsize))
263
- self.btn_edit_content_edit.SetMinSize(dip_size(self, minsize, minsize))
264
- self.btn_edit_content_paste.SetMinSize(dip_size(self, minsize, minsize))
265
-
266
- sizer_edit_wordlist_buttons.Add(self.btn_edit_wordlist_del, 0, wx.EXPAND, 0)
267
- sizer_edit_wordlist_buttons.Add(self.btn_edit_wordlist_edit, 0, wx.EXPAND, 0)
268
-
269
- sizer_edit_content_buttons.Add(self.btn_edit_content_add, 0, wx.EXPAND, 0)
270
- sizer_edit_content_buttons.Add(self.btn_edit_content_del, 0, wx.EXPAND, 0)
271
- sizer_edit_content_buttons.Add(self.btn_edit_content_edit, 0, wx.EXPAND, 0)
272
- sizer_edit_content_buttons.Add(self.btn_edit_content_paste, 0, wx.EXPAND, 0)
273
-
274
- sizer_index_right = wx.BoxSizer(wx.HORIZONTAL)
275
- label_2 = wx.StaticText(self, wx.ID_ANY, _("Start Index for field:"))
276
- sizer_index_right.Add(label_2, 0, wx.ALIGN_CENTER_VERTICAL, 0)
277
- self.cbo_index_single = wx.ComboBox(
278
- self, wx.ID_ANY, choices=[], style=wx.CB_DROPDOWN | wx.CB_READONLY
279
- )
280
- sizer_index_right.Add(self.cbo_index_single, 1, wx.ALIGN_CENTER_VERTICAL, 0)
281
-
282
- dummylabel = wx.StaticText(self, wx.ID_ANY, " ")
283
- sizer_index_right.Add(dummylabel, 1, wx.ALIGN_CENTER_VERTICAL, 0)
284
- sizer_index_right.Add(
285
- sizer_edit_content_buttons, 0, wx.ALIGN_CENTER_VERTICAL, 0
286
- )
287
-
288
- sizer_grid_right.Add(sizer_index_right, 0, wx.EXPAND, 0)
289
- sizer_grid_right.Add(self.grid_content, 1, wx.EXPAND, 0)
290
-
291
- self.txt_pattern = wx.TextCtrl(self, wx.ID_ANY, "")
292
-
293
- sizer_buttons = wx.BoxSizer(wx.HORIZONTAL)
294
- sizer_buttons.Add(self.txt_pattern, 1, wx.ALIGN_CENTER_VERTICAL, 0)
295
-
296
- self.btn_add = wx.Button(self, wx.ID_ANY, _("Add Text"))
297
- self.btn_add.SetToolTip(_("Add another wordlist entry"))
298
- sizer_buttons.Add(self.btn_add, 0, 0, 0)
299
-
300
- self.btn_add_counter = wx.Button(self, wx.ID_ANY, _("Add Counter"))
301
- sizer_buttons.Add(self.btn_add_counter, 0, 0, 0)
302
-
303
- self.btn_delete = wx.Button(self, wx.ID_ANY, _("Delete"))
304
- self.btn_delete.SetToolTip(_("Delete the current wordlist entry"))
305
- sizer_buttons.Add(self.btn_delete, 0, 0, 0)
306
-
307
- sizer_exit = wx.BoxSizer(wx.HORIZONTAL)
308
-
309
- self.btn_backup = wx.Button(self, wx.ID_ANY, _("Backup Wordlist"))
310
- self.btn_backup.SetToolTip(_("Save current wordlist to disk"))
311
- sizer_exit.Add(self.btn_backup, 0, wx.ALIGN_CENTER_VERTICAL, 0)
312
-
313
- self.btn_restore = wx.Button(self, wx.ID_ANY, _("Restore Wordlist"))
314
- self.btn_restore.SetToolTip(_("Load wordlist from disk"))
315
- sizer_exit.Add(self.btn_restore, 0, wx.ALIGN_CENTER_VERTICAL, 0)
316
-
317
- self.check_autosave = wx.CheckBox(self, wx.ID_ANY, _("Autosave"))
318
- self.check_autosave.SetToolTip(
319
- _("All changes to the wordlist will be saved immediately")
320
- )
321
- sizer_exit.Add(self.check_autosave, 0, wx.ALIGN_CENTER_VERTICAL, 0)
322
- self.check_autosave.SetValue(self.context.wordlist_autosave)
323
-
324
- self.lbl_message = wx.StaticText(self, wx.ID_ANY, "")
325
- sizer_exit.Add(self.lbl_message, 1, wx.ALIGN_CENTER_VERTICAL, 0)
326
-
327
- sizer_lower = wx.BoxSizer(wx.HORIZONTAL)
328
- sizer_lower.Add(sizer_buttons, 1, wx.ALL, 0)
329
- sizer_lower.Add(sizer_exit, 1, wx.ALL, 0)
330
-
331
- sizer_main.Add(sizer_grids, 1, wx.EXPAND, 0)
332
- sizer_main.Add(sizer_lower, 0, wx.ALL, 0)
333
-
334
- self.SetSizer(sizer_main)
335
- self.Layout()
336
- self._set_logic()
337
- self.populate_gui()
338
-
339
- def _set_logic(self):
340
- self.btn_add.Bind(wx.EVT_BUTTON, self.on_btn_add)
341
- self.btn_add_counter.Bind(wx.EVT_BUTTON, self.on_add_counter)
342
- self.btn_delete.Bind(wx.EVT_BUTTON, self.on_btn_delete)
343
- self.btn_backup.Bind(wx.EVT_BUTTON, self.on_backup)
344
- self.btn_restore.Bind(wx.EVT_BUTTON, self.on_restore)
345
- self.txt_pattern.Bind(wx.EVT_TEXT, self.on_patterntext_change)
346
- self.cbo_Index.Bind(wx.EVT_COMBOBOX, self.on_cbo_select)
347
- self.grid_wordlist.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_grid_wordlist)
348
- self.grid_wordlist.Bind(
349
- wx.EVT_LIST_BEGIN_LABEL_EDIT, self.on_begin_edit_wordlist
350
- )
351
- self.grid_wordlist.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.on_end_edit_wordlist)
352
- self.grid_content.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_grid_content)
353
- self.grid_content.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.on_begin_edit_content)
354
- self.grid_content.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.on_end_edit_content)
355
- self.grid_content.Bind(wx.EVT_LEFT_DCLICK, self.on_content_dblclick)
356
- self.check_autosave.Bind(wx.EVT_CHECKBOX, self.on_checkbox_autosave)
357
-
358
- self.btn_edit_wordlist_del.Bind(wx.EVT_LEFT_DOWN, self.on_btn_edit_wordlist_del)
359
- self.btn_edit_wordlist_edit.Bind(
360
- wx.EVT_LEFT_DOWN, self.on_btn_edit_wordlist_edit
361
- )
362
- self.btn_edit_content_add.Bind(wx.EVT_LEFT_DOWN, self.on_btn_edit_content_add)
363
- self.btn_edit_content_del.Bind(wx.EVT_LEFT_DOWN, self.on_btn_edit_content_del)
364
- self.btn_edit_content_edit.Bind(wx.EVT_LEFT_DOWN, self.on_btn_edit_content_edit)
365
- self.btn_edit_content_paste.Bind(
366
- wx.EVT_LEFT_DOWN, self.on_btn_edit_content_paste
367
- )
368
- self.cbo_index_single.Bind(wx.EVT_COMBOBOX, self.on_single_index)
369
- # Key handler for F2
370
- self.grid_content.Bind(wx.EVT_CHAR, self.on_key_grid)
371
- self.grid_wordlist.Bind(wx.EVT_CHAR, self.on_key_grid)
372
-
373
- def set_parent(self, par_panel):
374
- self.parent_panel = par_panel
375
-
376
- def pane_show(self):
377
- self.populate_gui()
378
- self.grid_wordlist.SetFocus()
379
-
380
- def pane_hide(self):
381
- pass
382
-
383
- def autosave(self):
384
- if self.check_autosave.GetValue():
385
- if self.wlist.default_filename is not None:
386
- self.wlist.save_data(self.wlist.default_filename)
387
- msg = _("Saved to ") + self.wlist.default_filename
388
- self.edit_message(msg)
389
-
390
- def on_key_grid(self, event):
391
- grid = event.GetEventObject()
392
- key = event.GetKeyCode()
393
- if key == wx.WXK_F2:
394
- index = grid.GetFirstSelected()
395
- if index >= 0:
396
- grid.EditLabel(index)
397
- # that consumes the key
398
- return
399
- # Let's make sure the keystroke is processed further
400
- event.Skip()
401
-
402
- def on_checkbox_autosave(self, event):
403
- self.context.wordlist_autosave = self.check_autosave.GetValue()
404
- event.Skip()
405
-
406
- def on_btn_edit_wordlist_del(self, event):
407
- index = self.grid_wordlist.GetFirstSelected()
408
- key = self.get_column_text(self.grid_wordlist, index, 0)
409
- if key in self.wlist.prohibited:
410
- msg = _("Can't delete internal variable {key}").format(key=key)
411
- self.edit_message(msg)
412
- return
413
- self.wlist.delete(key)
414
- self.autosave()
415
- self.refresh_grid_wordlist()
416
-
417
- def on_btn_edit_wordlist_edit(self, event):
418
- index = self.grid_wordlist.GetFirstSelected()
419
- if index >= 0:
420
- self.grid_wordlist.EditLabel(index)
421
-
422
- def on_btn_edit_content_add(self, event):
423
- skey = self.cur_skey
424
- if skey is None:
425
- return
426
- self.wlist.add_value(self.cur_skey, "---", 0)
427
- self.refresh_grid_content(skey, 0)
428
- self.autosave()
429
-
430
- def on_btn_edit_content_del(self, event):
431
- skey = self.cur_skey
432
- if skey is None:
433
- return
434
- index = self.grid_content.GetFirstSelected()
435
- if index >= 0:
436
- pass
437
- self.wlist.delete_value(skey, index)
438
- self.refresh_grid_content(skey, 0)
439
- self.autosave()
440
-
441
- def on_btn_edit_content_edit(self, event):
442
- skey = self.cur_skey
443
- if skey is None:
444
- return
445
- index = self.grid_content.GetFirstSelected()
446
- if index >= 0:
447
- self.cur_index = index
448
- self.grid_content.EditLabel(self.cur_index)
449
-
450
- def on_btn_edit_content_paste(self, event):
451
- skey = self.cur_skey
452
- if skey is None:
453
- return
454
- text_data = wx.TextDataObject()
455
- if wx.TheClipboard.Open():
456
- success = wx.TheClipboard.GetData(text_data)
457
- wx.TheClipboard.Close()
458
- if success:
459
- msg = text_data.GetText()
460
- if msg is not None and len(msg) > 0:
461
- lines = msg.splitlines()
462
- for entry in lines:
463
- self.wlist.add_value(skey, entry, 0)
464
- self.refresh_grid_content(skey, 0)
465
- self.autosave()
466
-
467
- def refresh_grid_wordlist(self):
468
- self.current_entry = None
469
- self.grid_wordlist.ClearAll()
470
- self.cur_skey = None
471
- self.grid_wordlist.InsertColumn(0, _("Name"))
472
- self.grid_wordlist.InsertColumn(1, _("Type"))
473
- self.grid_wordlist.InsertColumn(2, _("Index"))
474
- typestr = [_("Text"), _("CSV"), _("Counter")]
475
- for skey in self.wlist.content:
476
- index = self.grid_wordlist.InsertItem(
477
- self.grid_wordlist.GetItemCount(), skey
478
- )
479
- self.grid_wordlist.SetItem(index, 1, typestr[self.wlist.content[skey][0]])
480
- self.grid_wordlist.SetItem(index, 2, str(self.wlist.content[skey][1] - 2))
481
- self.grid_wordlist.SetColumnWidth(0, wx.LIST_AUTOSIZE)
482
- self.grid_wordlist.SetColumnWidth(1, wx.LIST_AUTOSIZE_USEHEADER)
483
- self.grid_wordlist.SetColumnWidth(2, wx.LIST_AUTOSIZE_USEHEADER)
484
-
485
- def get_column_text(self, grid, index, col):
486
- item = grid.GetItem(index, col)
487
- return item.GetText()
488
-
489
- def set_column_text(self, grid, index, col, value):
490
- # item = grid.GetItem(index)
491
- grid.SetItem(index, col, value)
492
-
493
- def refresh_grid_content(self, skey, current):
494
- self.cbo_index_single.Clear()
495
- choices = []
496
- selidx = 0
497
- self.grid_content.ClearAll()
498
- self.cur_skey = skey.lower()
499
- self.cur_index = None
500
- self.grid_content.InsertColumn(0, _("Content"))
501
-
502
- for idx in range(2, len(self.wlist.content[skey])):
503
- myidx = idx - 2
504
- s_entry = str(self.wlist.content[skey][idx])
505
- choices.append(f"{myidx:3d} - {s_entry[:10]}")
506
- index = self.grid_content.InsertItem(
507
- self.grid_content.GetItemCount(), s_entry
508
- )
509
- if idx == current + 2:
510
- selidx = current
511
- self.grid_content.SetItemTextColour(index, wx.RED)
512
- wsize = self.grid_content.GetSize()
513
- self.grid_content.SetColumnWidth(0, wsize[0] - 10)
514
- self.cbo_index_single.Set(choices)
515
- if selidx >= 0:
516
- self.cbo_index_single.SetSelection(selidx)
517
-
518
- def on_single_index(self, event):
519
- skey = self.cur_skey
520
- idx = self.cbo_index_single.GetSelection()
521
- if skey is None:
522
- return
523
- self.wlist.set_index(skey, idx)
524
- self.refresh_grid_content(skey, idx)
525
- # We need to refresh the main_index_column as well
526
- self.set_column_text(self.grid_wordlist, self.current_entry, 2, str(idx))
527
-
528
- def on_grid_wordlist(self, event):
529
- current_item = event.Index
530
- self.current_entry = current_item
531
- skey = self.get_column_text(self.grid_wordlist, current_item, 0).lower()
532
- try:
533
- current = int(self.get_column_text(self.grid_wordlist, current_item, 2))
534
- except ValueError:
535
- current = 0
536
-
537
- self.refresh_grid_content(skey, current)
538
- self.txt_pattern.SetValue(skey)
539
- event.Skip()
540
-
541
- def on_content_dblclick(self, event):
542
- index = self.grid_content.GetFirstSelected()
543
- if index >= 0:
544
- self.cbo_index_single.SetSelection(index)
545
- self.on_single_index(event)
546
-
547
- def on_grid_content(self, event):
548
- # Single Click
549
- event.Skip()
550
-
551
- def on_begin_edit_wordlist(self, event):
552
- index = self.grid_wordlist.GetFirstSelected()
553
- if index >= 0:
554
- # Is this a prevented value?
555
- skey = self.get_column_text(self.grid_wordlist, index, 0)
556
- self.to_save_wordlist = (skey, 0)
557
- if skey in self.wlist.prohibited:
558
- msg = _("Can't rename internal variable {key}").format(key=skey)
559
- self.edit_message(msg)
560
- self.to_save_wordlist = None
561
- event.Veto()
562
- return
563
- event.Allow()
564
-
565
- def on_end_edit_wordlist(self, event):
566
- if self.to_save_wordlist is None:
567
- self.edit_message(_("Update failed"))
568
- else:
569
- skey = self.to_save_wordlist[0]
570
- value = event.GetText().lower()
571
- # We need to replace it...
572
- self.wlist.rename_key(skey, value)
573
- self.autosave()
574
- self.cur_skey = value
575
- self.refresh_grid_content(value, 0)
576
- self.txt_pattern.SetValue(value)
577
-
578
- self.to_save_wordlist = None
579
- event.Allow()
580
-
581
- def on_begin_edit_content(self, event):
582
- index = self.grid_content.GetFirstSelected()
583
- if index >= 0:
584
- self.cur_index = index
585
- self.to_save_content = (self.cur_skey, self.cur_index)
586
- event.Allow()
587
-
588
- def on_end_edit_content(self, event):
589
- if self.to_save_content is None:
590
- self.edit_message(_("Update failed"))
591
- else:
592
- skey = self.to_save_content[0]
593
- index = self.to_save_content[1]
594
- value = event.GetText()
595
- self.wlist.set_value(skey, value, index)
596
- self.autosave()
597
- self.to_save_content = None
598
- event.Allow()
599
-
600
- def populate_gui(self):
601
- self.cbo_Index.Clear()
602
- self.cbo_Index.Enable(False)
603
- maxidx = -1
604
- self.grid_content.ClearAll()
605
- self.refresh_grid_wordlist()
606
- for skey in self.wlist.content:
607
- if self.wlist.content[skey][0] == 1: # CSV
608
- i = len(self.wlist.content[skey]) - 2
609
- if i > maxidx:
610
- maxidx = i
611
- if maxidx >= 0:
612
- for i in range(maxidx):
613
- self.cbo_Index.Append(str(i))
614
- self.cbo_Index.SetValue("0")
615
- self.cbo_Index.Enable(True)
616
- # Let's refresh the scene to acknowledge new changes
617
- self.context.signal("refresh_scene", "Scene")
618
-
619
- def edit_message(self, text):
620
- self.lbl_message.Label = text
621
- self.lbl_message.Refresh()
622
-
623
- def on_cbo_select(self, event):
624
- try:
625
- idx = int(self.cbo_Index.GetValue())
626
- except ValueError:
627
- idx = 0
628
- self.wlist.set_index(skey="@all", idx=idx)
629
- selidx = self.grid_wordlist.GetFirstSelected()
630
- if selidx < 0:
631
- selidx = 0
632
- self.refresh_grid_wordlist()
633
- self.grid_wordlist.Select(selidx, True)
634
- self.autosave()
635
-
636
- def on_btn_add(self, event):
637
- skey = self.txt_pattern.GetValue()
638
- if skey is not None and len(skey) > 0:
639
- if skey in self.wlist.content:
640
- self.wlist.delete(skey)
641
- self.wlist.add_value(skey, "---", 0)
642
- self.autosave()
643
- self.populate_gui()
644
- event.Skip()
645
-
646
- def on_patterntext_change(self, event):
647
- enab1 = False
648
- enab2 = False
649
- newname = self.txt_pattern.GetValue().lower()
650
- if len(newname) > 0:
651
- enab2 = True
652
- enab1 = True
653
- if newname in self.wlist.content:
654
- enab1 = False
655
- self.btn_add.Enable(enab1)
656
- self.btn_add_counter.Enable(enab1)
657
- self.btn_delete.Enable(enab2)
658
-
659
- def on_add_counter(self, event): # wxGlade: editWordlist.<event_handler>
660
- skey = self.txt_pattern.GetValue()
661
- if skey is not None and len(skey) > 0:
662
- if skey in self.wlist.content:
663
- self.wlist.delete(skey)
664
- self.wlist.add_value(skey, 1, 2)
665
- self.autosave()
666
- self.populate_gui()
667
- event.Skip()
668
-
669
- def on_btn_delete(self, event):
670
- skey = self.txt_pattern.GetValue()
671
- if skey is not None and len(skey) > 0:
672
- self.wlist.delete(skey)
673
- self.autosave()
674
- self.populate_gui()
675
- event.Skip()
676
-
677
- def on_backup(self, event):
678
- if self.wlist.default_filename is not None:
679
- self.wlist.save_data(self.wlist.default_filename)
680
- msg = _("Saved to ") + self.wlist.default_filename
681
- self.edit_message(msg)
682
- event.Skip()
683
-
684
- def on_restore(self, event):
685
- if self.wlist.default_filename is not None:
686
- self.wlist.load_data(self.wlist.default_filename)
687
- msg = _("Loaded from ") + self.wlist.default_filename
688
- self.edit_message(msg)
689
- self.populate_gui()
690
- event.Skip()
691
-
692
- @signal_listener("wordlist")
693
- def signal_wordlist(self, origin, *args):
694
- self.autosave()
695
- self.refresh_grid_wordlist()
696
-
697
-
698
- class ImportPanel(wx.Panel):
699
- def __init__(self, *args, context=None, **kwds):
700
- kwds["style"] = kwds.get("style", 0) | wx.TAB_TRAVERSAL
701
- wx.Panel.__init__(self, *args, **kwds)
702
- self.parent_panel = None
703
- self.context = context
704
- self.wlist = self.context.elements.mywordlist
705
- main_sizer = wx.BoxSizer(wx.VERTICAL)
706
- info_box = StaticBoxSizer(self, wx.ID_ANY, _("Import CSV"), wx.VERTICAL)
707
- sizer_csv = wx.BoxSizer(wx.HORIZONTAL)
708
- info_box.Add(sizer_csv, 1, wx.EXPAND, 0)
709
-
710
- label_1 = wx.StaticText(self, wx.ID_ANY, _("Import CSV-File"))
711
- sizer_csv.Add(label_1, 0, wx.ALIGN_CENTER_VERTICAL, 0)
712
-
713
- self.txt_filename = wx.TextCtrl(self, wx.ID_ANY, "")
714
- sizer_csv.Add(self.txt_filename, 1, wx.ALIGN_CENTER_VERTICAL, 0)
715
-
716
- self.btn_fileDialog = wx.Button(self, wx.ID_ANY, "...")
717
- self.btn_fileDialog.SetMinSize(dip_size(self, 23, 23))
718
- sizer_csv.Add(self.btn_fileDialog, 0, wx.ALIGN_CENTER_VERTICAL, 0)
719
-
720
- self.btn_import = wx.Button(self, wx.ID_ANY, _("Import CSV"))
721
- sizer_csv.Add(self.btn_import, 0, wx.ALIGN_CENTER_VERTICAL, 0)
722
-
723
- sizer_header = wx.BoxSizer(wx.HORIZONTAL)
724
- self.rbox_header = wx.RadioBox(
725
- self,
726
- wx.ID_ANY,
727
- _("What does the first row contain:"),
728
- choices=(
729
- _("Auto-Detect"),
730
- _("Contains Data"),
731
- _("Contains Variable-Names"),
732
- ),
733
- majorDimension=3,
734
- style=wx.RA_SPECIFY_COLS,
735
- )
736
- self.rbox_header.SetSelection(0)
737
- sizer_header.Add(self.rbox_header, 1, wx.EXPAND, 0)
738
- info_box.Add(sizer_header, 0, wx.EXPAND)
739
-
740
- self.text_preview = wx.TextCtrl(
741
- self, wx.ID_ANY, style=wx.TE_READONLY | wx.TE_MULTILINE
742
- )
743
-
744
- main_sizer.Add(info_box, 0, wx.EXPAND, 0)
745
- main_sizer.Add(self.text_preview, 1, wx.EXPAND, 0)
746
- self.SetSizer(main_sizer)
747
- self.Layout()
748
- self.btn_fileDialog.Bind(wx.EVT_BUTTON, self.on_btn_file)
749
- self.btn_import.Bind(wx.EVT_BUTTON, self.on_btn_import)
750
- self.txt_filename.Bind(wx.EVT_TEXT, self.on_filetext_change)
751
- self.btn_import.Enable(False)
752
-
753
- def set_parent(self, par_panel):
754
- self.parent_panel = par_panel
755
-
756
- def on_btn_file(self, event):
757
- myfile = ""
758
- mydlg = wx.FileDialog(
759
- self,
760
- message=_("Choose a csv-file"),
761
- wildcard="CSV-Files (*.csv)|*.csv|Text files (*.txt)|*.txt|All files (*.*)|*.*",
762
- style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST | wx.FD_PREVIEW,
763
- )
764
- if mydlg.ShowModal() == wx.ID_OK:
765
- # This returns a Python list of files that were selected.
766
- myfile = mydlg.GetPath()
767
- mydlg.Destroy()
768
- if myfile != "":
769
- self.txt_filename.SetValue(myfile)
770
- # self.on_btn_import(None)
771
-
772
- def import_csv(self, myfile):
773
- if os.path.exists(myfile):
774
- self.txt_filename.SetValue(myfile)
775
- force_header = None
776
- if self.rbox_header.GetSelection() == 1:
777
- force_header = False
778
- elif self.rbox_header.GetSelection() == 2:
779
- force_header = True
780
- ct, colcount, headers = self.wlist.load_csv_file(
781
- myfile, force_header=force_header
782
- )
783
- msg = _("Imported file, {col} fields, {row} rows").format(
784
- col=colcount, row=ct
785
- )
786
- if self.parent_panel is not None:
787
- self.parent_panel.edit_message(msg)
788
- self.parent_panel.populate_gui()
789
- return True
790
- return False
791
-
792
- def on_btn_import(self, event):
793
- myfile = self.txt_filename.GetValue()
794
- if os.path.exists(myfile):
795
- force_header = None
796
- if self.rbox_header.GetSelection() == 1:
797
- force_header = False
798
- elif self.rbox_header.GetSelection() == 2:
799
- force_header = True
800
- ct, colcount, headers = self.wlist.load_csv_file(
801
- myfile, force_header=force_header
802
- )
803
- msg = _("Imported file, {col} fields, {row} rows").format(
804
- col=colcount, row=ct
805
- )
806
- if self.parent_panel is not None:
807
- self.parent_panel.edit_message(msg)
808
- self.parent_panel.populate_gui()
809
-
810
- def on_filetext_change(self, event):
811
- myfile = self.txt_filename.GetValue()
812
- enab = False
813
- if os.path.exists(myfile):
814
- enab = True
815
- self.btn_import.Enable(enab)
816
- self.preview_it()
817
-
818
- def preview_it(self):
819
- filename = self.txt_filename.GetValue()
820
- buffer = ""
821
- if os.path.exists(filename):
822
- try:
823
- with open(filename, newline="") as csvfile:
824
- buffer = csvfile.read(1024)
825
- except (PermissionError, OSError, FileNotFoundError):
826
- pass
827
- self.text_preview.SetValue(buffer)
828
-
829
-
830
- class AboutPanel(wx.Panel):
831
- def __init__(self, *args, context=None, **kwds):
832
- kwds["style"] = kwds.get("style", 0) | wx.TAB_TRAVERSAL
833
- wx.Panel.__init__(self, *args, **kwds)
834
- self.context = context
835
- main_sizer = wx.BoxSizer(wx.VERTICAL)
836
- info_box = StaticBoxSizer(self, wx.ID_ANY, _("How to use..."), wx.VERTICAL)
837
- self.parent_panel = None
838
- s = self.context.asset("wordlist_howto")
839
- info_label = wx.TextCtrl(
840
- self, wx.ID_ANY, value=s, style=wx.TE_READONLY | wx.TE_MULTILINE
841
- )
842
- font = wx.Font(
843
- 10,
844
- wx.FONTFAMILY_TELETYPE,
845
- wx.FONTSTYLE_NORMAL,
846
- wx.FONTWEIGHT_NORMAL,
847
- )
848
- info_label.SetFont(font)
849
- info_label.SetBackgroundColour(self.GetBackgroundColour())
850
- info_box.Add(info_label, 1, wx.EXPAND, 0)
851
- main_sizer.Add(info_box, 1, wx.EXPAND, 0)
852
- self.SetSizer(main_sizer)
853
- self.Layout()
854
-
855
- def set_parent(self, par_panel):
856
- self.parent_panel = par_panel
857
-
858
-
859
- class WordlistEditor(MWindow):
860
- def __init__(self, *args, **kwds):
861
- super().__init__(500, 530, *args, **kwds)
862
-
863
- self.panel_editor = WordlistPanel(self, wx.ID_ANY, context=self.context)
864
- self.panel_import = ImportPanel(self, wx.ID_ANY, context=self.context)
865
- self.panel_about = AboutPanel(self, wx.ID_ANY, context=self.context)
866
-
867
- self.panel_editor.set_parent(self)
868
- self.panel_import.set_parent(self)
869
- self.panel_about.set_parent(self)
870
-
871
- _icon = wx.NullIcon
872
- _icon.CopyFromBitmap(icons8_curly_brackets.GetBitmap())
873
- self.SetIcon(_icon)
874
- self.notebook_main = wx.aui.AuiNotebook(
875
- self,
876
- -1,
877
- style=wx.aui.AUI_NB_TAB_EXTERNAL_MOVE
878
- | wx.aui.AUI_NB_SCROLL_BUTTONS
879
- | wx.aui.AUI_NB_TAB_SPLIT
880
- | wx.aui.AUI_NB_TAB_MOVE,
881
- )
882
- self.notebook_main.AddPage(self.panel_editor, _("Editing"))
883
- self.notebook_main.AddPage(self.panel_import, _("Import/Export"))
884
- self.notebook_main.AddPage(self.panel_about, _("How to use"))
885
- # begin wxGlade: Keymap.__set_properties
886
- self.DragAcceptFiles(True)
887
- self.Bind(wx.EVT_DROP_FILES, self.on_drop_file)
888
- self.SetTitle(_("Wordlist Editor"))
889
-
890
- def on_drop_file(self, event):
891
- """
892
- Drop file handler
893
- Accepts only a single file drop.
894
- """
895
- for pathname in event.GetFiles():
896
- if self.panel_import.import_csv(pathname):
897
- break
898
-
899
- def delegates(self):
900
- yield self.panel_editor
901
- yield self.panel_import
902
- yield self.panel_about
903
-
904
- @staticmethod
905
- def sub_register(kernel):
906
- kernel.register(
907
- "button/config/Wordlist",
908
- {
909
- "label": _("Wordlist Editor"),
910
- "icon": icons8_curly_brackets,
911
- "tip": _("Manages Wordlist-Entries"),
912
- "action": lambda v: kernel.console("window toggle Wordlist\n"),
913
- },
914
- )
915
-
916
- def window_open(self):
917
- pass
918
-
919
- def window_close(self):
920
- pass
921
-
922
- def edit_message(self, msg):
923
- self.panel_editor.edit_message(msg)
924
-
925
- def populate_gui(self):
926
- self.panel_editor.populate_gui()
927
-
928
- @staticmethod
929
- def submenu():
930
- # Suppress to avoid double menu-appearance
931
- return "Editing", "Variables + Wordlists", True
1
+ import os
2
+ import re
3
+
4
+ import wx
5
+ from wx import aui
6
+
7
+ from ..extra.encode_detect import EncodingDetectFile
8
+ from ..kernel import signal_listener
9
+ from .icons import (
10
+ get_default_icon_size,
11
+ icon_add_new,
12
+ icon_edit,
13
+ icon_trash,
14
+ icons8_circled_left,
15
+ icons8_circled_right,
16
+ icons8_curly_brackets,
17
+ icons8_paste,
18
+ )
19
+ from .mwindow import MWindow
20
+ from .wxutils import (
21
+ StaticBoxSizer,
22
+ TextCtrl,
23
+ dip_size,
24
+ wxButton,
25
+ wxCheckBox,
26
+ wxComboBox,
27
+ wxListCtrl,
28
+ wxRadioBox,
29
+ wxStaticBitmap,
30
+ wxStaticText,
31
+ )
32
+
33
+ _ = wx.GetTranslation
34
+
35
+
36
+ def register_panel_wordlist(window, context):
37
+ pane = (
38
+ aui.AuiPaneInfo()
39
+ .Left()
40
+ .MinSize(150, 25)
41
+ .FloatingSize(150, 35)
42
+ .Caption(_("Wordlist"))
43
+ .CaptionVisible(not context.pane_lock)
44
+ .Name("wordlist")
45
+ .Hide()
46
+ )
47
+ pane.dock_proportion = 225
48
+ pane.control = WordlistMiniPanel(window, wx.ID_ANY, context=context)
49
+ pane.submenu = "_50_" + _("Tools")
50
+ pane.helptext = _("Display wordlist advancement controls")
51
+ window.on_pane_create(pane)
52
+ context.register("pane/wordlist", pane)
53
+
54
+
55
+ class WordlistMiniPanel(wx.Panel):
56
+ def __init__(self, *args, context=None, **kwds):
57
+ kwds["style"] = kwds.get("style", 0) | wx.TAB_TRAVERSAL
58
+ wx.Panel.__init__(self, *args, **kwds)
59
+ self.context = context
60
+ self.context.themes.set_window_colors(self)
61
+ self.SetHelpText("wordlist")
62
+
63
+ main_sizer = wx.BoxSizer(wx.HORIZONTAL)
64
+ self.button_edit = wxButton(self, wx.ID_ANY, _("Edit"))
65
+ self.button_edit.SetBitmap(
66
+ icons8_curly_brackets.GetBitmap(resize=0.5 * get_default_icon_size(self.context))
67
+ )
68
+ self.button_edit.SetToolTip(_("Manages Wordlist-Entries"))
69
+
70
+ self.button_next = wxButton(self, wx.ID_ANY, _("Next"))
71
+ self.button_next.SetBitmap(
72
+ icons8_circled_right.GetBitmap(resize=0.5 * get_default_icon_size(self.context))
73
+ )
74
+ self.button_next.SetToolTip(
75
+ _("Wordlist: go to next page (right-click to next entry)")
76
+ )
77
+
78
+ self.button_prev = wxButton(self, wx.ID_ANY, _("Prev"))
79
+ self.button_prev.SetBitmap(
80
+ icons8_circled_left.GetBitmap(resize=0.5 * get_default_icon_size(self.context))
81
+ )
82
+ self.button_prev.SetToolTip(
83
+ _("Wordlist: go to previous page (right-click to previous entry)")
84
+ )
85
+
86
+ main_sizer.Add(self.button_prev, 1, wx.EXPAND, 0)
87
+ main_sizer.Add(self.button_edit, 1, wx.EXPAND, 0)
88
+ main_sizer.Add(self.button_next, 1, wx.EXPAND, 0)
89
+
90
+ self.button_next.Bind(wx.EVT_BUTTON, self.on_button_next_page)
91
+ self.button_prev.Bind(wx.EVT_BUTTON, self.on_button_prev_page)
92
+ self.button_next.Bind(wx.EVT_RIGHT_DOWN, self.on_button_next)
93
+ self.button_prev.Bind(wx.EVT_RIGHT_DOWN, self.on_button_prev)
94
+ self.button_edit.Bind(wx.EVT_BUTTON, self.on_button_edit)
95
+
96
+ self.SetSizer(main_sizer)
97
+ self.Layout()
98
+
99
+ def establish_max_delta(self):
100
+ # try to establish the needed delta to satisfy all variables...
101
+ deltamin = 0
102
+ deltamax = 0
103
+ for node in self.context.elements.elems():
104
+ sample = ""
105
+ if node.type == "elem text":
106
+ if node.text is not None:
107
+ sample = str(node.text)
108
+ elif node.type == "elem path" and hasattr(node, "mktext"):
109
+ if node.mktext is not None:
110
+ sample = str(node.mktext)
111
+ if sample == "":
112
+ continue
113
+ # we can be rather agnostic on the individual variable,
114
+ # we are interested in the highest {variable#+offset} pattern
115
+ brackets = re.compile(r"\{[^}]+\}")
116
+ for bracketed_key in brackets.findall(sample):
117
+ # print(f"Key found: {bracketed_key}")
118
+ key = bracketed_key[1:-1].lower().strip()
119
+ relative = 0
120
+ pos = key.find("#")
121
+ if pos > 0: # Needs to be after first character
122
+ # Process offset modification.
123
+ index_string = key[pos + 1 :]
124
+ key = key[:pos].strip()
125
+ if index_string.startswith("+") or index_string.startswith("-"):
126
+ try:
127
+ # This covers +x, -x, x
128
+ relative = int(index_string)
129
+ except ValueError:
130
+ relative = 0
131
+ deltamin = min(relative, deltamin)
132
+ deltamax = max(relative, deltamax)
133
+
134
+ return deltamax
135
+
136
+ def on_button_edit(self, event):
137
+ self.context("window toggle Wordlist\n")
138
+
139
+ def on_button_prev(self, event):
140
+ self.context.elements.wordlist_advance(-1)
141
+
142
+ def on_button_prev_page(self, event):
143
+ delta = self.establish_max_delta()
144
+ self.context.elements.wordlist_advance(-1 - delta)
145
+
146
+ def on_button_next(self, event):
147
+ self.context.elements.wordlist_advance(+1)
148
+
149
+ def on_button_next_page(self, event):
150
+ delta = self.establish_max_delta()
151
+ self.context.elements.wordlist_advance(+1 + delta)
152
+
153
+
154
+ class WordlistPanel(wx.Panel):
155
+ def __init__(self, *args, context=None, **kwds):
156
+ kwds["style"] = kwds.get("style", 0) | wx.TAB_TRAVERSAL
157
+ wx.Panel.__init__(self, *args, **kwds)
158
+ self.context = context
159
+ self.context.themes.set_window_colors(self)
160
+ self.SetHelpText("wordlist")
161
+ self.parent_panel = None
162
+ self.wlist = self.context.elements.mywordlist
163
+ self.current_entry = None
164
+ # For Grid editing
165
+ self.cur_skey = None
166
+ self.cur_index = None
167
+ self.to_save_content = None
168
+ self.to_save_wordlist = None
169
+
170
+ self.context.setting(bool, "wordlist_autosave", False)
171
+
172
+ sizer_main = wx.BoxSizer(wx.VERTICAL)
173
+ sizer_grids = wx.BoxSizer(wx.HORIZONTAL)
174
+ sizer_grid_left = wx.BoxSizer(wx.VERTICAL)
175
+ sizer_grid_right = wx.BoxSizer(wx.VERTICAL)
176
+ sizer_grids.Add(sizer_grid_left, 1, wx.EXPAND, 0)
177
+ sizer_grids.Add(sizer_grid_right, 1, wx.EXPAND, 0)
178
+
179
+ sizer_index_left = wx.BoxSizer(wx.HORIZONTAL)
180
+ sizer_grid_left.Add(sizer_index_left, 0, wx.EXPAND, 0)
181
+
182
+ label_2 = wxStaticText(self, wx.ID_ANY, _("Start Index for CSV-based data:"))
183
+ sizer_index_left.Add(label_2, 0, wx.ALIGN_CENTER_VERTICAL, 0)
184
+ self.cbo_Index = wxComboBox(
185
+ self, wx.ID_ANY, choices=[], style=wx.CB_DROPDOWN | wx.CB_READONLY
186
+ )
187
+ sizer_index_left.Add(self.cbo_Index, 0, wx.ALIGN_CENTER_VERTICAL, 0)
188
+
189
+ self.grid_wordlist = wxListCtrl(
190
+ self,
191
+ wx.ID_ANY,
192
+ style=wx.LC_HRULES
193
+ | wx.LC_REPORT
194
+ | wx.LC_VRULES
195
+ | wx.LC_SINGLE_SEL
196
+ | wx.LC_EDIT_LABELS,
197
+ context=self.context,
198
+ list_name="list_wordlist"
199
+ )
200
+ sizer_grid_left.Add(self.grid_wordlist, 1, wx.EXPAND, 0)
201
+
202
+ self.grid_content = wxListCtrl(
203
+ self,
204
+ wx.ID_ANY,
205
+ style=wx.LC_HRULES
206
+ | wx.LC_REPORT
207
+ | wx.LC_VRULES
208
+ | wx.LC_SINGLE_SEL
209
+ | wx.LC_EDIT_LABELS,
210
+ context=self.context,
211
+ list_name="list_wordlist_content",
212
+ )
213
+
214
+ sizer_edit_wordlist_buttons = wx.BoxSizer(wx.HORIZONTAL)
215
+ sizer_edit_content_buttons = wx.BoxSizer(wx.HORIZONTAL)
216
+
217
+ dummylabel = wxStaticText(self, wx.ID_ANY, " ")
218
+ sizer_index_left.Add(dummylabel, 1, wx.ALIGN_CENTER_VERTICAL, 0)
219
+ sizer_index_left.Add(
220
+ sizer_edit_wordlist_buttons, 0, wx.ALIGN_CENTER_VERTICAL, 0
221
+ )
222
+ testsize = dip_size(self, 20, 20)
223
+ icon_size = testsize[0]
224
+
225
+ self.btn_edit_wordlist_del = wxStaticBitmap(
226
+ self, wx.ID_ANY, size=dip_size(self, 25, 25)
227
+ )
228
+ self.btn_edit_wordlist_edit = wxStaticBitmap(
229
+ self, wx.ID_ANY, size=dip_size(self, 25, 25)
230
+ )
231
+ self.btn_edit_content_add = wxStaticBitmap(
232
+ self, wx.ID_ANY, size=dip_size(self, 25, 25)
233
+ )
234
+ self.btn_edit_content_del = wxStaticBitmap(
235
+ self, wx.ID_ANY, size=dip_size(self, 25, 25)
236
+ )
237
+ self.btn_edit_content_edit = wxStaticBitmap(
238
+ self, wx.ID_ANY, size=dip_size(self, 25, 25)
239
+ )
240
+ self.btn_edit_content_paste = wxStaticBitmap(
241
+ self, wx.ID_ANY, size=dip_size(self, 25, 25)
242
+ )
243
+ # Circumvent a WXPython bug at high resolutions under Windows
244
+ bmp = icon_trash.GetBitmap(resize=icon_size, buffer=1)
245
+ self.btn_edit_wordlist_del.SetBitmap(bmp)
246
+ testsize = self.btn_edit_wordlist_del.GetBitmap().Size
247
+ if testsize[0] != icon_size:
248
+ icon_size = int(icon_size * icon_size / testsize[0])
249
+
250
+ self.btn_edit_wordlist_del.SetBitmap(
251
+ icon_trash.GetBitmap(resize=icon_size, buffer=1)
252
+ )
253
+ self.btn_edit_wordlist_edit.SetBitmap(
254
+ icon_edit.GetBitmap(resize=icon_size, buffer=1)
255
+ )
256
+ self.btn_edit_content_add.SetBitmap(
257
+ icon_add_new.GetBitmap(resize=icon_size, buffer=1)
258
+ )
259
+ self.btn_edit_content_del.SetBitmap(
260
+ icon_trash.GetBitmap(resize=icon_size, buffer=1)
261
+ )
262
+ self.btn_edit_content_edit.SetBitmap(
263
+ icon_edit.GetBitmap(resize=icon_size, buffer=1)
264
+ )
265
+ self.btn_edit_content_paste.SetBitmap(
266
+ icons8_paste.GetBitmap(resize=icon_size, buffer=1)
267
+ )
268
+
269
+ self.btn_edit_wordlist_del.SetToolTip(_("Delete the current variable"))
270
+ self.btn_edit_wordlist_edit.SetToolTip(
271
+ _("Edit the name of the active variable")
272
+ )
273
+ self.btn_edit_content_add.SetToolTip(
274
+ _("Add a new entry for the active variable")
275
+ )
276
+ self.btn_edit_content_del.SetToolTip(
277
+ _("Delete the current entry for the active variable")
278
+ )
279
+ self.btn_edit_content_edit.SetToolTip(
280
+ _("Edit the current entry for the active variable")
281
+ )
282
+ self.btn_edit_content_paste.SetToolTip(
283
+ _(
284
+ "Paste the clipboard as new entries for the active variable, any line as new entry"
285
+ )
286
+ )
287
+ minsize = 23
288
+ self.btn_edit_wordlist_del.SetMinSize(dip_size(self, minsize, minsize))
289
+ self.btn_edit_wordlist_edit.SetMinSize(dip_size(self, minsize, minsize))
290
+ self.btn_edit_content_add.SetMinSize(dip_size(self, minsize, minsize))
291
+ self.btn_edit_content_del.SetMinSize(dip_size(self, minsize, minsize))
292
+ self.btn_edit_content_edit.SetMinSize(dip_size(self, minsize, minsize))
293
+ self.btn_edit_content_paste.SetMinSize(dip_size(self, minsize, minsize))
294
+
295
+ sizer_edit_wordlist_buttons.Add(self.btn_edit_wordlist_del, 0, wx.EXPAND, 0)
296
+ sizer_edit_wordlist_buttons.Add(self.btn_edit_wordlist_edit, 0, wx.EXPAND, 0)
297
+
298
+ sizer_edit_content_buttons.Add(self.btn_edit_content_add, 0, wx.EXPAND, 0)
299
+ sizer_edit_content_buttons.Add(self.btn_edit_content_del, 0, wx.EXPAND, 0)
300
+ sizer_edit_content_buttons.Add(self.btn_edit_content_edit, 0, wx.EXPAND, 0)
301
+ sizer_edit_content_buttons.Add(self.btn_edit_content_paste, 0, wx.EXPAND, 0)
302
+
303
+ sizer_index_right = wx.BoxSizer(wx.HORIZONTAL)
304
+ label_2 = wxStaticText(self, wx.ID_ANY, _("Start Index for field:"))
305
+ sizer_index_right.Add(label_2, 0, wx.ALIGN_CENTER_VERTICAL, 0)
306
+ self.cbo_index_single = wxComboBox(
307
+ self, wx.ID_ANY, choices=[], style=wx.CB_DROPDOWN | wx.CB_READONLY
308
+ )
309
+ sizer_index_right.Add(self.cbo_index_single, 1, wx.ALIGN_CENTER_VERTICAL, 0)
310
+
311
+ dummylabel = wxStaticText(self, wx.ID_ANY, " ")
312
+ sizer_index_right.Add(dummylabel, 1, wx.ALIGN_CENTER_VERTICAL, 0)
313
+ sizer_index_right.Add(
314
+ sizer_edit_content_buttons, 0, wx.ALIGN_CENTER_VERTICAL, 0
315
+ )
316
+
317
+ sizer_grid_right.Add(sizer_index_right, 0, wx.EXPAND, 0)
318
+ sizer_grid_right.Add(self.grid_content, 1, wx.EXPAND, 0)
319
+
320
+ self.txt_pattern = TextCtrl(self, wx.ID_ANY, "")
321
+
322
+ sizer_buttons = wx.BoxSizer(wx.HORIZONTAL)
323
+ sizer_buttons.Add(self.txt_pattern, 1, wx.ALIGN_CENTER_VERTICAL, 0)
324
+
325
+ self.btn_add = wxButton(self, wx.ID_ANY, _("Add Text"))
326
+ self.btn_add.SetToolTip(_("Add another wordlist entry"))
327
+ sizer_buttons.Add(self.btn_add, 0, 0, 0)
328
+
329
+ self.btn_add_counter = wxButton(self, wx.ID_ANY, _("Add Counter"))
330
+ sizer_buttons.Add(self.btn_add_counter, 0, 0, 0)
331
+
332
+ self.btn_delete = wxButton(self, wx.ID_ANY, _("Delete"))
333
+ self.btn_delete.SetToolTip(_("Delete the current wordlist entry"))
334
+ sizer_buttons.Add(self.btn_delete, 0, 0, 0)
335
+
336
+ sizer_exit = wx.BoxSizer(wx.HORIZONTAL)
337
+
338
+ self.btn_backup = wxButton(self, wx.ID_ANY, _("Backup Wordlist"))
339
+ self.btn_backup.SetToolTip(_("Save current wordlist to disk"))
340
+ sizer_exit.Add(self.btn_backup, 0, wx.ALIGN_CENTER_VERTICAL, 0)
341
+
342
+ self.btn_restore = wxButton(self, wx.ID_ANY, _("Restore Wordlist"))
343
+ self.btn_restore.SetToolTip(_("Load wordlist from disk"))
344
+ sizer_exit.Add(self.btn_restore, 0, wx.ALIGN_CENTER_VERTICAL, 0)
345
+
346
+ self.check_autosave = wxCheckBox(self, wx.ID_ANY, _("Autosave"))
347
+ self.check_autosave.SetToolTip(
348
+ _("All changes to the wordlist will be saved immediately")
349
+ )
350
+ sizer_exit.Add(self.check_autosave, 0, wx.ALIGN_CENTER_VERTICAL, 0)
351
+ self.check_autosave.SetValue(self.context.wordlist_autosave)
352
+
353
+ self.lbl_message = wxStaticText(self, wx.ID_ANY, "")
354
+ sizer_exit.Add(self.lbl_message, 1, wx.ALIGN_CENTER_VERTICAL, 0)
355
+
356
+ sizer_lower = wx.BoxSizer(wx.HORIZONTAL)
357
+ sizer_lower.Add(sizer_buttons, 1, wx.ALL, 0)
358
+ sizer_lower.Add(sizer_exit, 1, wx.ALL, 0)
359
+
360
+ sizer_main.Add(sizer_grids, 1, wx.EXPAND, 0)
361
+ sizer_main.Add(sizer_lower, 0, wx.ALL, 0)
362
+
363
+ self.SetSizer(sizer_main)
364
+ self.Layout()
365
+ self._set_logic()
366
+ self.populate_gui()
367
+
368
+ def _set_logic(self):
369
+ self.btn_add.Bind(wx.EVT_BUTTON, self.on_btn_add)
370
+ self.btn_add_counter.Bind(wx.EVT_BUTTON, self.on_add_counter)
371
+ self.btn_delete.Bind(wx.EVT_BUTTON, self.on_btn_delete)
372
+ self.btn_backup.Bind(wx.EVT_BUTTON, self.on_backup)
373
+ self.btn_restore.Bind(wx.EVT_BUTTON, self.on_restore)
374
+ self.txt_pattern.Bind(wx.EVT_TEXT, self.on_patterntext_change)
375
+ self.cbo_Index.Bind(wx.EVT_COMBOBOX, self.on_cbo_select)
376
+ self.grid_wordlist.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_grid_wordlist)
377
+ self.grid_wordlist.Bind(
378
+ wx.EVT_LIST_BEGIN_LABEL_EDIT, self.on_begin_edit_wordlist
379
+ )
380
+ self.grid_wordlist.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.on_end_edit_wordlist)
381
+ self.grid_content.Bind(wx.EVT_LIST_ITEM_SELECTED, self.on_grid_content)
382
+ self.grid_content.Bind(wx.EVT_LIST_BEGIN_LABEL_EDIT, self.on_begin_edit_content)
383
+ self.grid_content.Bind(wx.EVT_LIST_END_LABEL_EDIT, self.on_end_edit_content)
384
+ self.grid_content.Bind(wx.EVT_LEFT_DCLICK, self.on_content_dblclick)
385
+ self.check_autosave.Bind(wx.EVT_CHECKBOX, self.on_checkbox_autosave)
386
+
387
+ self.btn_edit_wordlist_del.Bind(wx.EVT_LEFT_DOWN, self.on_btn_edit_wordlist_del)
388
+ self.btn_edit_wordlist_edit.Bind(
389
+ wx.EVT_LEFT_DOWN, self.on_btn_edit_wordlist_edit
390
+ )
391
+ self.btn_edit_content_add.Bind(wx.EVT_LEFT_DOWN, self.on_btn_edit_content_add)
392
+ self.btn_edit_content_del.Bind(wx.EVT_LEFT_DOWN, self.on_btn_edit_content_del)
393
+ self.btn_edit_content_edit.Bind(wx.EVT_LEFT_DOWN, self.on_btn_edit_content_edit)
394
+ self.btn_edit_content_paste.Bind(
395
+ wx.EVT_LEFT_DOWN, self.on_btn_edit_content_paste
396
+ )
397
+ self.cbo_index_single.Bind(wx.EVT_COMBOBOX, self.on_single_index)
398
+ # Key handler for F2
399
+ self.grid_content.Bind(wx.EVT_CHAR, self.on_key_grid)
400
+ self.grid_wordlist.Bind(wx.EVT_CHAR, self.on_key_grid)
401
+
402
+ def set_parent(self, par_panel):
403
+ self.parent_panel = par_panel
404
+
405
+ def pane_show(self):
406
+ self.grid_wordlist.load_column_widths()
407
+ self.grid_content.load_column_widths()
408
+ self.populate_gui()
409
+ self.grid_wordlist.SetFocus()
410
+
411
+ def pane_hide(self):
412
+ self.grid_wordlist.save_column_widths()
413
+ self.grid_content.save_column_widths()
414
+
415
+ def autosave(self):
416
+ if self.check_autosave.GetValue():
417
+ if self.wlist.default_filename is not None:
418
+ self.wlist.save_data(self.wlist.default_filename)
419
+ msg = _("Saved to ") + self.wlist.default_filename
420
+ self.edit_message(msg)
421
+
422
+ def on_key_grid(self, event):
423
+ grid = event.GetEventObject()
424
+ key = event.GetKeyCode()
425
+ if key == wx.WXK_F2:
426
+ index = grid.GetFirstSelected()
427
+ if index >= 0:
428
+ grid.EditLabel(index)
429
+ # that consumes the key
430
+ return
431
+ # Let's make sure the keystroke is processed further
432
+ event.Skip()
433
+
434
+ def on_checkbox_autosave(self, event):
435
+ self.context.wordlist_autosave = self.check_autosave.GetValue()
436
+ event.Skip()
437
+
438
+ def on_btn_edit_wordlist_del(self, event):
439
+ index = self.grid_wordlist.GetFirstSelected()
440
+ key = self.get_column_text(self.grid_wordlist, index, 0)
441
+ if key in self.wlist.prohibited:
442
+ msg = _("Can't delete internal variable {key}").format(key=key)
443
+ self.edit_message(msg)
444
+ return
445
+ self.wlist.delete(key)
446
+ self.autosave()
447
+ self.refresh_grid_wordlist()
448
+
449
+ def on_btn_edit_wordlist_edit(self, event):
450
+ index = self.grid_wordlist.GetFirstSelected()
451
+ if index >= 0:
452
+ self.grid_wordlist.EditLabel(index)
453
+
454
+ def on_btn_edit_content_add(self, event):
455
+ skey = self.cur_skey
456
+ if skey is None:
457
+ return
458
+ self.wlist.add_value(self.cur_skey, "---", 0)
459
+ self.refresh_grid_content(skey, 0)
460
+ self.autosave()
461
+
462
+ def on_btn_edit_content_del(self, event):
463
+ skey = self.cur_skey
464
+ if skey is None:
465
+ return
466
+ index = self.grid_content.GetFirstSelected()
467
+ if index >= 0:
468
+ pass
469
+ self.wlist.delete_value(skey, index)
470
+ self.refresh_grid_content(skey, 0)
471
+ self.autosave()
472
+
473
+ def on_btn_edit_content_edit(self, event):
474
+ skey = self.cur_skey
475
+ if skey is None:
476
+ return
477
+ index = self.grid_content.GetFirstSelected()
478
+ if index >= 0:
479
+ self.cur_index = index
480
+ self.grid_content.EditLabel(self.cur_index)
481
+
482
+ def on_btn_edit_content_paste(self, event):
483
+ skey = self.cur_skey
484
+ if skey is None:
485
+ return
486
+ text_data = wx.TextDataObject()
487
+ success = False
488
+ if wx.TheClipboard.Open():
489
+ success = wx.TheClipboard.GetData(text_data)
490
+ wx.TheClipboard.Close()
491
+ if not success:
492
+ return
493
+ msg = text_data.GetText()
494
+ if msg is not None and len(msg) > 0:
495
+ lines = msg.splitlines()
496
+ for entry in lines:
497
+ self.wlist.add_value(skey, entry, 0)
498
+ self.refresh_grid_content(skey, 0)
499
+ self.autosave()
500
+
501
+ def refresh_grid_wordlist(self):
502
+ self.current_entry = None
503
+ self.grid_wordlist.ClearAll()
504
+ self.cur_skey = None
505
+ self.grid_wordlist.InsertColumn(0, _("Name"))
506
+ self.grid_wordlist.InsertColumn(1, _("Type"))
507
+ self.grid_wordlist.InsertColumn(2, _("Index"))
508
+ typestr = [_("Text"), _("CSV"), _("Counter")]
509
+ for skey in self.wlist.content:
510
+ index = self.grid_wordlist.InsertItem(
511
+ self.grid_wordlist.GetItemCount(), skey
512
+ )
513
+ self.grid_wordlist.SetItem(index, 1, typestr[self.wlist.content[skey][0]])
514
+ self.grid_wordlist.SetItem(index, 2, str(self.wlist.content[skey][1] - 2))
515
+ self.grid_wordlist.SetColumnWidth(0, wx.LIST_AUTOSIZE)
516
+ self.grid_wordlist.SetColumnWidth(1, wx.LIST_AUTOSIZE_USEHEADER)
517
+ self.grid_wordlist.SetColumnWidth(2, wx.LIST_AUTOSIZE_USEHEADER)
518
+ self.grid_wordlist.resize_columns()
519
+
520
+ def get_column_text(self, grid, index, col):
521
+ item = grid.GetItem(index, col)
522
+ return item.GetText()
523
+
524
+ def set_column_text(self, grid, index, col, value):
525
+ # item = grid.GetItem(index)
526
+ grid.SetItem(index, col, value)
527
+
528
+ def refresh_grid_content(self, skey, current):
529
+ self.cbo_index_single.Clear()
530
+ choices = []
531
+ selidx = 0
532
+ self.grid_content.ClearAll()
533
+ self.cur_skey = skey.lower()
534
+ self.cur_index = None
535
+ self.grid_content.InsertColumn(0, _("Content"))
536
+
537
+ for idx in range(2, len(self.wlist.content[skey])):
538
+ myidx = idx - 2
539
+ s_entry = str(self.wlist.content[skey][idx])
540
+ choices.append(f"{myidx:3d} - {s_entry[:10]}")
541
+ index = self.grid_content.InsertItem(
542
+ self.grid_content.GetItemCount(), s_entry
543
+ )
544
+ if idx == current + 2:
545
+ selidx = current
546
+ self.grid_content.SetItemTextColour(index, wx.RED)
547
+ wsize = self.grid_content.GetSize()
548
+ self.grid_content.SetColumnWidth(0, wsize[0] - 10)
549
+ self.grid_content.resize_columns()
550
+
551
+ self.cbo_index_single.Set(choices)
552
+ if selidx >= 0:
553
+ self.cbo_index_single.SetSelection(selidx)
554
+
555
+ def on_single_index(self, event):
556
+ skey = self.cur_skey
557
+ idx = self.cbo_index_single.GetSelection()
558
+ if skey is None:
559
+ return
560
+ self.wlist.set_index(skey, idx)
561
+ self.refresh_grid_content(skey, idx)
562
+ # We need to refresh the main_index_column as well
563
+ self.set_column_text(self.grid_wordlist, self.current_entry, 2, str(idx))
564
+
565
+ def on_grid_wordlist(self, event):
566
+ current_item = event.Index
567
+ self.current_entry = current_item
568
+ skey = self.get_column_text(self.grid_wordlist, current_item, 0).lower()
569
+ try:
570
+ current = int(self.get_column_text(self.grid_wordlist, current_item, 2))
571
+ except ValueError:
572
+ current = 0
573
+
574
+ self.refresh_grid_content(skey, current)
575
+ self.txt_pattern.SetValue(skey)
576
+ event.Skip()
577
+
578
+ def on_content_dblclick(self, event):
579
+ index = self.grid_content.GetFirstSelected()
580
+ if index >= 0:
581
+ self.cbo_index_single.SetSelection(index)
582
+ self.on_single_index(event)
583
+
584
+ def on_grid_content(self, event):
585
+ # Single Click
586
+ event.Skip()
587
+
588
+ def on_begin_edit_wordlist(self, event):
589
+ index = self.grid_wordlist.GetFirstSelected()
590
+ if index >= 0:
591
+ # Is this a prevented value?
592
+ skey = self.get_column_text(self.grid_wordlist, index, 0)
593
+ self.to_save_wordlist = (skey, 0)
594
+ if skey in self.wlist.prohibited:
595
+ msg = _("Can't rename internal variable {key}").format(key=skey)
596
+ self.edit_message(msg)
597
+ self.to_save_wordlist = None
598
+ event.Veto()
599
+ return
600
+ event.Allow()
601
+
602
+ def on_end_edit_wordlist(self, event):
603
+ if self.to_save_wordlist is None:
604
+ self.edit_message(_("Update failed"))
605
+ else:
606
+ skey = self.to_save_wordlist[0]
607
+ value = event.GetText().lower()
608
+ # We need to replace it...
609
+ self.wlist.rename_key(skey, value)
610
+ self.autosave()
611
+ self.cur_skey = value
612
+ self.refresh_grid_content(value, 0)
613
+ self.txt_pattern.SetValue(value)
614
+
615
+ self.to_save_wordlist = None
616
+ event.Allow()
617
+
618
+ def on_begin_edit_content(self, event):
619
+ index = self.grid_content.GetFirstSelected()
620
+ if index >= 0:
621
+ self.cur_index = index
622
+ self.to_save_content = (self.cur_skey, self.cur_index)
623
+ event.Allow()
624
+
625
+ def on_end_edit_content(self, event):
626
+ if self.to_save_content is None:
627
+ self.edit_message(_("Update failed"))
628
+ else:
629
+ skey = self.to_save_content[0]
630
+ index = self.to_save_content[1]
631
+ value = event.GetText()
632
+ self.wlist.set_value(skey, value, index)
633
+ self.autosave()
634
+ self.to_save_content = None
635
+ event.Allow()
636
+
637
+ def populate_gui(self):
638
+ self.cbo_Index.Clear()
639
+ self.cbo_Index.Enable(False)
640
+ maxidx = -1
641
+ self.grid_content.ClearAll()
642
+ self.refresh_grid_wordlist()
643
+ for skey in self.wlist.content:
644
+ if self.wlist.content[skey][0] == 1: # CSV
645
+ i = len(self.wlist.content[skey]) - 2
646
+ if i > maxidx:
647
+ maxidx = i
648
+ if maxidx >= 0:
649
+ for i in range(maxidx):
650
+ self.cbo_Index.Append(str(i))
651
+ self.cbo_Index.SetValue("0")
652
+ self.cbo_Index.Enable(True)
653
+ # Let's refresh the scene to acknowledge new changes
654
+ self.context.signal("refresh_scene", "Scene")
655
+
656
+ def edit_message(self, text):
657
+ self.lbl_message.Label = text
658
+ self.lbl_message.Refresh()
659
+
660
+ def on_cbo_select(self, event):
661
+ try:
662
+ idx = int(self.cbo_Index.GetValue())
663
+ except ValueError:
664
+ idx = 0
665
+ self.wlist.set_index(skey="@all", idx=idx)
666
+ selidx = self.grid_wordlist.GetFirstSelected()
667
+ if selidx < 0:
668
+ selidx = 0
669
+ self.refresh_grid_wordlist()
670
+ self.grid_wordlist.Select(selidx, True)
671
+ self.autosave()
672
+
673
+ def on_btn_add(self, event):
674
+ skey = self.txt_pattern.GetValue()
675
+ if skey is not None and len(skey) > 0:
676
+ if skey in self.wlist.content:
677
+ self.wlist.delete(skey)
678
+ self.wlist.add_value(skey, "---", 0)
679
+ self.autosave()
680
+ self.populate_gui()
681
+ event.Skip()
682
+
683
+ def on_patterntext_change(self, event):
684
+ enab1 = False
685
+ enab2 = False
686
+ newname = self.txt_pattern.GetValue().lower()
687
+ if len(newname) > 0:
688
+ enab2 = True
689
+ enab1 = True
690
+ if newname in self.wlist.content:
691
+ enab1 = False
692
+ self.btn_add.Enable(enab1)
693
+ self.btn_add_counter.Enable(enab1)
694
+ self.btn_delete.Enable(enab2)
695
+
696
+ def on_add_counter(self, event): # wxGlade: editWordlist.<event_handler>
697
+ skey = self.txt_pattern.GetValue()
698
+ if skey is not None and len(skey) > 0:
699
+ if skey in self.wlist.content:
700
+ self.wlist.delete(skey)
701
+ self.wlist.add_value(skey, 1, 2)
702
+ self.autosave()
703
+ self.populate_gui()
704
+ event.Skip()
705
+
706
+ def on_btn_delete(self, event):
707
+ skey = self.txt_pattern.GetValue()
708
+ if skey is not None and len(skey) > 0:
709
+ self.wlist.delete(skey)
710
+ self.autosave()
711
+ self.populate_gui()
712
+ event.Skip()
713
+
714
+ def on_backup(self, event):
715
+ if self.wlist.default_filename is not None:
716
+ self.wlist.save_data(self.wlist.default_filename)
717
+ msg = _("Saved to ") + self.wlist.default_filename
718
+ self.edit_message(msg)
719
+ event.Skip()
720
+
721
+ def on_restore(self, event):
722
+ if self.wlist.default_filename is not None:
723
+ self.wlist.load_data(self.wlist.default_filename)
724
+ msg = _("Loaded from ") + self.wlist.default_filename
725
+ self.edit_message(msg)
726
+ self.populate_gui()
727
+ event.Skip()
728
+
729
+ @signal_listener("wordlist")
730
+ def signal_wordlist(self, origin, *args):
731
+ self.autosave()
732
+ self.refresh_grid_wordlist()
733
+
734
+
735
+ class ImportPanel(wx.Panel):
736
+ def __init__(self, *args, context=None, **kwds):
737
+ kwds["style"] = kwds.get("style", 0) | wx.TAB_TRAVERSAL
738
+ wx.Panel.__init__(self, *args, **kwds)
739
+ self.parent_panel = None
740
+ self.context = context
741
+ self.context.themes.set_window_colors(self)
742
+ self.wlist = self.context.elements.mywordlist
743
+ main_sizer = wx.BoxSizer(wx.VERTICAL)
744
+ info_box = StaticBoxSizer(self, wx.ID_ANY, _("Import CSV"), wx.VERTICAL)
745
+ sizer_csv = wx.BoxSizer(wx.HORIZONTAL)
746
+ info_box.Add(sizer_csv, 1, wx.EXPAND, 0)
747
+
748
+ label_1 = wxStaticText(self, wx.ID_ANY, _("Import CSV-File"))
749
+ sizer_csv.Add(label_1, 0, wx.ALIGN_CENTER_VERTICAL, 0)
750
+
751
+ self.txt_filename = TextCtrl(self, wx.ID_ANY, "")
752
+ sizer_csv.Add(self.txt_filename, 1, wx.ALIGN_CENTER_VERTICAL, 0)
753
+
754
+ self.btn_fileDialog = wxButton(self, wx.ID_ANY, "...")
755
+ self.btn_fileDialog.SetMinSize(dip_size(self, 23, 23))
756
+ sizer_csv.Add(self.btn_fileDialog, 0, wx.ALIGN_CENTER_VERTICAL, 0)
757
+
758
+ self.btn_import = wxButton(self, wx.ID_ANY, _("Import CSV"))
759
+ sizer_csv.Add(self.btn_import, 0, wx.ALIGN_CENTER_VERTICAL, 0)
760
+
761
+ sizer_header = wx.BoxSizer(wx.HORIZONTAL)
762
+ self.rbox_header = wxRadioBox(
763
+ self,
764
+ wx.ID_ANY,
765
+ _("What does the first row contain:"),
766
+ choices=(
767
+ _("Auto-Detect"),
768
+ _("Contains Data"),
769
+ _("Contains Variable-Names"),
770
+ ),
771
+ majorDimension=3,
772
+ style=wx.RA_SPECIFY_COLS,
773
+ )
774
+ self.rbox_header.SetSelection(0)
775
+ sizer_header.Add(self.rbox_header, 1, wx.EXPAND, 0)
776
+ info_box.Add(sizer_header, 0, wx.EXPAND)
777
+
778
+ self.text_preview = TextCtrl(
779
+ self, wx.ID_ANY, style=wx.TE_READONLY | wx.TE_MULTILINE
780
+ )
781
+
782
+ main_sizer.Add(info_box, 0, wx.EXPAND, 0)
783
+ main_sizer.Add(self.text_preview, 1, wx.EXPAND, 0)
784
+ self.SetSizer(main_sizer)
785
+ self.Layout()
786
+ self.btn_fileDialog.Bind(wx.EVT_BUTTON, self.on_btn_file)
787
+ self.btn_import.Bind(wx.EVT_BUTTON, self.on_btn_import)
788
+ self.txt_filename.Bind(wx.EVT_TEXT, self.on_filetext_change)
789
+ self.btn_import.Enable(False)
790
+
791
+ def set_parent(self, par_panel):
792
+ self.parent_panel = par_panel
793
+
794
+ def on_btn_file(self, event):
795
+ myfile = ""
796
+ mydlg = wx.FileDialog(
797
+ self,
798
+ message=_("Choose a csv-file"),
799
+ wildcard="CSV-Files (*.csv)|*.csv|Text files (*.txt)|*.txt|All files (*.*)|*.*",
800
+ style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST | wx.FD_PREVIEW,
801
+ )
802
+ if mydlg.ShowModal() == wx.ID_OK:
803
+ # This returns a Python list of files that were selected.
804
+ myfile = mydlg.GetPath()
805
+ mydlg.Destroy()
806
+ if myfile != "":
807
+ self.txt_filename.SetValue(myfile)
808
+ # self.on_btn_import(None)
809
+
810
+ def import_csv(self, myfile):
811
+ if os.path.exists(myfile):
812
+ self.txt_filename.SetValue(myfile)
813
+ force_header = None
814
+ if self.rbox_header.GetSelection() == 1:
815
+ force_header = False
816
+ elif self.rbox_header.GetSelection() == 2:
817
+ force_header = True
818
+ ct, colcount, headers = self.wlist.load_csv_file(
819
+ myfile, force_header=force_header
820
+ )
821
+ msg = _("Imported file, {col} fields, {row} rows").format(
822
+ col=colcount, row=ct
823
+ )
824
+ if self.parent_panel is not None:
825
+ self.parent_panel.edit_message(msg)
826
+ self.parent_panel.populate_gui()
827
+ return True
828
+ return False
829
+
830
+ def on_btn_import(self, event):
831
+ myfile = self.txt_filename.GetValue()
832
+ if os.path.exists(myfile):
833
+ force_header = None
834
+ if self.rbox_header.GetSelection() == 1:
835
+ force_header = False
836
+ elif self.rbox_header.GetSelection() == 2:
837
+ force_header = True
838
+ ct, colcount, headers = self.wlist.load_csv_file(
839
+ myfile, force_header=force_header
840
+ )
841
+ msg = _("Imported file, {col} fields, {row} rows").format(
842
+ col=colcount, row=ct
843
+ )
844
+ if self.parent_panel is not None:
845
+ self.parent_panel.edit_message(msg)
846
+ self.parent_panel.populate_gui()
847
+
848
+ def on_filetext_change(self, event):
849
+ myfile = self.txt_filename.GetValue()
850
+ enab = False
851
+ if os.path.exists(myfile):
852
+ enab = True
853
+ self.btn_import.Enable(enab)
854
+ self.preview_it()
855
+
856
+ def preview_it(self):
857
+ filename = self.txt_filename.GetValue()
858
+
859
+ buffer = ""
860
+ if os.path.exists(filename):
861
+ decoder = EncodingDetectFile()
862
+ result = decoder.load(filename)
863
+ if result:
864
+ encoding, bom_marker, file_content = result
865
+ buffer = file_content
866
+
867
+ self.text_preview.SetValue(buffer)
868
+
869
+
870
+ class AboutPanel(wx.Panel):
871
+ def __init__(self, *args, context=None, **kwds):
872
+ kwds["style"] = kwds.get("style", 0) | wx.TAB_TRAVERSAL
873
+ wx.Panel.__init__(self, *args, **kwds)
874
+ self.context = context
875
+ self.context.themes.set_window_colors(self)
876
+ main_sizer = wx.BoxSizer(wx.VERTICAL)
877
+ info_box = StaticBoxSizer(self, wx.ID_ANY, _("How to use..."), wx.VERTICAL)
878
+ self.parent_panel = None
879
+ s = self.context.asset("wordlist_howto")
880
+ info_label = TextCtrl(
881
+ self, wx.ID_ANY, value=s, style=wx.TE_READONLY | wx.TE_MULTILINE
882
+ )
883
+ font = wx.Font(
884
+ 10,
885
+ wx.FONTFAMILY_TELETYPE,
886
+ wx.FONTSTYLE_NORMAL,
887
+ wx.FONTWEIGHT_NORMAL,
888
+ )
889
+ info_label.SetFont(font)
890
+ info_label.SetBackgroundColour(self.GetBackgroundColour())
891
+ info_box.Add(info_label, 1, wx.EXPAND, 0)
892
+ main_sizer.Add(info_box, 1, wx.EXPAND, 0)
893
+ self.SetSizer(main_sizer)
894
+ self.Layout()
895
+
896
+ def set_parent(self, par_panel):
897
+ self.parent_panel = par_panel
898
+
899
+
900
+ class WordlistEditor(MWindow):
901
+ def __init__(self, *args, **kwds):
902
+ super().__init__(500, 530, *args, **kwds)
903
+
904
+ self.panel_editor = WordlistPanel(self, wx.ID_ANY, context=self.context)
905
+ self.panel_import = ImportPanel(self, wx.ID_ANY, context=self.context)
906
+ self.panel_about = AboutPanel(self, wx.ID_ANY, context=self.context)
907
+
908
+ self.panel_editor.set_parent(self)
909
+ self.panel_import.set_parent(self)
910
+ self.panel_about.set_parent(self)
911
+
912
+ _icon = wx.NullIcon
913
+ _icon.CopyFromBitmap(icons8_curly_brackets.GetBitmap())
914
+ self.SetIcon(_icon)
915
+ self.notebook_main = wx.aui.AuiNotebook(
916
+ self,
917
+ -1,
918
+ style=wx.aui.AUI_NB_TAB_EXTERNAL_MOVE
919
+ | wx.aui.AUI_NB_SCROLL_BUTTONS
920
+ | wx.aui.AUI_NB_TAB_SPLIT
921
+ | wx.aui.AUI_NB_TAB_MOVE,
922
+ )
923
+ self.window_context.themes.set_window_colors(self.notebook_main)
924
+ bg_std = self.window_context.themes.get("win_bg")
925
+ bg_active = self.window_context.themes.get("highlight")
926
+ self.notebook_main.GetArtProvider().SetColour(bg_std)
927
+ self.notebook_main.GetArtProvider().SetActiveColour(bg_active)
928
+
929
+ self.sizer.Add(self.notebook_main, 1, wx.EXPAND, 0)
930
+ self.notebook_main.AddPage(self.panel_editor, _("Editing"))
931
+ self.notebook_main.AddPage(self.panel_import, _("Import/Export"))
932
+ self.notebook_main.AddPage(self.panel_about, _("How to use"))
933
+ # begin wxGlade: Keymap.__set_properties
934
+ self.DragAcceptFiles(True)
935
+ self.Bind(wx.EVT_DROP_FILES, self.on_drop_file)
936
+ self.SetTitle(_("Wordlist Editor"))
937
+ self.restore_aspect()
938
+
939
+ def on_drop_file(self, event):
940
+ """
941
+ Drop file handler
942
+ Accepts only a single file drop.
943
+ """
944
+ for pathname in event.GetFiles():
945
+ if self.panel_import.import_csv(pathname):
946
+ break
947
+
948
+ def delegates(self):
949
+ yield self.panel_editor
950
+ yield self.panel_import
951
+ yield self.panel_about
952
+
953
+ @staticmethod
954
+ def sub_register(kernel):
955
+ kernel.register(
956
+ "button/config/Wordlist",
957
+ {
958
+ "label": _("Wordlist Editor"),
959
+ "icon": icons8_curly_brackets,
960
+ "tip": _("Manages Wordlist-Entries"),
961
+ "help": "wordlist",
962
+ "action": lambda v: kernel.console("window toggle Wordlist\n"),
963
+ },
964
+ )
965
+
966
+ def window_open(self):
967
+ pass
968
+
969
+ def window_close(self):
970
+ pass
971
+
972
+ def edit_message(self, msg):
973
+ self.panel_editor.edit_message(msg)
974
+
975
+ def populate_gui(self):
976
+ self.panel_editor.populate_gui()
977
+
978
+ @staticmethod
979
+ def submenu():
980
+ # Suppress to avoid double menu-appearance
981
+ return "Editing", "Variables + Wordlists", True
982
+
983
+ @staticmethod
984
+ def helptext():
985
+ return _("Configure the wordlist (text variable) entries")