| @@ -134,8 +134,11 @@ class Plot: | |||
| self.mission = Plot.miss[self.missIndex] | |||
| self.probIndex = probIndex if probIndex != None else r.randint(0, len(Plot.prob)-1) | |||
| self.problem = Plot.prob[self.probIndex] | |||
| if (secProblem != None and secProblem in list(filter(lambda x:x["id"], Plot.prob))): | |||
| self.secProblem = [x for x in Plot.prob if x["id"] == secProblem][0] | |||
| if secProblem != None: | |||
| try: | |||
| self.secProblem = [x for x in Plot.prob if x["id"] == secProblem][0] | |||
| except: | |||
| self.secProblem = None | |||
| else: | |||
| self.secProblem = None | |||
| self.thirdProblem = [x for x in Plot.prob if x["id"] == thirdProblem][0] if (thirdProblem != None and thirdProblem in list(filter(lambda x:x["id"], Plot.prob))) else None | |||