X-Git-Url: https://git.yukkurigames.com/?p=mlpccg-meta.git;a=blobdiff_plain;f=meta%2Fviews.py;h=a1cf06be9c5b3466152e379c4319112d6d88be1b;hp=fc1a7afa7c211ab561f8339e0a92f6c981bd58c8;hb=ac0c692e40ad6aca1b28199a4ef9c850fa680427;hpb=0f08a0d1ebcd8d825a7df448221a7eb23600bba5 diff --git a/meta/views.py b/meta/views.py index fc1a7af..a1cf06b 100644 --- a/meta/views.py +++ b/meta/views.py @@ -1,5 +1,10 @@ +import datetime +import difflib +import json from django.shortcuts import render -from meta.models import TournamentModel +from django.template import RequestContext +from django.http import HttpResponse +from meta.models import TournamentModel, RecordModel from mlpccg.CardDb import CARDDB from mlpccg.DeckList import DeckList from mlpccg.Clustering import Clustering @@ -7,21 +12,38 @@ from mlpccg.Clustering import Clustering def index(request): tournament_models = TournamentModel.objects.all().order_by('-date') tournament_data = [] - placements = [] for tournament in tournament_models: tournament_records = tournament.recordmodel_set.all().order_by('placement') - tournament_decks = [] - for record in tournament_records: - decklist = DeckList(name=record.decklist.name, url=record.decklist.url) - tournament_decks += [decklist] - if len(decklist.cards) > 0: - placements += [{'decklist': decklist, 'placement': record.placement}] - + tournament_decks = [DeckList(name=record.decklist.name, url=record.decklist.url) for record in tournament_records] tournament_data += [{'tournament': tournament, 'decklists': tournament_decks}] + records = RecordModel.objects.all() + placements = [{'decklist': DeckList(name=record.decklist.name, url=record.decklist.url), 'placement': record.placement} for record in records if record.decklist.url] clustering = Clustering(records=placements) ranking = [(int(avg), label, [decklist for decklist in clustering.clusters[label]]) for avg, label in clustering.ranking()] return render(request, 'index.html', { 'ranking': ranking, - 'tournaments': tournament_data}) + 'tournaments': tournament_data}, + context_instance=RequestContext(request)) + +def tournaments(request): + return HttpResponse(json.dumps([{'name': t.name} for t in TournamentModel.objects.all()]), content_type='application/json') + +def rate(request): + decklist = DeckList(url=request.POST['url']) + + records = RecordModel.objects.all() + placements = [{'decklist': DeckList(name=record.decklist.name, url=record.decklist.url), 'placement': record.placement, 'record': record} for record in records if record.decklist.url] + clustering = Clustering(records=placements) + label = clustering.predict(decklist) + + similiar = [] + for placement in placements: + for deck in clustering.clusters[label]: + if placement['decklist'] == deck: + similiar += [(deck, placement['record'])] + + return render(request, 'rate.html', { + 'deck': decklist, + 'similiar': similiar}, context_instance=RequestContext(request))