meerk40t 0.9.3001__py2.py3-none-any.whl → 0.9.7010__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 (445) 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 +1195 -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 +1844 -1507
  50. meerk40t/core/elements/clipboard.py +229 -219
  51. meerk40t/core/elements/element_treeops.py +4561 -2837
  52. meerk40t/core/elements/element_types.py +125 -105
  53. meerk40t/core/elements/elements.py +4329 -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 +933 -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/trace.py +651 -563
  66. meerk40t/core/elements/tree_commands.py +415 -409
  67. meerk40t/core/elements/undo_redo.py +116 -58
  68. meerk40t/core/elements/wordlist.py +319 -200
  69. meerk40t/core/exceptions.py +9 -9
  70. meerk40t/core/laserjob.py +220 -220
  71. meerk40t/core/logging.py +63 -63
  72. meerk40t/core/node/blobnode.py +83 -86
  73. meerk40t/core/node/bootstrap.py +105 -103
  74. meerk40t/core/node/branch_elems.py +40 -31
  75. meerk40t/core/node/branch_ops.py +45 -38
  76. meerk40t/core/node/branch_regmark.py +48 -41
  77. meerk40t/core/node/cutnode.py +29 -32
  78. meerk40t/core/node/effect_hatch.py +375 -257
  79. meerk40t/core/node/effect_warp.py +398 -0
  80. meerk40t/core/node/effect_wobble.py +441 -309
  81. meerk40t/core/node/elem_ellipse.py +404 -309
  82. meerk40t/core/node/elem_image.py +1082 -801
  83. meerk40t/core/node/elem_line.py +358 -292
  84. meerk40t/core/node/elem_path.py +259 -201
  85. meerk40t/core/node/elem_point.py +129 -102
  86. meerk40t/core/node/elem_polyline.py +310 -246
  87. meerk40t/core/node/elem_rect.py +376 -286
  88. meerk40t/core/node/elem_text.py +445 -418
  89. meerk40t/core/node/filenode.py +59 -40
  90. meerk40t/core/node/groupnode.py +138 -74
  91. meerk40t/core/node/image_processed.py +777 -766
  92. meerk40t/core/node/image_raster.py +156 -113
  93. meerk40t/core/node/layernode.py +31 -31
  94. meerk40t/core/node/mixins.py +135 -107
  95. meerk40t/core/node/node.py +1427 -1304
  96. meerk40t/core/node/nutils.py +117 -114
  97. meerk40t/core/node/op_cut.py +462 -335
  98. meerk40t/core/node/op_dots.py +296 -251
  99. meerk40t/core/node/op_engrave.py +414 -311
  100. meerk40t/core/node/op_image.py +755 -369
  101. meerk40t/core/node/op_raster.py +787 -522
  102. meerk40t/core/node/place_current.py +37 -40
  103. meerk40t/core/node/place_point.py +329 -126
  104. meerk40t/core/node/refnode.py +58 -47
  105. meerk40t/core/node/rootnode.py +225 -219
  106. meerk40t/core/node/util_console.py +48 -48
  107. meerk40t/core/node/util_goto.py +84 -65
  108. meerk40t/core/node/util_home.py +61 -61
  109. meerk40t/core/node/util_input.py +102 -102
  110. meerk40t/core/node/util_output.py +102 -102
  111. meerk40t/core/node/util_wait.py +65 -65
  112. meerk40t/core/parameters.py +709 -707
  113. meerk40t/core/planner.py +875 -785
  114. meerk40t/core/plotplanner.py +656 -652
  115. meerk40t/core/space.py +120 -113
  116. meerk40t/core/spoolers.py +706 -705
  117. meerk40t/core/svg_io.py +1836 -1549
  118. meerk40t/core/treeop.py +534 -445
  119. meerk40t/core/undos.py +278 -124
  120. meerk40t/core/units.py +784 -680
  121. meerk40t/core/view.py +393 -322
  122. meerk40t/core/webhelp.py +62 -62
  123. meerk40t/core/wordlist.py +513 -504
  124. meerk40t/cylinder/cylinder.py +247 -0
  125. meerk40t/cylinder/gui/cylindersettings.py +41 -0
  126. meerk40t/cylinder/gui/gui.py +24 -0
  127. meerk40t/device/__init__.py +1 -1
  128. meerk40t/device/basedevice.py +322 -123
  129. meerk40t/device/devicechoices.py +50 -0
  130. meerk40t/device/dummydevice.py +163 -128
  131. meerk40t/device/gui/defaultactions.py +618 -602
  132. meerk40t/device/gui/effectspanel.py +114 -0
  133. meerk40t/device/gui/formatterpanel.py +253 -290
  134. meerk40t/device/gui/warningpanel.py +337 -260
  135. meerk40t/device/mixins.py +13 -13
  136. meerk40t/dxf/__init__.py +1 -1
  137. meerk40t/dxf/dxf_io.py +766 -554
  138. meerk40t/dxf/plugin.py +47 -35
  139. meerk40t/external_plugins.py +79 -79
  140. meerk40t/external_plugins_build.py +28 -28
  141. meerk40t/extra/cag.py +112 -116
  142. meerk40t/extra/coolant.py +403 -0
  143. meerk40t/extra/encode_detect.py +198 -0
  144. meerk40t/extra/ezd.py +1165 -1165
  145. meerk40t/extra/hershey.py +835 -340
  146. meerk40t/extra/imageactions.py +322 -316
  147. meerk40t/extra/inkscape.py +630 -622
  148. meerk40t/extra/lbrn.py +424 -424
  149. meerk40t/extra/outerworld.py +284 -0
  150. meerk40t/extra/param_functions.py +1542 -1556
  151. meerk40t/extra/potrace.py +257 -253
  152. meerk40t/extra/serial_exchange.py +118 -0
  153. meerk40t/extra/updater.py +602 -453
  154. meerk40t/extra/vectrace.py +147 -146
  155. meerk40t/extra/winsleep.py +83 -83
  156. meerk40t/extra/xcs_reader.py +597 -0
  157. meerk40t/fill/fills.py +781 -335
  158. meerk40t/fill/patternfill.py +1061 -1061
  159. meerk40t/fill/patterns.py +614 -567
  160. meerk40t/grbl/control.py +87 -87
  161. meerk40t/grbl/controller.py +990 -903
  162. meerk40t/grbl/device.py +1081 -768
  163. meerk40t/grbl/driver.py +989 -771
  164. meerk40t/grbl/emulator.py +532 -497
  165. meerk40t/grbl/gcodejob.py +783 -767
  166. meerk40t/grbl/gui/grblconfiguration.py +373 -298
  167. meerk40t/grbl/gui/grblcontroller.py +485 -271
  168. meerk40t/grbl/gui/grblhardwareconfig.py +269 -153
  169. meerk40t/grbl/gui/grbloperationconfig.py +105 -0
  170. meerk40t/grbl/gui/gui.py +147 -116
  171. meerk40t/grbl/interpreter.py +44 -44
  172. meerk40t/grbl/loader.py +22 -22
  173. meerk40t/grbl/mock_connection.py +56 -56
  174. meerk40t/grbl/plugin.py +294 -264
  175. meerk40t/grbl/serial_connection.py +93 -88
  176. meerk40t/grbl/tcp_connection.py +81 -79
  177. meerk40t/grbl/ws_connection.py +112 -0
  178. meerk40t/gui/__init__.py +1 -1
  179. meerk40t/gui/about.py +2042 -296
  180. meerk40t/gui/alignment.py +1644 -1608
  181. meerk40t/gui/autoexec.py +199 -0
  182. meerk40t/gui/basicops.py +791 -670
  183. meerk40t/gui/bufferview.py +77 -71
  184. meerk40t/gui/busy.py +170 -133
  185. meerk40t/gui/choicepropertypanel.py +1673 -1469
  186. meerk40t/gui/consolepanel.py +706 -542
  187. meerk40t/gui/devicepanel.py +687 -581
  188. meerk40t/gui/dialogoptions.py +110 -107
  189. meerk40t/gui/executejob.py +316 -306
  190. meerk40t/gui/fonts.py +90 -90
  191. meerk40t/gui/functionwrapper.py +252 -0
  192. meerk40t/gui/gui_mixins.py +729 -0
  193. meerk40t/gui/guicolors.py +205 -182
  194. meerk40t/gui/help_assets/help_assets.py +218 -201
  195. meerk40t/gui/helper.py +154 -0
  196. meerk40t/gui/hersheymanager.py +1430 -846
  197. meerk40t/gui/icons.py +3422 -2747
  198. meerk40t/gui/imagesplitter.py +555 -508
  199. meerk40t/gui/keymap.py +354 -344
  200. meerk40t/gui/laserpanel.py +892 -806
  201. meerk40t/gui/laserrender.py +1470 -1232
  202. meerk40t/gui/lasertoolpanel.py +805 -793
  203. meerk40t/gui/magnetoptions.py +436 -0
  204. meerk40t/gui/materialmanager.py +2917 -0
  205. meerk40t/gui/materialtest.py +1722 -1694
  206. meerk40t/gui/mkdebug.py +646 -359
  207. meerk40t/gui/mwindow.py +163 -140
  208. meerk40t/gui/navigationpanels.py +2605 -2467
  209. meerk40t/gui/notes.py +143 -142
  210. meerk40t/gui/opassignment.py +414 -410
  211. meerk40t/gui/operation_info.py +310 -299
  212. meerk40t/gui/plugin.py +494 -328
  213. meerk40t/gui/position.py +714 -669
  214. meerk40t/gui/preferences.py +901 -650
  215. meerk40t/gui/propertypanels/attributes.py +1461 -1131
  216. meerk40t/gui/propertypanels/blobproperty.py +117 -114
  217. meerk40t/gui/propertypanels/consoleproperty.py +83 -80
  218. meerk40t/gui/propertypanels/gotoproperty.py +77 -0
  219. meerk40t/gui/propertypanels/groupproperties.py +223 -217
  220. meerk40t/gui/propertypanels/hatchproperty.py +489 -469
  221. meerk40t/gui/propertypanels/imageproperty.py +2244 -1384
  222. meerk40t/gui/propertypanels/inputproperty.py +59 -58
  223. meerk40t/gui/propertypanels/opbranchproperties.py +82 -80
  224. meerk40t/gui/propertypanels/operationpropertymain.py +1890 -1638
  225. meerk40t/gui/propertypanels/outputproperty.py +59 -58
  226. meerk40t/gui/propertypanels/pathproperty.py +389 -380
  227. meerk40t/gui/propertypanels/placementproperty.py +1214 -383
  228. meerk40t/gui/propertypanels/pointproperty.py +140 -136
  229. meerk40t/gui/propertypanels/propertywindow.py +313 -181
  230. meerk40t/gui/propertypanels/rasterwizardpanels.py +996 -912
  231. meerk40t/gui/propertypanels/regbranchproperties.py +76 -0
  232. meerk40t/gui/propertypanels/textproperty.py +770 -755
  233. meerk40t/gui/propertypanels/waitproperty.py +56 -55
  234. meerk40t/gui/propertypanels/warpproperty.py +121 -0
  235. meerk40t/gui/propertypanels/wobbleproperty.py +255 -204
  236. meerk40t/gui/ribbon.py +2468 -2210
  237. meerk40t/gui/scene/scene.py +1100 -1051
  238. meerk40t/gui/scene/sceneconst.py +22 -22
  239. meerk40t/gui/scene/scenepanel.py +439 -349
  240. meerk40t/gui/scene/scenespacewidget.py +365 -365
  241. meerk40t/gui/scene/widget.py +518 -505
  242. meerk40t/gui/scenewidgets/affinemover.py +215 -215
  243. meerk40t/gui/scenewidgets/attractionwidget.py +315 -309
  244. meerk40t/gui/scenewidgets/bedwidget.py +120 -97
  245. meerk40t/gui/scenewidgets/elementswidget.py +137 -107
  246. meerk40t/gui/scenewidgets/gridwidget.py +785 -745
  247. meerk40t/gui/scenewidgets/guidewidget.py +765 -765
  248. meerk40t/gui/scenewidgets/laserpathwidget.py +66 -66
  249. meerk40t/gui/scenewidgets/machineoriginwidget.py +86 -86
  250. meerk40t/gui/scenewidgets/nodeselector.py +28 -28
  251. meerk40t/gui/scenewidgets/rectselectwidget.py +589 -346
  252. meerk40t/gui/scenewidgets/relocatewidget.py +33 -33
  253. meerk40t/gui/scenewidgets/reticlewidget.py +83 -83
  254. meerk40t/gui/scenewidgets/selectionwidget.py +2952 -2756
  255. meerk40t/gui/simpleui.py +357 -333
  256. meerk40t/gui/simulation.py +2431 -2094
  257. meerk40t/gui/snapoptions.py +208 -203
  258. meerk40t/gui/spoolerpanel.py +1227 -1180
  259. meerk40t/gui/statusbarwidgets/defaultoperations.py +480 -353
  260. meerk40t/gui/statusbarwidgets/infowidget.py +520 -483
  261. meerk40t/gui/statusbarwidgets/opassignwidget.py +356 -355
  262. meerk40t/gui/statusbarwidgets/selectionwidget.py +172 -171
  263. meerk40t/gui/statusbarwidgets/shapepropwidget.py +754 -236
  264. meerk40t/gui/statusbarwidgets/statusbar.py +272 -260
  265. meerk40t/gui/statusbarwidgets/statusbarwidget.py +268 -270
  266. meerk40t/gui/statusbarwidgets/strokewidget.py +267 -251
  267. meerk40t/gui/themes.py +200 -78
  268. meerk40t/gui/tips.py +591 -0
  269. meerk40t/gui/toolwidgets/circlebrush.py +35 -35
  270. meerk40t/gui/toolwidgets/toolcircle.py +248 -242
  271. meerk40t/gui/toolwidgets/toolcontainer.py +82 -77
  272. meerk40t/gui/toolwidgets/tooldraw.py +97 -90
  273. meerk40t/gui/toolwidgets/toolellipse.py +219 -212
  274. meerk40t/gui/toolwidgets/toolimagecut.py +25 -132
  275. meerk40t/gui/toolwidgets/toolline.py +39 -144
  276. meerk40t/gui/toolwidgets/toollinetext.py +79 -236
  277. meerk40t/gui/toolwidgets/toollinetext_inline.py +296 -0
  278. meerk40t/gui/toolwidgets/toolmeasure.py +160 -216
  279. meerk40t/gui/toolwidgets/toolnodeedit.py +2088 -2074
  280. meerk40t/gui/toolwidgets/toolnodemove.py +92 -94
  281. meerk40t/gui/toolwidgets/toolparameter.py +754 -668
  282. meerk40t/gui/toolwidgets/toolplacement.py +108 -108
  283. meerk40t/gui/toolwidgets/toolpoint.py +68 -59
  284. meerk40t/gui/toolwidgets/toolpointlistbuilder.py +294 -0
  285. meerk40t/gui/toolwidgets/toolpointmove.py +183 -0
  286. meerk40t/gui/toolwidgets/toolpolygon.py +288 -403
  287. meerk40t/gui/toolwidgets/toolpolyline.py +38 -196
  288. meerk40t/gui/toolwidgets/toolrect.py +211 -207
  289. meerk40t/gui/toolwidgets/toolrelocate.py +72 -72
  290. meerk40t/gui/toolwidgets/toolribbon.py +598 -113
  291. meerk40t/gui/toolwidgets/tooltabedit.py +546 -0
  292. meerk40t/gui/toolwidgets/tooltext.py +98 -89
  293. meerk40t/gui/toolwidgets/toolvector.py +213 -204
  294. meerk40t/gui/toolwidgets/toolwidget.py +39 -39
  295. meerk40t/gui/usbconnect.py +98 -91
  296. meerk40t/gui/utilitywidgets/buttonwidget.py +18 -18
  297. meerk40t/gui/utilitywidgets/checkboxwidget.py +90 -90
  298. meerk40t/gui/utilitywidgets/controlwidget.py +14 -14
  299. meerk40t/gui/utilitywidgets/cyclocycloidwidget.py +343 -340
  300. meerk40t/gui/utilitywidgets/debugwidgets.py +148 -0
  301. meerk40t/gui/utilitywidgets/handlewidget.py +27 -27
  302. meerk40t/gui/utilitywidgets/harmonograph.py +450 -447
  303. meerk40t/gui/utilitywidgets/openclosewidget.py +40 -40
  304. meerk40t/gui/utilitywidgets/rotationwidget.py +54 -54
  305. meerk40t/gui/utilitywidgets/scalewidget.py +75 -75
  306. meerk40t/gui/utilitywidgets/seekbarwidget.py +183 -183
  307. meerk40t/gui/utilitywidgets/togglewidget.py +142 -142
  308. meerk40t/gui/utilitywidgets/toolbarwidget.py +8 -8
  309. meerk40t/gui/wordlisteditor.py +985 -931
  310. meerk40t/gui/wxmeerk40t.py +1444 -1169
  311. meerk40t/gui/wxmmain.py +5578 -4112
  312. meerk40t/gui/wxmribbon.py +1591 -1076
  313. meerk40t/gui/wxmscene.py +1635 -1453
  314. meerk40t/gui/wxmtree.py +2410 -2089
  315. meerk40t/gui/wxutils.py +1769 -1099
  316. meerk40t/gui/zmatrix.py +102 -102
  317. meerk40t/image/__init__.py +1 -1
  318. meerk40t/image/dither.py +429 -0
  319. meerk40t/image/imagetools.py +2778 -2269
  320. meerk40t/internal_plugins.py +150 -130
  321. meerk40t/kernel/__init__.py +63 -12
  322. meerk40t/kernel/channel.py +259 -212
  323. meerk40t/kernel/context.py +538 -538
  324. meerk40t/kernel/exceptions.py +41 -41
  325. meerk40t/kernel/functions.py +463 -414
  326. meerk40t/kernel/jobs.py +100 -100
  327. meerk40t/kernel/kernel.py +3809 -3571
  328. meerk40t/kernel/lifecycles.py +71 -71
  329. meerk40t/kernel/module.py +49 -49
  330. meerk40t/kernel/service.py +147 -147
  331. meerk40t/kernel/settings.py +383 -343
  332. meerk40t/lihuiyu/controller.py +883 -876
  333. meerk40t/lihuiyu/device.py +1181 -1069
  334. meerk40t/lihuiyu/driver.py +1466 -1372
  335. meerk40t/lihuiyu/gui/gui.py +127 -106
  336. meerk40t/lihuiyu/gui/lhyaccelgui.py +377 -363
  337. meerk40t/lihuiyu/gui/lhycontrollergui.py +741 -651
  338. meerk40t/lihuiyu/gui/lhydrivergui.py +470 -446
  339. meerk40t/lihuiyu/gui/lhyoperationproperties.py +238 -237
  340. meerk40t/lihuiyu/gui/tcpcontroller.py +226 -190
  341. meerk40t/lihuiyu/interpreter.py +53 -53
  342. meerk40t/lihuiyu/laserspeed.py +450 -450
  343. meerk40t/lihuiyu/loader.py +90 -90
  344. meerk40t/lihuiyu/parser.py +404 -404
  345. meerk40t/lihuiyu/plugin.py +101 -102
  346. meerk40t/lihuiyu/tcp_connection.py +111 -109
  347. meerk40t/main.py +231 -165
  348. meerk40t/moshi/builder.py +788 -781
  349. meerk40t/moshi/controller.py +505 -499
  350. meerk40t/moshi/device.py +495 -442
  351. meerk40t/moshi/driver.py +862 -696
  352. meerk40t/moshi/gui/gui.py +78 -76
  353. meerk40t/moshi/gui/moshicontrollergui.py +538 -522
  354. meerk40t/moshi/gui/moshidrivergui.py +87 -75
  355. meerk40t/moshi/plugin.py +43 -43
  356. meerk40t/network/console_server.py +102 -57
  357. meerk40t/network/kernelserver.py +10 -9
  358. meerk40t/network/tcp_server.py +142 -140
  359. meerk40t/network/udp_server.py +103 -77
  360. meerk40t/network/web_server.py +390 -0
  361. meerk40t/newly/controller.py +1158 -1144
  362. meerk40t/newly/device.py +874 -732
  363. meerk40t/newly/driver.py +540 -412
  364. meerk40t/newly/gui/gui.py +219 -188
  365. meerk40t/newly/gui/newlyconfig.py +116 -101
  366. meerk40t/newly/gui/newlycontroller.py +193 -186
  367. meerk40t/newly/gui/operationproperties.py +51 -51
  368. meerk40t/newly/mock_connection.py +82 -82
  369. meerk40t/newly/newly_params.py +56 -56
  370. meerk40t/newly/plugin.py +1214 -1246
  371. meerk40t/newly/usb_connection.py +322 -322
  372. meerk40t/rotary/gui/gui.py +52 -46
  373. meerk40t/rotary/gui/rotarysettings.py +240 -232
  374. meerk40t/rotary/rotary.py +202 -98
  375. meerk40t/ruida/control.py +291 -91
  376. meerk40t/ruida/controller.py +138 -1088
  377. meerk40t/ruida/device.py +672 -231
  378. meerk40t/ruida/driver.py +534 -472
  379. meerk40t/ruida/emulator.py +1494 -1491
  380. meerk40t/ruida/exceptions.py +4 -4
  381. meerk40t/ruida/gui/gui.py +71 -76
  382. meerk40t/ruida/gui/ruidaconfig.py +239 -72
  383. meerk40t/ruida/gui/ruidacontroller.py +187 -184
  384. meerk40t/ruida/gui/ruidaoperationproperties.py +48 -47
  385. meerk40t/ruida/loader.py +54 -52
  386. meerk40t/ruida/mock_connection.py +57 -109
  387. meerk40t/ruida/plugin.py +124 -87
  388. meerk40t/ruida/rdjob.py +2084 -945
  389. meerk40t/ruida/serial_connection.py +116 -0
  390. meerk40t/ruida/tcp_connection.py +146 -0
  391. meerk40t/ruida/udp_connection.py +73 -0
  392. meerk40t/svgelements.py +9671 -9669
  393. meerk40t/tools/driver_to_path.py +584 -579
  394. meerk40t/tools/geomstr.py +5583 -4680
  395. meerk40t/tools/jhfparser.py +357 -292
  396. meerk40t/tools/kerftest.py +904 -890
  397. meerk40t/tools/livinghinges.py +1168 -1033
  398. meerk40t/tools/pathtools.py +987 -949
  399. meerk40t/tools/pmatrix.py +234 -0
  400. meerk40t/tools/pointfinder.py +942 -942
  401. meerk40t/tools/polybool.py +940 -940
  402. meerk40t/tools/rasterplotter.py +1660 -547
  403. meerk40t/tools/shxparser.py +989 -901
  404. meerk40t/tools/ttfparser.py +726 -446
  405. meerk40t/tools/zinglplotter.py +595 -593
  406. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7010.dist-info}/LICENSE +21 -21
  407. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7010.dist-info}/METADATA +150 -139
  408. meerk40t-0.9.7010.dist-info/RECORD +445 -0
  409. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7010.dist-info}/WHEEL +1 -1
  410. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7010.dist-info}/top_level.txt +0 -1
  411. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7010.dist-info}/zip-safe +1 -1
  412. meerk40t/balormk/elementlightjob.py +0 -159
  413. meerk40t-0.9.3001.dist-info/RECORD +0 -437
  414. test/bootstrap.py +0 -63
  415. test/test_cli.py +0 -12
  416. test/test_core_cutcode.py +0 -418
  417. test/test_core_elements.py +0 -144
  418. test/test_core_plotplanner.py +0 -397
  419. test/test_core_viewports.py +0 -312
  420. test/test_drivers_grbl.py +0 -108
  421. test/test_drivers_lihuiyu.py +0 -443
  422. test/test_drivers_newly.py +0 -113
  423. test/test_element_degenerate_points.py +0 -43
  424. test/test_elements_classify.py +0 -97
  425. test/test_elements_penbox.py +0 -22
  426. test/test_file_svg.py +0 -176
  427. test/test_fill.py +0 -155
  428. test/test_geomstr.py +0 -1523
  429. test/test_geomstr_nodes.py +0 -18
  430. test/test_imagetools_actualize.py +0 -306
  431. test/test_imagetools_wizard.py +0 -258
  432. test/test_kernel.py +0 -200
  433. test/test_laser_speeds.py +0 -3303
  434. test/test_length.py +0 -57
  435. test/test_lifecycle.py +0 -66
  436. test/test_operations.py +0 -251
  437. test/test_operations_hatch.py +0 -57
  438. test/test_ruida.py +0 -19
  439. test/test_spooler.py +0 -22
  440. test/test_tools_rasterplotter.py +0 -29
  441. test/test_wobble.py +0 -133
  442. test/test_zingl.py +0 -124
  443. {test → meerk40t/cylinder}/__init__.py +0 -0
  444. /meerk40t/{core/element_commands.py → cylinder/gui/__init__.py} +0 -0
  445. {meerk40t-0.9.3001.dist-info → meerk40t-0.9.7010.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")