from django.shortcuts import render, get_object_or_404 from django.contrib.auth.decorators import login_required from django.contrib.admin.views.decorators import staff_member_required #from django.db.models import Q from django.contrib.auth.forms import AuthenticationForm, PasswordChangeForm from django.http import HttpResponseRedirect from django.contrib import messages from django.urls import reverse from django.contrib.auth import login as auth_login from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger from django.utils import timezone import datetime from .models import User, Contest, Frequency, Reference, QSO, ShadowCall from .forms import UpdateRefForm, QSOForm, QSOFormWithTime, CustomUserCreationForm, ShadowCallAddForm, UpdateCategoryForm def index(request): if request.user.is_authenticated(): return HttpResponseRedirect(reverse("contest:index")) return render(request, "index.html", {"loginForm": AuthenticationForm()}) @login_required def contestIndex(request): qsoform = QSOForm(request.user) contest = Contest.objects.get(id=1) qrgs = Frequency.objects.filter(band__contest=contest).order_by("channel") return render(request, 'contest/index.html', {"qsoform": qsoform, "contest": contest, "qrgs": qrgs}) @login_required def log(request): if not request.user.ref: return HttpResponseRedirect(reverse("contest:index")) form = None qsos = QSO.objects.filter(owner=request.user).order_by("-ownNo") if request.method == 'POST': form = QSOForm(user=request.user, data=request.POST) form.helper.form_tag = False if form.is_valid(): l = form.instance if not l.time: # set current time l.time = datetime.datetime.now() l.owner = request.user l.save() messages.success(request, "QSO saved!") return HttpResponseRedirect(reverse("contest:log")) else: data = { "ownNo": qsos[0].ownNo + 1 if len(qsos) > 0 else 1, "reportRX": "59", "reportTX": "59", } if qsos.count() > 0: data["band"] = qsos[0].band form = QSOForm(request.user, initial=data) form.helper.form_tag = False return render(request, 'contest/log.html', {'form': form, 'qsos': qsos}) @login_required def logEdit(request, qsoid): if not request.user.ref: return HttpResponseRedirect(reverse("contest:index")) qso = QSO.objects.get(id=qsoid, owner=request.user) form = None if request.method == 'POST': form = QSOFormWithTime(user=request.user, instance=qso, data=request.POST) if form.is_valid(): form.instance.save() messages.info(request, "QSO has been edited") return HttpResponseRedirect(reverse("contest:log")) else: form = QSOFormWithTime(user=request.user, instance=qso) return render(request, 'contest/logEdit.html', {'form': form, "qso": qso}) def logDelete(request, qsoid): if not request.user.ref: return HttpResponseRedirect(reverse("contest:index")) qso = QSO.objects.get(id=qsoid, owner=request.user) if request.method == 'POST': if "delete" in request.POST: if request.POST["delete"].lower() == "yes": qso.delete() messages.info(request, "QSO has been deleted") return HttpResponseRedirect(reverse("contest:log")) elif request.POST["delete"].lower() == "no": return HttpResponseRedirect(reverse("contest:log")) return render(request, 'contest/logDelete.html', {"qso": qso}) @staff_member_required def registerRefs(request): allUser = User.objects.all() shadows = ShadowCall.objects.all() qsos = QSO.objects.all().order_by("-time")[0:10] shadowForm = None if request.method == 'POST': shadowForm = ShadowCallAddForm(data=request.POST) if shadowForm.is_valid(): shadowForm.instance.save() messages.success(request, "Successfully added shadow user %s" % (shadowForm.instance.username,)) return HttpResponseRedirect(reverse("contest:registerRefs")) else: shadowForm = ShadowCallAddForm() return render(request, 'contest/registerRefs.html', {'alluser': allUser, "qsos": qsos, "shadowForm": shadowForm, "shadows": shadows}) def getPage(paginator, pageNo): try: return paginator.page(pageNo) except PageNotAnInteger: return paginator.page(1) except EmptyPage: return paginator.page(paginator.num_pages) @staff_member_required def recheckAllQSOs(request): if request.method == "POST": cnt = 0 for qso in QSO.objects.all(): if qso.checkQSOData(): print(qso) qso.save() cnt += 1 messages.success(request, "Allo QSO have been checked against each other (%d QSOs modified)" % (cnt,)) return HttpResponseRedirect(reverse("contest:registerRefs")) return render(request, "contest/checkAllQSOs.html", {}) @staff_member_required def viewUserQSOs(request, uid, page=1): user = get_object_or_404(User, id=uid) qsos = QSO.objects.filter(owner=user).order_by("-time") qsoPager = Paginator(qsos, 50) qsoPage = getPage(qsoPager, request.GET.get('page')) userRefs = set(map(lambda _x: _x["refStr"], user.qso_set.filter(ref__isnull=False).values("ref", "refStr"))) return render(request, "contest/viewUserQSOs.html", {'owner': user, 'qsos': qsos, 'qsoPage': qsoPage, 'userRefs': userRefs}) @staff_member_required def updateRef(request, shadow, uid): user = None form = None if shadow: user = get_object_or_404(ShadowCall, id=uid) else: user = get_object_or_404(User, id=uid) if request.method == 'POST': form = UpdateRefForm(data=request.POST) if form.is_valid(): ref = None if form.cleaned_data["existingRef"]: ref = form.cleaned_data["existingRef"] else: ref = Reference(name=form.cleaned_data["newRefName"]) ref.save() messages.info(request, "New Ref '%s' created" % ref) user.ref = ref user.regTime = form.cleaned_data["regTime"] user.location = form.cleaned_data["location"] user.opName = form.cleaned_data["opName"] user.save() messages.success(request, "%s%s ref set to %s" % ("(shadow) " if shadow else "", user, ref)) return HttpResponseRedirect(reverse("contest:registerRefs")) else: data = {} if user.ref: data["existingRef"] = user.ref if user.regTime: data["regTime"] = user.regTime else: data["regTime"] = timezone.now() data["opName"] = user.opName data["location"] = user.location form = UpdateRefForm(initial=data) return render(request, 'contest/updateRef.html', {'userobj': user, 'form': form, "shadow": shadow}) @staff_member_required def viewAllQSOs(request, page=1): qsos = QSO.objects.all().order_by("-time") qsoPager = Paginator(qsos, 10) qsoPage = getPage(qsoPager, request.GET.get('page')) return render(request, 'contest/viewAllQSOs.html', {'qsoPage': qsoPage}) def overview(request): # FIXME: Hardcoded for cqtu... everywhere c = Contest.objects.get(id=1) qrgs = Frequency.objects.filter(band__contest=c).order_by("channel") return render(request, 'contest/overview.html', {'contest': c, 'qrgs': qrgs}) def register(request): form = None if request.method == 'POST': form = CustomUserCreationForm(data=request.POST) if form.is_valid(): form.save() auth_login(request, form.instance) messages.info(request, "Registration as user %s successfull!" % form.instance.username) return HttpResponseRedirect(reverse("contest:index")) else: form = CustomUserCreationForm() return render(request, 'registration/register.html', {"form": form}) @login_required def profile(request): pwForm = None catForm = None if request.method == 'POST': if request.POST.get("submit", None) == "pwchange": pwForm = PasswordChangeForm(user=request.user, data=request.POST) if pwForm.is_valid(): pwForm.save() auth_login(request, pwForm.user) messages.success(request, "Password changed") return HttpResponseRedirect(reverse("profile")) else: catForm = UpdateCategoryForm(data=request.POST) if catForm.is_valid(): request.user.cat = catForm.cleaned_data["entry"] request.user.save() messages.success(request, "Entry category set") return HttpResponseRedirect(reverse("profile")) if not pwForm: pwForm = PasswordChangeForm(user=request.user) if not catForm: catForm = UpdateCategoryForm(initial={'entry': request.user.cat}) return render(request, 'registration/profile.html', {"pwForm": pwForm, "catForm": catForm})