working on code generation
This commit is contained in:
paul
@@ -2,6 +2,78 @@ import { createFilesAndDownload } from "../../../codeEngine/utils"
|
||||
import MainWindow from "../widgets/mainWindow"
|
||||
|
||||
import { message } from "antd"
|
||||
import TopLevel from "../widgets/toplevel"
|
||||
|
||||
|
||||
// FIXME: if the toplevel comes first, before the MainWindow in widgetlist the root may become null
|
||||
// Recursive function to generate the code list, imports, requirements, and track mainVariable
|
||||
function generateTkinterCodeList(widgetList = [], widgetRefs = [], parentVariable = null, mainVariable = "") {
|
||||
let variableNames = [] // {widgetId: "", name: "", count: 1}
|
||||
let imports = new Set([])
|
||||
let requirements = new Set([])
|
||||
|
||||
let code = []
|
||||
|
||||
for (let widget of widgetList) {
|
||||
// console.log("Key: ", widget)
|
||||
const widgetRef = widgetRefs[widget.id].current
|
||||
|
||||
let varName = widgetRef.getVariableName()
|
||||
|
||||
imports.add(widgetRef.getImports())
|
||||
requirements.add(widgetRef.getRequirements())
|
||||
|
||||
// Set main variable if the widget is MainWindow
|
||||
if (widget.widgetType === MainWindow) {
|
||||
mainVariable = varName
|
||||
}
|
||||
|
||||
if (!variableNames.find((val) => val.variable === varName)) {
|
||||
variableNames.push({ widgetId: widgetRef.id, name: varName, count: 1 })
|
||||
} else {
|
||||
// Avoid duplicate names
|
||||
const existingVariable = variableNames.find((val) => val.variable === varName)
|
||||
const existingCount = existingVariable.count
|
||||
existingVariable.count += 1
|
||||
varName = `${existingVariable.name}${existingCount}`
|
||||
variableNames.push({ widgetId: widgetRef.id, name: varName, count: 1 })
|
||||
}
|
||||
|
||||
const currentParentVariable = variableNames.find(val => val.widgetId === widget.id)?.name || parentVariable
|
||||
|
||||
let widgetCode = widgetRef.generateCode(varName, currentParentVariable)
|
||||
|
||||
if (!(widgetCode instanceof Array)) {
|
||||
throw new Error("Generate code function should return array, each new line should be a new item")
|
||||
}
|
||||
|
||||
// Add \n after every line
|
||||
widgetCode = widgetCode.flatMap((item, index) => index < widgetCode.length - 1 ? [item, "\n"] : [item])
|
||||
|
||||
code.push(...widgetCode)
|
||||
code.push("\n\n")
|
||||
|
||||
// Recursively handle child widgets, which are full widget objects
|
||||
if (widget.children && widget.children.length > 0) {
|
||||
const childResult = generateTkinterCodeList(widget.children, widgetRefs, varName, mainVariable)
|
||||
|
||||
// Merge child imports, requirements, and code
|
||||
imports = new Set([...imports, ...childResult.imports])
|
||||
requirements = new Set([...requirements, ...childResult.requirements])
|
||||
code.push(...childResult.code)
|
||||
|
||||
// Track the main variable returned by the child
|
||||
mainVariable = childResult.mainVariable || mainVariable
|
||||
}
|
||||
}
|
||||
|
||||
return {
|
||||
imports: Array.from(imports), // Convert Set to Array
|
||||
code: code,
|
||||
requirements: Array.from(requirements), // Convert Set to Array
|
||||
mainVariable
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
async function generateTkinterCode(projectName, widgetList=[], widgetRefs=[]){
|
||||
@@ -10,7 +82,10 @@ async function generateTkinterCode(projectName, widgetList=[], widgetRefs=[]){
|
||||
|
||||
let mainWindowCount = 0
|
||||
|
||||
for (let widget of widgetList){
|
||||
// only MainWindow and/or the TopLevel should be on the canvas
|
||||
const filteredWidgetList = widgetList.filter(widget => widget.widgetType === MainWindow || widget.widgetType === TopLevel)
|
||||
|
||||
for (let widget of filteredWidgetList){
|
||||
if (widget.widgetType === MainWindow){
|
||||
mainWindowCount += 1
|
||||
}
|
||||
@@ -27,64 +102,47 @@ async function generateTkinterCode(projectName, widgetList=[], widgetRefs=[]){
|
||||
return
|
||||
}
|
||||
|
||||
let variableNames = [] // {widgetId: "", name: "", count: 1}
|
||||
// let code = [`# This code is generated by PyUIbuilder: https://github.com/paulledemon \n`]
|
||||
// code.push("\n", "import tkinter as tk", "\n\n")
|
||||
|
||||
let imports = new Set([])
|
||||
let requirements = new Set([])
|
||||
// widget - {id, widgetType: widgetComponentType, children: [], parent: "", initialData: {}}
|
||||
|
||||
const generatedObject = generateTkinterCodeList(filteredWidgetList, widgetRefs, "", "")
|
||||
|
||||
let code = [`# This code is generated by PyUIbuilder: https://github.com/paulledemon \n`]
|
||||
code.push("\n", "import tkinter as tk", "\n\n")
|
||||
const {code: codeLines, imports, requirements, mainVariable} = generatedObject
|
||||
|
||||
const code = [
|
||||
"# This code is generated by PyUIbuilder: https://github.com/paulledemon",
|
||||
"\n\n",
|
||||
...imports,
|
||||
"\n\n",
|
||||
...codeLines,
|
||||
"\n",
|
||||
`${mainVariable}.mainloop()`,
|
||||
]
|
||||
|
||||
let mainVariable = "" // the main window variable
|
||||
|
||||
for (let widget of widgetList){
|
||||
console.log("Key: ", widget)
|
||||
const widgetRef = widgetRefs[widget.id].current
|
||||
|
||||
let varName = widgetRef.getVariableName()
|
||||
|
||||
imports.add(widgetRef.getImports())
|
||||
requirements.add(widgetRef.getRequirements())
|
||||
|
||||
if (widget.widgetType === MainWindow){
|
||||
mainVariable = varName
|
||||
}
|
||||
|
||||
if (!variableNames.find((val) => val.variable === varName)){
|
||||
variableNames.push({widgetId: widgetRef.id, name: varName, count: 1})
|
||||
}else{
|
||||
// Avoid duplicate names
|
||||
const existingVariable = variableNames.find((val) => val.variable === varName)
|
||||
const existingCount = existingVariable.count
|
||||
existingVariable.count += 1
|
||||
varName = `${existingVariable.name}${existingCount}`
|
||||
variableNames.push({widgetId: widgetRef.id, name: varName, count: 1})
|
||||
}
|
||||
|
||||
const parentVariable = variableNames.find(val => val.widgetId === widget.id)?.name || null
|
||||
|
||||
console.log("widget ref: ", widgetRef)
|
||||
let widgetCode = widgetRef.generateCode(varName, parentVariable)
|
||||
|
||||
if (!widgetCode instanceof Array){
|
||||
throw new Error("Generate code function should return array, each new line should be a new item")
|
||||
}
|
||||
// add \n after every line
|
||||
widgetCode = widgetCode.flatMap((item, index) => index < code.length - 1 ? [item, "\n"] : [item])
|
||||
|
||||
code.push(...widgetCode)
|
||||
code.push("\n")
|
||||
|
||||
}
|
||||
|
||||
code.push(`\n${mainVariable}.mainloop()`) // start the main loop for tkinter
|
||||
|
||||
// TODO: create requirements.txt file
|
||||
console.log("Code: ", code.join(""))
|
||||
|
||||
message.info("starting zipping files, download will start in a few seconds")
|
||||
|
||||
// createFilesAndDownload([], projectName).then(() => {
|
||||
const createFileList = [
|
||||
{
|
||||
fileData: code.join(""),
|
||||
fileName: "main.py",
|
||||
folder: ""
|
||||
}
|
||||
]
|
||||
|
||||
if (requirements.length > 0){
|
||||
createFileList.push({
|
||||
fileData: requirements.join("\n"),
|
||||
fileName: "requirements.txt",
|
||||
folder: ""
|
||||
})
|
||||
}
|
||||
|
||||
// createFilesAndDownload(createFileList, projectName).then(() => {
|
||||
// message.success("Download complete")
|
||||
// }).catch(() => {
|
||||
// message.error("Error while downloading")
|
||||
|
||||
Reference in New Issue
Block a user