newmediamonitoring/dashboard/views.py

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})