django crm01

客户管理系统

01 实现登录注册，利用modelform 添加用户

from django.shortcuts import render,HttpResponse,redirect
import os
from django.contrib import auth
from app01.models import UserInfo
from crm import settings
from django.views import View
from django import forms
from app01 import models
from django.utils.decorators import method_decorator
from django.contrib.auth.decorators import login_required
# Create your views here.



class RegForm(forms.Form):
    username = forms.CharField(
        label='用户名',
        max_length=12,
        min_length=6,
        error_messages={
            'max_length':'用户名不能超过12位',
            'min_length':'用户名不能低于6位',
            'required':'用户名不能为空',
        }
    )
    password = forms.CharField(
        label='密码',
        max_length=16,
        min_length=6,
        error_messages={
            'max_length': '密码不能超过12位',
            'min_length': '密码不能低于6位',
            'required': '密码不能为空',
        },
        widget=forms.widgets.PasswordInput(render_value=True),

    )

    r_password = forms.CharField(
        label='密码',
        max_length=16,
        min_length=6,
        error_messages={
            'max_length': '密码不能超过12位',
            'min_length': '密码不能低于6位',
            'required': '密码不能为空',
        },
        widget=forms.widgets.PasswordInput,

    )



    def __init__(self,*args,**kwargs):
        super().__init__(*args,**kwargs)
        for field in self.fields:
            self.fields[field].widget.attrs.update({'class':'form-control'})

def get_valid_img(request):

    from PIL import Image
    #终极版，方式5
    import random
    def get_random_color():
        return (random.randint(0,255),random.randint(0,255),random.randint(0,255))
    from PIL import Image,ImageDraw,ImageFont
    img_obj = Image.new('RGB', (236, 34), get_random_color()) #图片对象
    draw_obj = ImageDraw.Draw(img_obj)  #通过图片对象生成一个画笔对象
    font_path = os.path.join(settings.BASE_DIR,'statics/font/ziti.TTF') #获取字体,注意有些字体文件不能识别数字，所以如果数字显示不出来，就换个字体
    font_obj = ImageFont.truetype(font_path,16) #创建字体对象
    sum_str = ''  #这个数据就是用户需要输入的验证码的内容
    for i in range(6):
        a = random.choice([str(random.randint(0,9)), chr(random.randint(97,122)), chr(random.randint(65,90))])  #4  a  5  D  6  S
        sum_str += a
    print(sum_str)
    draw_obj.text((84,10),sum_str,fill=get_random_color(),font=font_obj) #通过画笔对象，添加文字

    width=236
    height=34
    # 添加噪线
    for i in range(5): #添加了5条线
        #一个坐标表示一个点，两个点就可以连成一条线
        x1=random.randint(0,width)
        x2=random.randint(0,width)
        y1=random.randint(0,height)
        y2=random.randint(0,height)
        draw_obj.line((x1,y1,x2,y2),fill=get_random_color())
    # # 添加噪点
    for i in range(10):
        #这是添加点，50个点
        draw_obj.point([random.randint(0, width), random.randint(0, height)], fill=get_random_color())
        #下面是添加很小的弧线，看上去类似于一个点，50个小弧线
        x = random.randint(0, width)
        y = random.randint(0, height)
        draw_obj.arc((x, y, x + 4, y + 4), 0, 90, fill=get_random_color()) #x, y是弧线的起始点位置，x + 4, y + 4是弧线的结束点位置

    from io import BytesIO
    f = BytesIO()  #操作内存的把手
    img_obj.save(f,'png')  #
    data = f.getvalue()

    # 存这个验证码的方式1：赋值给全局变量的简单测试
    # global valid_str
    # valid_str = sum_str
    # 方式2：将验证码存在各用户自己的session中,session的应用其实还有很多
    request.session['valid_str'] = sum_str
    return HttpResponse(data)


class Login(View):
    def get(self,request):
        return render(request,'login.html')
    def post(self,request):
        data = request.POST
        if data.get('code') == request.session['valid_str']:
            user_obj = auth.authenticate(username=data.get('username'),password=data.get('password'))
            if user_obj:
                auth.login(request,user_obj)
                return redirect('index')
            else:
                return redirect('login')
        else:
            return render(request,'login.html')

class Register(View):
    def get(self, request):
        form_obj = RegForm()
        return render(request, 'register.html',{"form_obj":form_obj})
    def post(self,request):
        data = request.POST
        form_obj = RegForm(data)
        code = data.get('code')
        error = ''
        if code == request.session['valid_str']:
            if data.get('password') == data.get('r_password'):
                if form_obj.is_valid():
                    user_obj = UserInfo.objects.create_user(username=data.get('username'),password=data.get('password'))
                    return redirect('login')
            else:
                erro = '两次密码不一致'
        else:
            erro = '验证码错误'
        return render(request, 'register.html',{"form_obj":form_obj,'error':erro})

#
from app01.page import PageNation
class Index(View):
    @method_decorator(login_required())
    def dispatch(self, request, *args, **kwargs):
        ret = super(Index, self).dispatch(request,*args,**kwargs)
        return ret

    def get(self,request):

        all_costomer = models.Customer.objects.all()
        #self,base_url,current_page_num,total_counts,per_page_counts=10,page_number=5
        costomer_obj = PageNation(request.path,request.GET.get('page'),all_costomer.count())
        start_num = costomer_obj.start_num
        end_num = costomer_obj.end_num
        page_html = costomer_obj.page_html()
        # print(page_html)
        costomers = all_costomer[start_num:end_num]
        return render(request,'index.html',{'all_costomer':costomers,'page_html':page_html})

# @login_required()
# def index(request):
#     return render(request,'index.html')

class Logout(View):

    def get(self,request):
        auth.logout(request)
        return redirect('login')

class Start(View):

    def get(self,request):
        return render(request,'starter.html')


from app01 import myform
class Addcostomer(View):
    def get(self,request):
        customerform = myform.CustomerForm()
        return render(request,'addcustomer.html',{'customerform':customerform})


    def post(self,request):
        data = request.POST
        customerform = myform.CustomerForm(data)
        if customerform.is_valid():
            customerform.save()
            return redirect('index')
        else:
            return render(request,'index.html',{'customerform':customerform})