def doRestore(self,user_name,set_name): sys.stderr.write("Starting Restore\n") self.doClear(user_name) session = profileManager.doLogin(user_name) profile = UserProfile() restoreSet = self.getSet(set_name) for k,v in restoreSet.iteritems(): liveAnswer = profile.getAnswerFromSession(session,k) sys.stderr.write("[%8d] - %s\n" % (k,liveAnswer.Selected)) if liveAnswer.Selected != 0: sys.stderr.write("[%8d] - Already Set - we have a problem\n" % (k)) else: sys.stderr.write("[%8d] - Starting Set\n" % (k)) profile.setAnswerToSession(session,v)
setManager = SetManager() if (options.save or options.restore or options.examine or options.clear or options.diff): profileManager = ProfileManager() userName = args[0] if len(args) == 2: setName = args[1] if userName not in profileManager.getProfileNames(): sys.stderr.write("No such profile stored\n") sys.exit(0) if options.examine: session = profileManager.doLogin(userName) profile = UserProfile() if len(args) == 2: answer = profile.getAnswerFromSession(session,int(args[1])) sys.stderr.write("Answer [%s]\n" % answer) setManager.printSet({answer.Id : answer}) else: profile.loadFromSession(session,userName) setManager.printSet(profile.Answers) elif options.save: session = profileManager.doLogin(userName) profile = UserProfile() profile.loadFromSession(session,userName) setManager.saveSet(setName,profile.Answers) elif options.restore: setManager.doRestore(userName,setName) elif options.clear: setManager.doClear(userName) elif options.diff: