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 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): #messages.debug(request, "Debug GLITCHHHHH") #messages.info(request, "This info is very educational") #messages.warning(request, "You got a warning") #messages.error(request, "Error!!!") #messages.success(request, "Great Success") 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: # FIXME: data initial my qso number 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 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, 'qsoPager': qsoPager, '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.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 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', {'qsoPager': qsoPager, '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.instance.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})