diff --git a/dashboard/views.py b/dashboard/views.py index d86e798..acb2382 100644 --- a/dashboard/views.py +++ b/dashboard/views.py @@ -1,198 +1,3 @@ -<<<<<<< HEAD -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}) -======= from captcha.helpers import captcha_image_url from captcha.models import CaptchaStore from django.contrib import messages @@ -386,4 +191,196 @@ def get_village(request): for i in village: res.append([i.code, i.name, i.level, i.pcode]) return JsonResponse({"village": res}) ->>>>>>> 8abe1a2f5f39f2a197eb308666a6c3267907510e +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})