def but(): if not list1.currentItem(): QMessageBox.critical(window,'error','Not league selected') else: defense = [] ln=str(list1.currentItem().text()) loopvar = QMessageBox.Yes rn,test = QInputDialog.getText(window,'New roster','Please enter the name for the new roster you\'d like to add:').strip() rn = str(rn) while loopvar == QMessageBox.Yes: dn,test = QInputDialog.getText(window,'Defense','Please enter a team to use as defense:').strip() if test: d = nflgame.standard_team(str(dn)) if d is None: QMessageBox.critical(window,'error','Invalid team name') else: defense.append(d) if defense: loopvar = QMessageBox.question(window,'???','Would you like to add another defense team?',QMessageBox.Yes|QMessageBox.No,QMessageBox.Yes) elif not defense: QMessageBox.critical(window,'error','Cannot have no defense') else: loopvar = QMessageBox.No status = fbTool.addRoster(ln,rn,defense) if status >=3: errstr = 'defense team ' + defense[status-3] + ' is already used' QMessageBox.critical(window,'error',errstr) elif status ==1: QMessageBox.critical(window,'error','Team already exists.') elif status: QMessageBox.critical(window,'error','Something is horribly wrong') mainMenu.updateTree()
def addRosterUI(ln,tn,defense): inp='y' if not defense: while inp.lower() not in ['n' or 'no']: if inp.lower()=='y' or inp.lower()=='yes': defen=raw_input('Please enter team name for defense: ').strip() while nflgame.standard_team(defen) is None: defen=raw_input('Invalid team name, please try again: ').strip() defen=nflgame.standard_team(defen) defense.append(defen) else: print 'invalid input, try again.' inp = raw_input('Would you like to add another team for defense? (y/n): ' ).strip() testVar=fbTool.addRoster(ln,tn,defense) if not testVar: print 'Succesfully added roster %s!' % tn else: print 'Failed to add team...' if testVar == 1: print 'Team %s already exists in league %s' %(tn,ln) anotherVar = ' ' while anotherVar not in ['y','yes','n','no']: anotherVar =raw_input('Try again?(y/n)').strip() if anotherVar.lower() in ['y','yes']: tn=getTeamName(ln) return addRosterUI(ln,tn,defense) if anotherVar.lower() in ['n','no']: print 'Team not added...' return -1 if anotherVar.lower() not in ['y','yes','n','no']: print 'Invalid input. Please try again.' if testVar == 2: print 'League %s does not exist.' userInVar = ' ' while userInVar not in ['y','yes','n''no']: userInVar = raw_input('Would you like to add it? (y/n): ').strip() if userInVar.lower() not in ['y','yes','n','no']: print 'invalid input. please try again.' if userInVar.lower() in ['y','yes']: if not fbTool.addLeague(ln): return addRosterUI(ln,tn,defense) if userInVar.lower() in ['n','no']: print 'team not added.' return 1 if testVar >= 3: print '%s already used as defense team in league %s'%(defense[testVar-3],ln) oneFinalVar = ' ' while oneFinalVar not in ['y','yes','n','no']: oneFinalVar = raw_input('Try again?(y/n) ').strip() if oneFinalVar.lower() in ['y','yes']: addRosterUI(ln,tn,[]) if oneFinalVar.lower() in ['n','no']: print 'Team not added...' return -1 if oneFinalVar.lower() not in ['y','yes','n','no']: print 'Invalid input, please try again.' return 0