axmol/tools/project-creator/module/ui.py

#!/usr/bin/python
#coding=utf-8
# ui.py
# Create cross-platform cocos2d-x project
# Copyright (c) 2012 cocos2d-x.org
# Author: chuanwei


import platform
import os, os.path
import shutil
import threading
import Queue
import time

from core import CocosProject

#import head files by python version.
if int(platform.python_version().split('.')[0])>=3:
    from tkinter import *
    from tkinter.filedialog import *
    from tkinter.messagebox import *
else:
    from Tkinter import *
    from tkFileDialog import *
    from tkMessageBox import *


class ThreadedTask(threading.Thread):
    """Create cocos project thread.
    """
    def __init__(self, queue, projectName, packageName, language, projectPath):
        threading.Thread.__init__(self)
        self.queue = queue
        self.projectName = projectName
        self.packageName = packageName
        self.language = language
        self.projectPath = projectPath

    def run(self):
        """Create cocos project.
        custom message rules to notify ui
        As follow:
            begin@%d@%d@%s  --- create before
            doing@%d@%d@%s  --- creating
            end@%d@%d@%s    --- create after
        """
        #delete exist project.
        if os.path.exists(os.path.join(self.projectPath, self.projectName)):
            print ("###begin remove... " + self.projectName)
            shutil.rmtree(os.path.join(self.projectPath, self.projectName))
            print ("###remove finish:  " + self.projectName)
        putMsg = "begin@%d@%d@%s" %(0, 100, "begin create")
        self.queue.put(putMsg)

        project = CocosProject()
        breturn = project.createPlatformProjects(
            self.projectName,
            self.packageName,
            self.language,
            self.projectPath,
            self.newProjectCallBack
        )
        if breturn:
            putMsg = "end@%d@%d@%s" %(100, 100, "create successful")
        else:
            putMsg = "end@%d@%d@%s" %(100, 100, "create failure")
        self.queue.put(putMsg)

    def newProjectCallBack(self, step, totalStep, showMsg):
        """Creating cocos project callback.
        """
        putMsg = "doing@%d@%d@%s" %(step, totalStep, showMsg)
        self.queue.put(putMsg)

class StdoutRedirector(object):
    """Redirect output.
    """
    def __init__(self, text_area):
        self.text_area = text_area

    def write(self, str):
        self.text_area.insert(END, str)
        self.text_area.see(END)

