Attempt to fix WindowsError related bug.
[bertos.git] / wizard / BFolderPage.py
1 #!/usr/bin/env python
2 # encoding: utf-8
3 #
4 # This file is part of BeRTOS.
5 #
6 # Bertos is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
10 #
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 # GNU General Public License for more details.
15 #
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19 #
20 # As a special exception, you may use this file as part of a free software
21 # library without restriction.  Specifically, if other files instantiate
22 # templates or use macros or inline functions from this file, or you compile
23 # this file and link it with other files to produce an executable, this
24 # file does not by itself cause the resulting executable to be covered by
25 # the GNU General Public License.  This exception does not however
26 # invalidate any other reasons why the executable file might be covered by
27 # the GNU General Public License.
28 #
29 # Copyright 2008 Develer S.r.l. (http://www.develer.com/)
30 #
31 # $Id$
32 #
33 # Author: Lorenzo Berni <duplo@develer.com>
34 #
35
36 import os
37
38 from PyQt4.QtGui import *
39 from BWizardPage import *
40 import bertos_utils
41
42 from BCpuPage import BCpuPage
43 from BBoardPage import BBoardPage
44
45 from const import *
46
47 class BFolderPage(BWizardPage):
48     """
49     Initial page of the wizard. Permit to select the project name and the directory
50     where the project will be created.
51     """
52     
53     def __init__(self):
54         BWizardPage.__init__(self, UI_LOCATION + "/dir_select.ui")
55         self.setTitle(self.tr("Select the project name"))
56         self.initializeAttributes()
57     
58     ## Overloaded QWizardPage methods ##
59
60     def isComplete(self):
61         """
62         Overload of the QWizardPage isComplete method.
63         """
64         self.setDefaultFolder(self._destination_folder)
65         if self.pageContent.projectPath.text() != "None":
66             self.setProjectInfo("PROJECT_PATH", unicode(self.pageContent.projectPath.text()))
67             self.setProjectInfo("PROJECT_NAME", os.path.basename(unicode(self.pageContent.projectPath.text())))
68             self.setProjectInfo("PROJECT_SRC_PATH", os.path.join(self.projectInfo("PROJECT_PATH"), self.projectInfo("PROJECT_NAME")))
69             self.setProjectInfo("PROJECT_HW_PATH", os.path.join(self.projectInfo("PROJECT_PATH"), self.projectInfo("PROJECT_NAME")))
70             self.setProjectInfo("ROUTE", self.next_page)
71             self.setProjectInfo("PROJECT_FROM_PRESET", self.from_preset)
72             return True
73         else:
74             return False
75     
76     ####
77
78     ## Overloaded BWizardPage methods ##
79
80     def setupUi(self):
81         """
82         Overload of the BWizardPage setupUi method.
83         """
84         self.pageContent.warningLabel.setVisible(False)
85     
86     def connectSignals(self):
87         """
88         Overload of the BWizardPage connectSignals method.
89         """
90         self.connect(self.pageContent.nameEdit, SIGNAL("textChanged(const QString)"), self.nameChanged)
91         self.connect(self.pageContent.directoryEdit, SIGNAL("textChanged(const QString)"), self.directoryChanged)
92         self.connect(self.pageContent.directoryButton, SIGNAL("clicked()"), self.selectDirectory)
93         self.connect(self.pageContent.customButton, SIGNAL("toggled(bool)"), self.isComplete)
94     
95     ####
96
97     ## Slots ##
98     
99     def nameChanged(self, name):
100         """
101         Slot called when the project name is changed manually by the user.
102         """
103         try:
104             name = unicode(name).encode("ascii")
105         except UnicodeEncodeError:
106             name = self._project_name
107             self.pageContent.nameEdit.setText(name)
108         self._project_name = unicode(name).replace(" ", "_")
109         self.setProjectPath()
110     
111     def directoryChanged(self, directory):
112         """
113         Slot called when the project folder is changed manually by the user.
114         """
115         try:
116             directory = unicode(directory).encode("ascii")
117         except UnicodeEncodeError:
118             directory = self._destination_folder
119             self.pageContent.directoryEdit.setText(directory)
120         self._destination_folder = directory
121         self.setProjectPath()
122
123     def selectDirectory(self):
124         """
125         Slot called when the project folder is changed using the file dialog.
126         """
127         directory = unicode(QFileDialog.getExistingDirectory(self, self.tr("Open Directory"), self.pageContent.directoryEdit.text(), QFileDialog.ShowDirsOnly))
128         if len(directory) > 0:
129             self.pageContent.directoryEdit.setText(QDir.toNativeSeparators(directory))
130
131     ####
132
133     @property
134     def next_page(self):
135         """
136         Contains the next page class.
137         """
138         if self.from_preset:
139             return BBoardPage
140         else:
141             return BCpuPage
142
143     @property
144     def from_preset(self):
145         return self.pageContent.predefinedButton.isChecked()
146     
147     def initializeAttributes(self):
148         """
149         Initializes the page attributes to the default values.
150         """
151         self._project_name = ""
152         stored_folder = self.defaultFolder()
153         if stored_folder != "":
154             self._destination_folder = stored_folder
155         elif os.name == "nt":
156             from win32com.shell import shell, shellcon
157             self._destination_folder = shell.SHGetFolderPath(0, shellcon.CSIDL_PERSONAL, 0, 0)
158             del shell
159             del shellcon
160         else:
161             self._destination_folder = os.path.expanduser("~")
162         self.pageContent.directoryEdit.setText(self._destination_folder)
163     
164     def setProjectPath(self):
165         """
166         Analyzes the page attributes and generates the path string.
167         """
168         if self._destination_folder != "" and self._project_name <> "":
169             if not self._destination_folder.endswith(os.sep):
170                 self._destination_folder += os.sep
171             self.pageContent.projectPath.setText(QDir.toNativeSeparators(self._destination_folder + self._project_name))
172             if os.path.exists(self._destination_folder + self._project_name):
173                 self.pageContent.warningLabel.setVisible(True)
174                 self.pageContent.warningLabel.setText(self.tr("<font color='#FF0000'>Warning: the selected directory exists, \
175                     it will be destroyed with all contained subdirectories and files...</font>"))
176             else:
177                 self.pageContent.warningLabel.setVisible(False)
178                 self.pageContent.warningLabel.setText("")
179         else:
180             self.pageContent.projectPath.setText("None")
181             self.pageContent.warningLabel.setVisible(False)
182             self.pageContent.warningLabel.setText("")
183         self.emit(SIGNAL("completeChanged()"))