Total Complexity | 61 |
Total Lines | 314 |
Duplicated Lines | 7.64 % |
Changes | 0 |
Duplicate code is one of the most pungent code smells. A rule that is often used is to re-structure code once it is duplicated in three or more places.
Common duplication problems, and corresponding solutions are:
Complex classes like osci.dialog.NewGlobalTaskDlg often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
1 | # |
||
2 | # Copyright 2001 - 2016 Ludek Smid [http://www.ospace.net/] |
||
3 | # |
||
4 | # This file is part of Outer Space. |
||
5 | # |
||
6 | # Outer Space 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 | # Outer Space 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 Outer Space; if not, write to the Free Software |
||
18 | # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
||
19 | # |
||
20 | |||
21 | import pygameui as ui |
||
22 | from TechInfoDlg import TechInfoDlg |
||
23 | from ConstructionDlg import ConstructionDlg |
||
24 | import ige.ospace.Const as Const |
||
25 | from ige.ospace import Rules |
||
26 | from osci import gdata, client, res |
||
27 | from ige import GameException |
||
28 | import math |
||
29 | import Utils |
||
30 | |||
31 | class NewGlobalTaskDlg: |
||
32 | |||
33 | def __init__(self, app): |
||
34 | self.app = app |
||
35 | self.showShips = 1 |
||
36 | self.showOther = 0 |
||
37 | self.techID = 0 |
||
38 | self.showLevels = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 99] |
||
39 | self.techInfoDlg = TechInfoDlg(app) |
||
40 | self.constructionDlg = ConstructionDlg(app) |
||
41 | self.createUI() |
||
42 | self.win.setTagAttr('ship', 'visible', 1) |
||
43 | # set default sorting for technologies |
||
44 | self.win.vTechs.setSort("text") |
||
45 | |||
46 | def display(self, caller, queue, structToDemolish = Const.OID_NONE): |
||
47 | if gdata.config.defaults.reportfinalization != None: |
||
48 | val = gdata.config.defaults.reportfinalization |
||
49 | self.win.vReportFin.checked = val == 'yes' |
||
50 | |||
51 | self.caller = caller |
||
52 | self.playerID = client.getPlayerID() |
||
53 | self.maxTechLevel = 0 |
||
54 | self.quantity = 1 |
||
55 | self.queue = queue |
||
56 | self.showTechs() |
||
57 | self.win.show() |
||
58 | gdata.updateDlgs.append(self) |
||
59 | |||
60 | def hide(self): |
||
61 | self.win.setStatus(_("Ready.")) |
||
62 | if self in gdata.updateDlgs: |
||
63 | gdata.updateDlgs.remove(self) |
||
64 | self.win.hide() |
||
65 | |||
66 | def update(self): |
||
67 | if self.win.visible: |
||
68 | if self.showShips: |
||
69 | self.win.vInfo.enabled = Utils.enableConstruction(client) |
||
70 | self.showTechs() |
||
71 | |||
72 | def showTechs(self): |
||
73 | # techs |
||
74 | items = [] |
||
75 | select = None |
||
76 | |||
77 | showSmall = self.win.vSmall.checked |
||
78 | showMed = self.win.vMedium.checked |
||
79 | showLarge = self.win.vLarge.checked |
||
80 | showCivShip = self.win.vCivShip.checked |
||
81 | showMilShip = self.win.vMilShip.checked |
||
82 | |||
83 | for techID in client.getPlayer().techs.keys(): |
||
84 | tech = client.getTechInfo(techID) |
||
85 | |||
86 | # hide pirate techs |
||
87 | if tech.level == 99: |
||
88 | continue |
||
89 | |||
90 | if tech.isStructure or tech.globalDisabled: |
||
91 | continue |
||
92 | |||
93 | etc = _("N/A") |
||
94 | item = ui.Item(tech.name, |
||
95 | tLevel = tech.level, |
||
96 | tProd = tech.buildProd, |
||
97 | techID = techID, |
||
98 | tIsShip = 0, |
||
99 | ) |
||
100 | if item.tLevel > self.maxTechLevel: |
||
101 | self.maxTechLevel = item.tLevel |
||
102 | |||
103 | if item.tLevel in self.showLevels and \ |
||
104 | (self.showOther and (tech.isProject)): |
||
105 | items.append(item) |
||
106 | if techID == self.techID: |
||
107 | select = item |
||
108 | |||
109 | # special handling for ships |
||
110 | player = client.getPlayer() |
||
111 | if self.showShips: |
||
112 | for designID in player.shipDesigns.keys(): |
||
113 | tech = player.shipDesigns[designID] |
||
114 | |||
115 | if not ((tech.combatClass == 0 and showSmall) or (tech.combatClass == 1 and showMed) or (tech.combatClass == 2 and showLarge)): |
||
116 | continue |
||
117 | |||
118 | if not ((tech.isMilitary and showMilShip) or (not tech.isMilitary and showCivShip)): |
||
119 | continue |
||
120 | |||
121 | if tech.upgradeTo != Const.OID_NONE: |
||
122 | # skip ships that are set to upgrade |
||
123 | continue |
||
124 | etc = _("N/A") |
||
125 | item = ui.Item(tech.name, |
||
126 | tLevel = tech.level, |
||
127 | tProd = tech.buildProd, |
||
128 | techID = designID, |
||
129 | tIsShip = 1, |
||
130 | ) |
||
131 | items.append(item) |
||
132 | if designID == self.techID: |
||
133 | select = item |
||
134 | # sort it by level and then by name |
||
135 | items.sort(key=lambda a: (100 - a.tLevel, a.text)) |
||
136 | self.win.vTechs.items = items |
||
137 | self.win.vTechs.itemsChanged() |
||
138 | self.win.vTechs.selectItem(select) |
||
139 | # filter |
||
140 | for i in xrange(1, 10): |
||
|
|||
141 | widget = getattr(self.win, 'vLevel%d' % i) |
||
142 | if i in self.showLevels and i <= self.maxTechLevel: |
||
143 | widget.visible = 1 |
||
144 | widget.pressed = 1 |
||
145 | elif i not in self.showLevels and i <= self.maxTechLevel: |
||
146 | widget.visible = 1 |
||
147 | widget.pressed = 0 |
||
148 | else: |
||
149 | widget.visible = 0 |
||
150 | self.win.vShipsToggle.pressed = self.showShips |
||
151 | self.win.vOtherToggle.pressed = self.showOther |
||
152 | # quantity |
||
153 | self.win.vQuantity.text = str(self.quantity) |
||
154 | |||
155 | View Code Duplication | def showSlots(self): |
|
156 | # techs |
||
157 | items = [] |
||
158 | techs = {} |
||
159 | if self.showStructures: |
||
160 | player = client.getPlayer() |
||
161 | target = client.get(self.targetID, noUpdate = 1) |
||
162 | if hasattr(target, 'slots') and target.owner == player.oid: |
||
163 | if len(target.slots) < target.plSlots: |
||
164 | item = ui.Item(_("Free slot"), techID = 0) |
||
165 | items.append(item) |
||
166 | for struct in target.slots: |
||
167 | if not struct[Const.STRUCT_IDX_TECHID] in techs: |
||
168 | techs[struct[Const.STRUCT_IDX_TECHID]] = 1 |
||
169 | else: |
||
170 | techs[struct[Const.STRUCT_IDX_TECHID]] += 1 |
||
171 | for tech in techs.keys(): |
||
172 | techInfo = client.getTechInfo(tech) |
||
173 | item = ui.Item("%s (%d)" % (techInfo.name, techs[tech]), techID = tech) |
||
174 | items.append(item) |
||
175 | |||
176 | self.win.vTSlots.items = items |
||
177 | self.win.vTSlots.itemsChanged() |
||
178 | self.structToDemolish = Const.OID_NONE |
||
179 | |||
180 | |||
181 | def onSelectTech(self, widget, action, data): |
||
182 | self.techID = data.techID |
||
183 | |||
184 | def onToggleLevel(self, widget, action, data): |
||
185 | i = widget.data |
||
186 | if i in self.showLevels: |
||
187 | self.showLevels.remove(i) |
||
188 | else: |
||
189 | self.showLevels.append(i) |
||
190 | self.update() |
||
191 | |||
192 | def onCancel(self, widget, action, data): |
||
193 | self.hide() |
||
194 | |||
195 | def onConstruct(self, widget, action, data): |
||
196 | if not self.techID: |
||
197 | self.win.setStatus(_('Select technology to construct.')) |
||
198 | return |
||
199 | try: |
||
200 | self.quantity = int(self.win.vQuantity.text) |
||
201 | except ValueError: |
||
202 | self.win.setStatus(_('Specify quantity (1, 2, 3, ...).')) |
||
203 | return |
||
204 | try: |
||
205 | self.win.setStatus(_('Executing START CONSTRUCTION command...')) |
||
206 | player = client.getPlayer() |
||
207 | |||
208 | player.prodQueues[self.queue], player.stratRes = client.cmdProxy.startGlobalConstruction(self.playerID, self.techID, self.quantity, self.techID < 1000, self.win.vReportFin.checked, self.queue) |
||
209 | self.win.setStatus(_('Command has been executed.')) |
||
210 | except GameException, e: |
||
211 | self.win.setStatus(e.args[0]) |
||
212 | return |
||
213 | self.hide() |
||
214 | self.caller.update() |
||
215 | |||
216 | def onToggleShips(self, widget, action, data): |
||
217 | self.quantity = int(self.win.vQuantity.text) |
||
218 | self.showStructures = 0 |
||
219 | self.showShips = 1 |
||
220 | self.showOther = 0 |
||
221 | self.win.setTagAttr('struct', 'visible', 0) |
||
222 | self.win.setTagAttr('ship', 'visible', 1) |
||
223 | self.update() |
||
224 | |||
225 | def onToggleOther(self, widget, action, data): |
||
226 | self.quantity = int(self.win.vQuantity.text) |
||
227 | self.showStructures = 0 |
||
228 | self.showShips = 0 |
||
229 | self.showOther = 1 |
||
230 | self.win.setTagAttr('struct', 'visible', 0) |
||
231 | self.win.setTagAttr('ship', 'visible', 0) |
||
232 | self.update() |
||
233 | |||
234 | def onInfo(self, widget, action, data): |
||
235 | if len(self.win.vTechs.selection) == 0: |
||
236 | return |
||
237 | task = self.win.vTechs.selection[0] |
||
238 | if not task.tIsShip: |
||
239 | self.techInfoDlg.display(task.techID) |
||
240 | else: |
||
241 | self.constructionDlg.selectedDesignID = task.techID; |
||
242 | self.constructionDlg.display() |
||
243 | |||
244 | def onFilter(self, widget, action, data): |
||
245 | self.update() |
||
246 | |||
247 | def createUI(self): |
||
248 | w, h = gdata.scrnSize |
||
249 | cols = 30 |
||
250 | rows = 24 #was 23 |
||
251 | dlgWidth = cols * 20 + 4 |
||
252 | dlgHeight = rows * 20 + 4 |
||
253 | self.win = ui.Window(self.app, |
||
254 | modal = 1, |
||
255 | escKeyClose = 1, |
||
256 | movable = 0, |
||
257 | title = _('Select new global task'), |
||
258 | rect = ui.Rect((w - dlgWidth) / 2, (h - dlgHeight) / 2, dlgWidth, dlgHeight), |
||
259 | layoutManager = ui.SimpleGridLM(), |
||
260 | tabChange = True |
||
261 | ) |
||
262 | self.win.subscribeAction('*', self) |
||
263 | ui.Title(self.win, layout = (0, 0, 19, 1), text = _('Technology'), |
||
264 | align = ui.ALIGN_W, font = 'normal-bold') |
||
265 | ui.Listbox(self.win, layout = (0, 1, 19, 19), id = 'vTechs', |
||
266 | columns = ((_('Name'), 'text', 13, ui.ALIGN_W), (_('Lvl'), 'tLevel', 2, ui.ALIGN_E), |
||
267 | (_('Constr'), 'tProd', 3, ui.ALIGN_E)), |
||
268 | columnLabels = 1, action = 'onSelectTech') |
||
269 | # filter |
||
270 | ui.Button(self.win, layout = (0, 20, 3, 1), text = _('Ships'), toggle = 1, |
||
271 | id = 'vShipsToggle', action = 'onToggleShips') |
||
272 | ui.Button(self.win, layout = (3, 20, 3, 1), text = _('Misc'), toggle = 1, |
||
273 | id = 'vOtherToggle', action = 'onToggleOther') |
||
274 | ui.Button(self.win, layout = (6, 20, 1, 1), text = _('1'), id = 'vLevel1', |
||
275 | toggle = 1, action = 'onToggleLevel', data = 1) |
||
276 | ui.Button(self.win, layout = (7, 20, 1, 1), text = _('2'), id = 'vLevel2', |
||
277 | toggle = 1, action = 'onToggleLevel', data = 2) |
||
278 | ui.Button(self.win, layout = (8, 20, 1, 1), text = _('3'), id = 'vLevel3', |
||
279 | toggle = 1, action = 'onToggleLevel', data = 3) |
||
280 | ui.Button(self.win, layout = (9, 20, 1, 1), text = _('4'), id = 'vLevel4', |
||
281 | toggle = 1, action = 'onToggleLevel', data = 4) |
||
282 | ui.Button(self.win, layout = (10, 20, 1, 1), text = _('5'), id = 'vLevel5', |
||
283 | toggle = 1, action = 'onToggleLevel', data = 5) |
||
284 | ui.Button(self.win, layout = (11, 20, 1, 1), text = _('6'), id = 'vLevel6', |
||
285 | toggle = 1, action = 'onToggleLevel', data = 6) |
||
286 | ui.Button(self.win, layout = (12, 20, 1, 1), text = _('7'), id = 'vLevel7', |
||
287 | toggle = 1, action = 'onToggleLevel', data = 7) |
||
288 | ui.Button(self.win, layout = (13, 20, 1, 1), text = _('8'), id = 'vLevel8', |
||
289 | toggle = 1, action = 'onToggleLevel', data = 8) |
||
290 | ui.Button(self.win, layout = (14, 20, 1, 1), text = _('9'), id = 'vLevel9', |
||
291 | toggle = 1, action = 'onToggleLevel', data = 9) |
||
292 | ui.Button(self.win, layout = (15, 20, 4, 1), text = _('Info'), action = 'onInfo', |
||
293 | id = 'vInfo') |
||
294 | # ship types |
||
295 | ui.Check(self.win, layout = (0, 21, 6, 1), text = _('Small'), tags = ['ship'], |
||
296 | id = 'vSmall', checked = 1, align = ui.ALIGN_W, action = 'onFilter') |
||
297 | ui.Check(self.win, layout = (6, 21, 6, 1), text = _('Medium'), tags = ['ship'], |
||
298 | id = 'vMedium', checked = 1, align = ui.ALIGN_W, action = 'onFilter') |
||
299 | ui.Check(self.win, layout = (12, 21, 6, 1), text = _('Large'), tags = ['ship'], |
||
300 | id = 'vLarge', checked = 1, align = ui.ALIGN_W, action = 'onFilter') |
||
301 | ui.Check(self.win, layout = (18, 21, 6, 1), text = _('Civilian'), tags = ['ship'], |
||
302 | id = 'vCivShip', checked = 1, align = ui.ALIGN_W, action = 'onFilter') |
||
303 | ui.Check(self.win, layout = (24, 21, 6, 1), text = _('Military'), tags = ['ship'], |
||
304 | id = 'vMilShip', checked = 1, align = ui.ALIGN_W, action = 'onFilter') |
||
305 | # build |
||
306 | ui.Title(self.win, layout = (19, 18, 11, 1), text = _('Options'), |
||
307 | align = ui.ALIGN_W, font = 'normal-bold') |
||
308 | ui.Label(self.win, layout = (19, 19, 5, 1), text = _('Quantity'), align = ui.ALIGN_W) |
||
309 | ui.Entry(self.win, layout = (24, 19, 6, 1), id = 'vQuantity', align = ui.ALIGN_E, orderNo = 1) |
||
310 | ui.Check(self.win, layout = (23, 20, 7, 1), id = 'vReportFin', text = _('Report finalization')) |
||
311 | ui.Title(self.win, layout = (0, 22, 20, 1), id = 'vStatusBar', align = ui.ALIGN_W) |
||
312 | ui.TitleButton(self.win, layout = (20, 22, 5, 1), text = _('Cancel'), action = 'onCancel') |
||
313 | ui.TitleButton(self.win, layout = (25, 22, 5, 1), text = _('Construct'), action = 'onConstruct') |
||
314 |