class TkCocosDialog(Frame):
    def __init__(self, parent):
        Frame.__init__(self,parent)

        self.projectName = ""
        self.packageName = ""
        self.language = ""

        self.parent = parent
        self.columnconfigure(3, weight=1)
        self.rowconfigure(5, weight=1)

        # project name frame
        self.labName = Label(self, text="projectName:")
        self.strName = StringVar()
        self.strName.set("MyGame")
        self.editName = Entry(self, textvariable=self.strName)
        self.labName.grid(sticky=W, pady=4, padx=5)
        self.editName.grid(row=0, column=1, columnspan=3,padx=5, pady=2,sticky=E+W)

        # package name frame
        self.labPackage = Label(self, text="packageName:")
        self.strPackage=StringVar()
        self.strPackage.set("com.MyCompany.AwesomeGame")
        self.editPackage = Entry(self, textvariable=self.strPackage)
        self.labPackage.grid(row=1, column=0,sticky=W, padx=5)
        self.editPackage.grid(row=1, column=1, columnspan=3,padx=5, pady=2,sticky=E+W)

        # project path frame
        self.labPath = Label(self, text="projectPath:")
        self.editPath = Entry(self)
        self.btnPath = Button(self, text="...", width = 6, command = self.pathCallback)
        self.labPath.grid(row=2, column=0,sticky=W, pady=4, padx=5)
        self.editPath.grid(row=2, column=1, columnspan=3,padx=5, pady=2, sticky=E+W)
        self.btnPath.grid(row=2, column=4,)

        # language frame
        self.labLanguage = Label(self, text="language:")
        self.var=IntVar()
        self.var.set(1)
        self.checkcpp = Radiobutton(self, text="cpp", variable=self.var, value=1)
        self.checklua = Radiobutton(self, text="lua", variable=self.var, value=2)
        self.checkjs = Radiobutton(self, text="javascript", variable=self.var, value=3)
        self.labLanguage.grid(row=3, column=0,sticky=W, padx=5)
        self.checkcpp.grid(row=3, column=1,sticky=N+W)
        self.checklua.grid(row=3, column=2,padx=5,sticky=N+W)
        self.checkjs.grid(row=3, column=3,padx=5,sticky=N+W)

        # show progress
        self.progress = Scale(self, state= DISABLED, from_=0, to=100, orient=HORIZONTAL)
        self.progress.set(0)
        self.progress.grid(row=4, column=0, columnspan=4,padx=5, pady=2,sticky=E+W+S+N)

        # msg text frame
        self.text=Text(self,background = '#d9efff')
        self.text.bind("<KeyPress>", lambda e : "break")
        self.text.grid(row=5, column=0, columnspan=4, rowspan=1, padx=5, sticky=E+W+S+N)

        # new project button
        self.btnCreate = Button(self, text="create", command = self.createBtnCallback)
        self.btnCreate.grid(row=7, column=3, columnspan=1, rowspan=1,pady=2,ipadx=15,ipady =10, sticky=W)

        #center window on desktop
        curWidth = 500
        curHeight = 450
        scnWidth = self.parent.winfo_screenwidth()
        scnHeight = self.parent.winfo_screenheight()
        tmpcnf = '%dx%d+%d+%d'%(curWidth, curHeight, int((scnWidth-curWidth)/2), int((scnHeight-curHeight)/2))
        self.parent.geometry(tmpcnf)
        self.parent.title("CocosCreateProject")

        #fix size
        #self.parent.maxsize(curWidth, curHeight)
        #self.parent.minsize(curWidth, curHeight)

        #redirect out to text
        self.pack(fill=BOTH, expand=1)
        sys.stdout = StdoutRedirector(self.text)

    def process_queue(self):
        """
        """
        #message is empty
        if self.queue.empty():
            self.parent.after(100, self.process_queue)
            return

        #parse message
        msg = self.queue.get(0)
        msglist = msg.split("@")
        if len(msglist) < 4:
            return

        #begin
        if msglist[0] == "begin":
            self.progress['state'] = NORMAL
        #doing
        elif msglist[0] == "doing":
            pass

        self.progress.set(int(int(msglist[1])*100/int(msglist[2])))
        #end
        if msglist[0] == "end":
            showwarning("create", msglist[3])
            self.progress.set(0)
            self.text.insert(END,"=================END==============\n")
            self.progress['state'] = DISABLED
            self.btnCreate['state'] = NORMAL
            return
        self.parent.after(100, self.process_queue)

    def createBtnCallback(self):
        """Create button event.
        """
        #Check project name
        projectName = self.editName.get()
        if projectName == "":
            showwarning("warning", "projectName is empty")
            return

        #Check the package name is effective
        packageName = self.editPackage.get()
        packageList = packageName.split(".")
        if len(packageList) < 2:
            showwarning("warning", "packageName format  error!")
            return
        for index in range(len(packageList)):
            if (packageList[index] == "") or (packageList[index][0].isdigit()):
                showwarning("warning", "packageName format  error!")
                return

        # get select language type
        language = "cpp"
        if self.var.get() == 1:
            language = "cpp"
        elif self.var.get() == 2:
            language = "lua"
        elif self.var.get() == 3:
            language = "javascript"

        projectPath = self.editPath.get()
        if projectPath == "":
            showwarning("warning", "projectPath is empty")
            return

        # if project has already exist,....
        if os.path.exists(os.path.join(projectPath, projectName)):
            if not askyesno("warning", "%s had exist,do you want to recreate!" %projectName ):
                return

        #create a new thread to deal with create new project.
        self.btnCreate['state'] = DISABLED
        self.queue = Queue.Queue()
        ThreadedTask(self.queue, projectName, packageName, language, projectPath).start()
        self.parent.after(100, self.process_queue)

    def pathCallback(self):
        """Paht button event.
        """
        filepath = askdirectory()
        if filepath:
            self.editPath.delete(0, END)
            self.editPath.insert(0, filepath)

def createTkCocosDialog():

    old_stdout = sys.stdout
    root = Tk()
    app = TkCocosDialog(root)
    root.mainloop()
    sys.stdout = old_stdout

if __name__ =='__main__':
    createTkCocosDialog()