def writeInAllSets(self): try: sets = Set.where().all() except: raise Exception("Error in connection to database") setNames = "" for set in sets: setNames += set.code + "|" + set.name + "|" if set.block is not None: setNames += set.block + "\n" else: setNames += "\n" setNames = setNames.rstrip("\n") with open(self.filePathSets, "w") as file: file.write(setNames)
def form_valid(self, form): form.instance.user = self.request.user versions = [] name = form.cleaned_data['name'] set_name = form.cleaned_data['set_name'] sets = MtgSet.where(name=set_name).all() if len(sets) > 0: code = sets[0].code versions = MtgCard.where(name=name, set=code).all() else: versions = MtgCard.where(name=name).all() if len(versions) > 0: retval = versions[0].image_url else: retval = "https://static.wikia.nocookie.net/mtgsalvation_gamepedia/images/f/f8/Magic_card_back.jpg" if form.is_valid(): obj = form.save(commit=False) obj.image_url = retval #form.cleaned_data['image_url'] = retval obj.save() return super().form_valid(form)
def readSetsFromOnline(self): try: setObjects = Set.where().all() except OperationalError: raise OperationalError("Error in connection to database") sets = [] setInfo = [] for set in setObjects: setInfo = [ set.code, set.name.replace(u"\u2014", "-").replace(u"\u2022", "*") ] if set.block is None: setInfo.append("") else: setInfo.append(set.block) sets.append(setInfo) return sets
def test_where_filters_on_name(self): with vcr.use_cassette('fixtures/filtered_sets.yaml'): sets = Set.where(name='khans of tarkir promos').all() self.assertEqual(1, len(sets)) self.assertEqual('PKTK', sets[0].code)
def test_where_filters_on_name(self): with vcr.use_cassette('fixtures/filtered_sets.yaml'): sets = Set.where(name='khans').all() self.assertEqual(1, len(sets)) self.assertEqual('KTK', sets[0].code)