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)
