diff --git a/dashboard/views.py b/dashboard/views.py index 623aa25..d86e798 100644 --- a/dashboard/views.py +++ b/dashboard/views.py @@ -1,3 +1,4 @@ +<<<<<<< HEAD from captcha.helpers import captcha_image_url from captcha.models import CaptchaStore from django.contrib import messages @@ -191,3 +192,198 @@ def get_village(request): for i in village: res.append([i.code, i.name, i.level, i.pcode]) return JsonResponse({"village": res}) +======= +from captcha.helpers import captcha_image_url +from captcha.models import CaptchaStore +from django.contrib import messages +from django.contrib.auth import logout, authenticate, login +from django.contrib.auth.models import User +from django.db.models import Q +from django.http import HttpResponseRedirect, JsonResponse +from django.shortcuts import render +import datetime + + +# Create your views here. +from dashboard.models import Userprofile, Organization, Area_code_2020, Weixin, Weibo, Toutiao, Qita + + +def refresh_captcha(request): + to_json_response = dict() + to_json_response['status'] = 1 + to_json_response['new_cptch_key'] = CaptchaStore.generate_key() + to_json_response['new_cptch_image'] = captcha_image_url(to_json_response['new_cptch_key']) + return JsonResponse(to_json_response) +def index(request): + weixin_count = Weixin.objects.all().count() + weibo_count = Weibo.objects.all().count() + toutiao_count = Toutiao.objects.all().count() + qita_count = Qita.objects.all().count() + return render(request,'dashboard/index.html',{'weixin_count':weixin_count,'weibo_count':weibo_count,'toutiao_count':toutiao_count,'qita_count':qita_count}) +def user_login(request): + username = None + password = None + hash_key = CaptchaStore.generate_key() + image_url = captcha_image_url(hash_key) + if request.method == 'POST': + if not request.POST.get('username'): + messages.error(request, '请输入用户名') + else: + username = request.POST.get('username') + if not request.POST.get('password'): + messages.error(request, '请输入密码') + else: + password = request.POST.get('password') + captcha_input = request.POST.get('captcha_1') + captcha_hashkey = request.POST.get('captcha_0') + + try: + CaptchaStore.objects.get(response=captcha_input.lower(), hashkey=captcha_hashkey, + expiration__gt=datetime.datetime.now()).delete() + except CaptchaStore.DoesNotExist: + messages.error(request, '验证码错误') + return HttpResponseRedirect('/login/') + if username is not None and password is not None: + try: + # user = authenticate(username=username, password=password) + user = User.objects.get(Q(username=username) | Q(first_name=username) | Q(email=username)) + if user is not None: + if user.check_password(password): + login(request, user) + return HttpResponseRedirect('/') + else: + messages.error(request, '无效的账号') + else: + messages.error(request, '账号或密码错误,请您确认账号和密码') + except: + messages.error(request, '账号或密码错误,请您确认账号和密码') + return render(request,'dashboard/login.html',{'hash_key':hash_key,'image_url':image_url}) +def user_logout(request): + logout(request) + return HttpResponseRedirect('/') +def register(request): + username = None + email = None + password = None + confirm_password = None + name = None + phone = None + image = None + flag = False + o = None + sex = None + hash_key = CaptchaStore.generate_key() + image_url = captcha_image_url(hash_key) + organization = Organization.objects.all() + if request.method == 'POST': + if not request.POST.get('organization'): + messages.error(request, '请选择单位') + else: + o = request.POST.get('organization') + if not request.POST.get('name'): + messages.error(request, '请输入姓名') + else: + name = request.POST.get('name') + if not request.POST.get('email'): + messages.error(request, '请输入邮箱') + else: + email = request.POST.get('email') + if not request.POST.get('phone'): + messages.error(request, '请输入电话') + else: + phone = request.POST.get('phone') + if not request.POST.get('username'): + messages.error(request, '请输入用户名') + else: + username = request.POST.get('username') + + if not request.POST.get('password'): + messages.error(request, '请输入密码') + else: + password = request.POST.get('password') + if not request.POST.get('confirm_password'): + messages.error(request, '请再次输入密码') + else: + confirm_password = request.POST.get('confirm_password') + if not request.FILES.get('image'): + messages.error(request, '请选择头像') + else: + image = request.FILES.get('image') + print(str(image)+"1111111111111111111111111111111111111111111") + if request.POST.get('sex') == 0: + sex = '男' + else: + sex = '女' + captcha_input = request.POST.get('captcha_1') + captcha_hashkey = request.POST.get('captcha_0') + + try: + CaptchaStore.objects.get(response=captcha_input.lower(), hashkey=captcha_hashkey, + expiration__gt=datetime.datetime.now()).delete() + except CaptchaStore.DoesNotExist: + messages.error(request, '验证码错误') + return HttpResponseRedirect('/register/') + if password is not None and confirm_password is not None: + if password == confirm_password: + flag = True + else: + messages.error(request, '两次输入的密码不一致,请重新输入') + + filter_result = User.objects.filter(username=username) + if len(filter_result) > 0: + messages.error(request, '对不起,您输入的用户名已被注册') + return render(request, "dashboard/register.html") + + if username is not None and password is not None and confirm_password is not None and email is not None and flag: + user = User.objects.create_user(username, email, password) + user.is_active = True + user.is_staff = True + user.first_name = phone + user.save() + userprofile = Userprofile(name=name,image=image,user_id=user.id,organization_id=o,sex=sex) + userprofile.save() + messages.success(request, '注册成功,请登录') + return HttpResponseRedirect('/login/') + return render(request, 'dashboard/register.html', {'hash_key': hash_key, 'image_url': image_url,'organization':organization}) + +def get_province(request): + #省 + province = Area_code_2020.objects.filter(level=1) + res = [] + for i in province: + res.append([i.code,i.name,i.level,i.pcode]) + return JsonResponse({"province":res}) +def get_city(request): + code = request.GET.get('code') + print(code) + #市 + cities = Area_code_2020.objects.filter(pcode=code) + res = [] + for i in cities: + res.append([i.code,i.name,i.level,i.pcode]) + return JsonResponse({"city": res}) +def get_district(request): + code = request.GET.get('code') + #县 + district = Area_code_2020.objects.filter(pcode=code) + res = [] + for i in district: + res.append([i.code, i.name, i.level, i.pcode]) + return JsonResponse({"district": res}) +def get_town(request): + code = request.GET.get('code') + # 乡 + town = Area_code_2020.objects.filter(pcode=code) + res = [] + for i in town: + res.append([i.code, i.name, i.level, i.pcode]) + return JsonResponse({"town": res}) +def get_village(request): + code = request.GET.get('code') + # 村 + village = Area_code_2020.objects.filter(pcode=code) + res = [] + for i in village: + res.append([i.code, i.name, i.level, i.pcode]) + return JsonResponse({"village": res}) +>>>>>>> 8abe1a2f5f39f2a197eb308666a6c3267907510e