def test_get_flashcard_set(self): setID = 19049486 cardSet = quizlet.get_flashcard_set(setID) self.assertEqual(cardSet['http_code'], 200) self.assertEqual(cardSet['id'], setID) self.assertEqual(cardSet['creator']['username'], "JoeQuery") setID = -1 cardSet = quizlet.get_flashcard_set(setID) self.assertEqual(cardSet['http_code'], 410)
def test_load_flashcard_set_terms_from_file(self): f = tempfile.TemporaryFile() setID = 19049486 cardSet = quizlet.get_flashcard_set(setID) quizlet.save_flashcard_set_terms_to_file(cardSet, f) f.seek(0) terms = quizlet.load_flashcard_set_terms_from_file(f) f.close() self.assertEqual(637906377, terms[0]['id'])
def test_save_flashcard_set_terms_to_file(self): f = tempfile.TemporaryFile() setID = 19049486 cardSet = quizlet.get_flashcard_set(setID) quizlet.save_flashcard_set_terms_to_file(cardSet, f) f.seek(0) termJSON = f.read() terms = json.loads(termJSON) f.close() self.assertEqual(637906377, terms[0]['id'])
def download_flashcard_set(setID): flashcardSet = quizlet.get_flashcard_set(setID) if (flashcardSet['http_code'] != 200): print("Unable to access flashcard set %s" % setID) return # Get computer friendly set name cardURL = flashcardSet['url'].split('/')[-2] setFilename = cardURL + ".quiz" setPath = os.path.join(quizlet.SET_DIR, setFilename) f = open(setPath, 'w') quizlet.save_flashcard_set_terms_to_file(flashcardSet, f) f.close() title = flashcardSet['title'] print("Downloaded '%s' set to sets/%s" % (title, setFilename))