對於stark(curd)插件的使用簡單介紹

一、創建表

from django.db import models

from django.db import models

class Department(models.Model):
    """
    部門表
    市場部     1000
    銷售      1001
    """
    title = models.CharField(verbose_name='部門名稱', max_length=16)
    code = models.IntegerField(verbose_name='部門編號',unique=True,null=False)

    def __str__(self):
        return self.title


class UserInfo(models.Model):
    """
    員工表
    """
    # auth = models.OneToOneField(verbose_name='用戶權限', to=rbac_model.User)
    name = models.CharField(verbose_name='員工姓名', max_length=16)
    username = models.CharField(verbose_name='用戶名', max_length=32)
    password = models.CharField(verbose_name='密碼', max_length=64)
    email = models.EmailField(verbose_name='郵箱', max_length=64)

    depart = models.ForeignKey(verbose_name='部門', to="Department",to_field="code")

    def __str__(self):
        return self.name


class Course(models.Model):
    """
    課程表
    如：
        Linux基礎
        Linux架構師
        Python自動化開發精英班
        Python自動化開發架構師班
    """
    name = models.CharField(verbose_name='課程名稱', max_length=32)

    def __str__(self):
        return self.name


class School(models.Model):
    """
    校區表
    如：
        北京海淀校區
        北京昌平校區
        上海虹口校區
        廣州白雲山校區
    """
    title = models.CharField(verbose_name='校區名稱', max_length=32)

    def __str__(self):
        return self.title


class ClassList(models.Model):
    """
    班級表
    如：
        Python全棧  面授班  5期  10000  2017-11-11  2018-5-11
    """
    school = models.ForeignKey(verbose_name='校區', to='School')
    course = models.ForeignKey(verbose_name='課程名稱', to='Course')

    semester = models.IntegerField(verbose_name="班級(期)")
    price = models.IntegerField(verbose_name="學費")
    start_date = models.DateField(verbose_name="開班日期")
    graduate_date = models.DateField(verbose_name="結業日期", null=True, blank=True)
    memo = models.CharField(verbose_name='說明', max_length=256, blank=True, null=True)
    teachers = models.ManyToManyField(verbose_name='任課老師', to='UserInfo', related_name='teach_classes')
    tutor = models.ForeignKey(verbose_name='班主任', to='UserInfo', related_name='classes')

    def __str__(self):
        return "{0}({1}期)".format(self.course.name, self.semester)


class Customer(models.Model):
    """
    客戶表
    """
    qq = models.CharField(verbose_name='qq', max_length=64, unique=True, help_text='QQ號必須唯一')

    name = models.CharField(verbose_name='學生姓名', max_length=16)
    gender_choices = ((1, '男'), (2, '女'))
    gender = models.SmallIntegerField(verbose_name='性別', choices=gender_choices)

    education_choices = (
        (1, '重點大學'),
        (2, '普通本科'),
        (3, '獨立院校'),
        (4, '民辦本科'),
        (5, '大專'),
        (6, '民辦專科'),
        (7, '高中'),
        (8, '其他')
    )
    education = models.IntegerField(verbose_name='學歷', choices=education_choices, blank=True, null=True, )
    graduation_school = models.CharField(verbose_name='畢業學校', max_length=64, blank=True, null=True)
    major = models.CharField(verbose_name='所學專業', max_length=64, blank=True, null=True)

    experience_choices = [
        (1, '在校生'),
        (2, '應屆畢業'),
        (3, '半年以內'),
        (4, '半年至一年'),
        (5, '一年至三年'),
        (6, '三年至五年'),
        (7, '五年以上'),
    ]
    experience = models.IntegerField(verbose_name='工作經驗', blank=True, null=True, choices=experience_choices)
    work_status_choices = [
        (1, '在職'),
        (2, '無業')
    ]
    work_status = models.IntegerField(verbose_name="職業狀態", choices=work_status_choices, default=1, blank=True,
                                      null=True)
    company = models.CharField(verbose_name="目前就職公司", max_length=64, blank=True, null=True)
    salary = models.CharField(verbose_name="當前薪資", max_length=64, blank=True, null=True)

    source_choices = [
        (1, "qq群"),
        (2, "內部轉介紹"),
        (3, "官方網站"),
        (4, "百度推廣"),
        (5, "360推廣"),
        (6, "搜狗推廣"),
        (7, "騰訊課堂"),
        (8, "廣點通"),
        (9, "高校宣講"),
        (10, "渠道代理"),
        (11, "51cto"),
        (12, "智匯推"),
        (13, "網盟"),
        (14, "DSP"),
        (15, "SEO"),
        (16, "其它"),
    ]
    source = models.SmallIntegerField('客戶來源', choices=source_choices, default=1)
    referral_from = models.ForeignKey(
        'self',
        blank=True,
        null=True,
        verbose_name="轉介紹自學員",
        help_text="若此客戶是轉介紹自內部學員,請在此處選擇內部學員姓名",
        related_name="internal_referral"
    )
    course = models.ManyToManyField(verbose_name="咨詢課程", to="Course")

    status_choices = [
        (1, "已報名"),
        (2, "未報名")
    ]
    status = models.IntegerField(
        verbose_name="狀態",
        choices=status_choices,
        default=2,
        help_text=u"選擇客戶此時的狀態"
    )
    consultant = models.ForeignKey(verbose_name="課程顧問", to='UserInfo', related_name='consultant',limit_choices_to={'depart_id':1001})
    date = models.DateField(verbose_name="咨詢日期", auto_now_add=True)
    last_consult_date = models.DateField(verbose_name="最后跟進日期", auto_now_add=True)

    def __str__(self):
        return "姓名:{0},QQ:{1}".format(self.name, self.qq, )


class ConsultRecord(models.Model):
    """
    客戶跟進記錄
    """
    customer = models.ForeignKey(verbose_name="所咨詢客戶", to='Customer')
    consultant = models.ForeignKey(verbose_name="跟蹤人", to='UserInfo')
    date = models.DateField(verbose_name="跟進日期", auto_now_add=True)
    note = models.TextField(verbose_name="跟進內容...")


class PaymentRecord(models.Model):
    """
    繳費記錄
    """
    customer = models.ForeignKey(Customer, verbose_name="客戶")

    class_list = models.ForeignKey(verbose_name="班級", to="ClassList", blank=True, null=True)

    pay_type_choices = [
        (1, "訂金/報名費"),
        (2, "學費"),
        (3, "轉班"),
        (4, "退學"),
        (5, "退款"),
    ]
    pay_type = models.IntegerField(verbose_name="費用類型", choices=pay_type_choices, default=1)
    paid_fee = models.IntegerField(verbose_name="費用數額", default=0)
    turnover = models.IntegerField(verbose_name="成交金額", blank=True, null=True)
    quote = models.IntegerField(verbose_name="報價金額", blank=True, null=True)
    note = models.TextField(verbose_name="備注", blank=True, null=True)
    date = models.DateTimeField(verbose_name="交款日期", auto_now_add=True)
    consultant = models.ForeignKey(verbose_name="負責老師", to='UserInfo', help_text="誰簽的單就選誰")


class Student(models.Model):
    """
    學生表（已報名）
    """
    customer = models.OneToOneField(verbose_name='客戶信息', to='Customer')

    username = models.CharField(verbose_name='用戶名', max_length=32)
    password = models.CharField(verbose_name='密碼', max_length=64)
    emergency_contract = models.CharField(max_length=32, blank=True, null=True, verbose_name='緊急聯系人')
    class_list = models.ManyToManyField(verbose_name="已報班級", to='ClassList', blank=True)

    company = models.CharField(verbose_name='公司', max_length=128, blank=True, null=True)
    location = models.CharField(max_length=64, verbose_name='所在區域', blank=True, null=True)
    position = models.CharField(verbose_name='崗位', max_length=64, blank=True, null=True)
    salary = models.IntegerField(verbose_name='薪資', blank=True, null=True)
    welfare = models.CharField(verbose_name='福利', max_length=256, blank=True, null=True)
    date = models.DateField(verbose_name='入職時間', help_text='格式yyyy-mm-dd', blank=True, null=True)
    memo = models.CharField(verbose_name='備注', max_length=256, blank=True, null=True)

    def __str__(self):
        return self.username


class CourseRecord(models.Model):
    """
    上課記錄表
    """
    class_obj = models.ForeignKey(verbose_name="班級", to="ClassList")
    day_num = models.IntegerField(verbose_name="節次", help_text=u"此處填寫第幾節課或第幾天課程...,必須為數字")
    teacher = models.ForeignKey(verbose_name="講師", to='UserInfo')
    date = models.DateField(verbose_name="上課日期", auto_now_add=True)

    course_title = models.CharField(verbose_name='本節課程標題', max_length=64, blank=True, null=True)
    course_memo = models.TextField(verbose_name='本節課程內容概要', blank=True, null=True)
    has_homework = models.BooleanField(default=True, verbose_name="本節有作業")
    homework_title = models.CharField(verbose_name='本節作業標題', max_length=64, blank=True, null=True)
    homework_memo = models.TextField(verbose_name='作業描述', max_length=500, blank=True, null=True)
    exam = models.TextField(verbose_name='踩分點', max_length=300, blank=True, null=True)

    def __str__(self):
        return "{0} day{1}".format(self.class_obj, self.day_num)


class StudyRecord(models.Model):
    course_record = models.ForeignKey(verbose_name="第幾天課程", to="CourseRecord")
    student = models.ForeignKey(verbose_name="學員", to='Student')
    record_choices = (('checked', "已簽到"),
                      ('vacate', "請假"),
                      ('late', "遲到"),
                      ('noshow', "缺勤"),
                      ('leave_early', "早退"),
                      )
    record = models.CharField("上課紀錄", choices=record_choices, default="checked", max_length=64)
    score_choices = ((100, 'A+'),
                     (90, 'A'),
                     (85, 'B+'),
                     (80, 'B'),
                     (70, 'B-'),
                     (60, 'C+'),
                     (50, 'C'),
                     (40, 'C-'),
                     (0, ' D'),
                     (-1, 'N/A'),
                     (-100, 'COPY'),
                     (-1000, 'FAIL'),
                     )
    score = models.IntegerField("本節成績", choices=score_choices, default=-1)
    homework_note = models.CharField(verbose_name='作業評語', max_length=255, blank=True, null=True)
    note = models.CharField(verbose_name="備注", max_length=255, blank=True, null=True)

    homework = models.FileField(verbose_name='作業文件', blank=True, null=True, default=None)
    stu_memo = models.TextField(verbose_name='學員備注', blank=True, null=True)
    date = models.DateTimeField(verbose_name='提交作業日期', auto_now_add=True)

    def __str__(self):
        return "{0}-{1}".format(self.course_record, self.student)

二、注冊表需要知道的知識點

1、部門表相關

　　需求：默認編輯不要了，通過字段變成可點擊的，具體是哪個字段不知道，可以自定義

edit_link = ["title"]

 

方法：拿到字段后判斷這個字段是都在edit_link里面，
　　　　- 值變成可點擊的
　　　　- 反向生成url
　　　　- 獲取url的參數，拼接成_listfilter

具體實現：

    StarkConfig類
      # =============編輯鏈接================
        edit_link = []
        def get_edit_link(self):
            result = []
            if self.edit_link:
                result.extend(self.edit_link)
            return result

    ChangeList類
    self.edit_link = config.get_edit_link()

    def edit_link_tag(self,pk,text):
            params = QueryDict(mutable=True)
            query_str = self.request.GET.urlencode()  #page=1
            params[self.config._query_param_key] = query_str
            #要跳轉的路徑http://127.0.0.1:8080/index/crm/department/1/change/?_listfilter=None%3D
            return mark_safe("<a href='%s?%s'>%s</a>"%(self.config.get_change_url(pk),params.urlencode(),text))

        
    def body_list():
         #用於定制編輯列
         if field_name in self.edit_link:
            val = self.edit_link_tag(row.pk,val)

 

4、=============重寫get_list_display()方法=================
    實現：
     # 重寫get_list_display(),因為父類有這個方法，如果這里不寫就會繼承父類的，寫了就優先執行自己的
        def get_list_display(self):
            result = []

            result.extend(self.list_display)
            result.insert(0,v1.StarkConfig.checkbox)
            result.append(v1.StarkConfig.edit)
            result.append(v1.StarkConfig.delete)

            return result
  重寫構造方法需要注意的： 函數和方法的區別：由於我們 val = field_name(self.config, row) ，傳了一個self， 所以用第二種方式。用方式一就報錯了 方式一：self.edit自己沒有就找父類的，自己有就用自己的 方式二；v1.StarkConofig.edit 如果是函數自己寫一個self =========寫代碼的時候注意加注釋==========================
        對類的注釋
        對於文件的注釋
        對於函數的注釋
   

2、員工表

　　1、錯誤信息的顯示
　　　　add_view
　　　　給errors加上個樣式

2、自定義ModelForm
      def get_model_form_class(self):
        '''自定義ModelForm'''
        class MyModelForm(ModelForm):
            class Meta:
                model = models.UserInfo
                fields = "__all__"
                error_messages = {
                    "name":{"required":"姓名不能為空"},
                    "username":{"required":"用戶名不能為空"},
                    "password":{"required":"密碼不能為空"},
                    "email":{"required":"郵箱不能為空","invalid":"郵箱格式不正確"},
                    "depart":{"required":"用戶名不能不選",},
                }
        return MyModelForm

　　3、組合搜索的篩選條件出錯了:

組合搜索的篩選條件出錯了:
            出的錯誤：當你看到明明有數據，但是卻沒有查到
            出錯原因：是由於關聯的字段變了depart = models.ForeignKey(verbose_name='部門', to="Department",to_field="code")
                      以前關聯的是pk,現在關聯的是code。所以對應的值就不一樣了。所以找不到了
                      
            解決辦法：
                在FilterOption又加了兩個參數
                text_func_name   ：組合搜素時，組合搜索時，頁面上生成顯示文本的函數
                val_func_name  ：組合搜素時，頁面上生成a標簽的value值的函數，也就是pk
                class FilterOption:
                     def __init__(self, field_name, is_multi=False, is_choice=False, condition=None,text_func_name=None,val_func_name=None):
                        self.text_func_name = text_func_name
                        self.val_func_name = val_func_name
                        
        # ================普通按鈕====================
        for val in self.data:   #單選
            if self.option.is_choice: # ((1,'男'),(2,'女'),(1,'男'))
                pk, text = str(val[0]),val[1]
            else:  # 外鍵或多對多
                '''現在的做法，是由於關聯的字段變了depart = models.ForeignKey(verbose_name='部門', to="Department",to_field="code")'''
                   text = str(self.option.text_func_name(val)) if self.option.text_func_name else str(val)
                   pk = str(self.option.val_func_name(val)) if self.option.val_func_name else str(val.pk)
                #原來的做法 pk, text = str(val.pk), str(val)
        stark.py
             comb_filter = [
                    v1.FilterOption("depart",val_func_name=lambda x: x.code,),
             ]

stark.py

comb_filter = [
        v1.FilterOption("depart",val_func_name=lambda x: x.code,),
    ]  #分組搜索

3、班級管理

4、學校

5、課程

6、客戶以及跟進記錄

客戶信息;
   需求： 咨詢課程
    吧咨詢課程以標簽的形式顯示
    並且加樣式一點擊X就給刪除了，
    額外的增加個url
    def extra_url(self):
        ...
        
    客戶的狀態。對於未報名的學生，有個跟進記錄
    跟進記錄：一點跳轉到詳細的根進記錄，這個根進記錄的url
    加上是否顯示組合搜索。。。。。
    自己可以查看自己的根進記錄，不可以查看別人的（自己寫一個視圖函數
        繼承父類的方法，
    ）

代碼實現

class ConsultRecordConfig(v1.StarkConfig):
    def customer_display(self,obj=None,is_header=False):
        if is_header:
            return "所咨詢客戶"
        return obj.customer.name
    list_display = [customer_display,"consultant","date","note"]
    comb_filter = [
        v1.FilterOption("customer"),
    ]   #組合搜索默認不顯示，但是卻又篩選的功能

    def change_views(self, request,*args, **kwargs):
        customer = request.GET.get('customer')
        # session中獲取當前用戶ID
        current_login_user_id = 6
        ct = models.Customer.objects.filter(consultant=current_login_user_id, id=customer).count()
        if not ct:
            return HttpResponse('無權訪問.')
        return super(ConsultRecordConfig,self).change_views(request, *args, **kwargs)
v1.site.register(models.ConsultRecord,ConsultRecordConfig)

v1.py
 # =============組合搜索====================
    comb_filter = []
    def get_comb_filter(self):
        result = []
        if self.comb_filter:
            result.extend(self.comb_filter)
        return result

    show_comb_filter = False
    def get_show_comb_filter(self):
        return self.show_comb_filter
    
self.show_comb_filter = config.get_show_comb_filter()
在頁面
  {% if cl.show_comb_filter %}
           <div class="list-filter">
            {% for item in cl.gen_comb_filter %}
                <div>
                {% for col in item %}
                      {{ col }}
                {% endfor %}
                </div>
            {% endfor %}
        </div>
        {% endif %}

圖示：

三、注冊表的具體使用方法

#!usr/bin/env python
# -*- coding:utf-8 -*-
from django.forms import ModelForm
from django.shortcuts import render,HttpResponse,redirect
from django.utils.safestring import mark_safe
from stark.service import v1
from crm import models
from django.conf.urls import url

class DepartmentConfig(v1.StarkConfig):
    list_display = ["title","code"]
    edit_link = ["title"]   #自定制鏈接，指定字段可編輯

    # 重寫get_list_display(),因為父類有這個方法，如果這里不寫就會繼承父類的，寫了就優先執行自己的
    def get_list_display(self):
        result = []

        result.extend(self.list_display)
        result.insert(0,v1.StarkConfig.checkbox)
        result.append(v1.StarkConfig.edit)
        result.append(v1.StarkConfig.delete)

        return result

v1.site.register(models.Department,DepartmentConfig)

class UserInfoConfig(v1.StarkConfig):
    edit_link = ["name"]

    def depart_dispaly(self,obj=None,is_header=False):
        if is_header:
            return "所屬部門"
        return obj.depart.title

    def get_model_form_class(self):
        '''自定義ModelForm'''
        class MyModelForm(ModelForm):
            class Meta:
                model = models.UserInfo
                fields = "__all__"
                error_messages = {
                    "name":{"required":"姓名不能為空"},
                    "username":{"required":"用戶名不能為空"},
                    "password":{"required":"密碼不能為空"},
                    "email":{"required":"郵箱不能為空","invalid":"郵箱格式不正確"},
                    "depart":{"required":"用戶名不能不選",},
                }
        return MyModelForm

    list_display = ["name","username","email",depart_dispaly]

    comb_filter = [
        v1.FilterOption("depart",val_func_name=lambda x: x.code,),
    ]  #分組搜索

    def delete_view(self, request,nid, *args, **kwargs):
        '''重寫視圖函數'''
        if request.method=="GET":
            return render(request,"stark/delete_view.html",{"quxiao_url":self.get_list_url()})
        else:
            self.model_class.objects.filter(pk=nid).delete()
            return redirect(self.get_list_url())

v1.site.register(models.UserInfo,UserInfoConfig)

class CourseConfig(v1.StarkConfig):
    list_display = ["name"]
    edit_link = ["name"]
    show_actions =True  #顯示actions

    def mutil_delete(self,request):
        if request.method =="POST":
            pk_list = request.POST.getlist("pk")
            print(pk_list,"000")
            self.model_class.objects.filter(id__in=pk_list).delete()

    mutil_delete.short_desc = "批量刪除"
    def init_func(self):
        pass
    init_func.short_desc = "初始化"
    actions = [mutil_delete,init_func]   #actios操作


    search_fields = ["name__contains"]   #按照name搜索
    show_search_form = True

v1.site.register(models.Course,CourseConfig)

class SchoolConfig(v1.StarkConfig):
    list_display = ["title"]
    edit_link = ["title"]

v1.site.register(models.School,SchoolConfig)

class ClassListConfig(v1.StarkConfig):
    def teachers_display(self,obj=None,is_header=False):
        if is_header:
            return "任教老師"
        user_list = obj.teachers.all()
        html = []
        for i in user_list:
            html.append(i.name)
        return ','.join(html)

    def display_graduate_date(self,obj=None,is_header=False):
        if is_header:
            return "結業日期"
        return '' if not obj.graduate_date else obj.graduate_date

    def  display_memo(self,obj=None,is_header=False):
        if is_header:
            return "說明"
        return '' if not obj.memo else obj.memo

    def course_semester(self,obj=None,is_header=False):
        if is_header:
            return "課程（班級）"
        return "%s(%s期)"%(obj.course,obj.semester)

    #列舉這個班級的人數
    def num(self,obj=None,is_header=False):
        if is_header:
            return "人數"
        print(obj.student_set.all().count())
        return obj.student_set.all().count()
    list_display = ["school",course_semester,num,"price","start_date",display_graduate_date,display_memo,teachers_display,"tutor"]
    edit_link = ["school"]
v1.site.register(models.ClassList,ClassListConfig)

class CustomerConfig(v1.StarkConfig):
    def display_gender(self,obj=None,is_header=False):
        if is_header:
            return "性別"
        return obj.get_gender_display()
    def display_education(self,obj=None,is_header=False):
        if is_header:
            return "學歷"
        return obj.get_education_display()

    def display_status(self, obj=None, is_header=False):
        if is_header:
            return '狀態'
        return obj.get_status_display()
    def recode(self, obj=None, is_header=False):
        if is_header:
            return "跟進記錄"
        return mark_safe("<a href='/index/crm/consultrecord/?customer=%s'>查看跟進記錄</a>" %(obj.pk,))

    def display_course(self,obj=None, is_header=False):
        if is_header:
            return "咨詢課程"
        course_list = obj.course.all()
        html = []
        for item in course_list:
            temp = "<a style='display:inline-block;padding:3px 5px;border:1px solid blue;margin:2px;' href='/index/crm/customer/%s/%s/dc/'>%s X</a>" %(obj.pk,item.pk,item.name)
            html.append(temp)
        return "".join(html)

    def extra_urls(self):
        # 由於沒有路徑，我們可以額外的增加一個路徑,重新走一個delete_course視圖
        app_model_name = (self.model_class._meta.app_label, self.model_class._meta.model_name)
        urlpatterns =[
            url(r'^(\d+)/(\d+)/dc/$', self.wrap(self.delete_course), name="%s_%s_delete" % app_model_name)

        ]
        return urlpatterns
    def delete_course(self, request,customer_id,course_id):
        '''
        刪除當前用戶感興趣的課程
        :param request:
        :param customer_id:
        :param course_id:
        :return:
        '''
        customer_obj = self.model_class.objects.filter(pk=customer_id).first()
        customer_obj.course.remove(course_id)
        return redirect(self.get_list_url())

    list_display = ["qq","name","graduation_school",display_course,display_gender,display_status,display_education,recode]
    edit_link = ["name","graduation_school"]
    search_fields = ["name__contains"]
    show_search_form = True

    show_actions = True
    comb_filter = [
        v1.FilterOption("gender",is_choice=True),
    ]

v1.site.register(models.Customer, CustomerConfig)

class ConsultRecordConfig(v1.StarkConfig):
    def customer_display(self,obj=None,is_header=False):
        if is_header:
            return "所咨詢客戶"
        return obj.customer.name
    list_display = [customer_display,"consultant","date","note"]
    comb_filter = [
        v1.FilterOption("customer"),
    ]   #組合搜索默認不顯示，但是卻又篩選的功能

    def change_views(self, request,*args, **kwargs):
        customer = request.GET.get('customer')
        # session中獲取當前用戶ID
        current_login_user_id = 6
        ct = models.Customer.objects.filter(consultant=current_login_user_id, id=customer).count()
        if not ct:
            return HttpResponse('無權訪問.')
        return super(ConsultRecordConfig,self).change_views(request, *args, **kwargs)
v1.site.register(models.ConsultRecord,ConsultRecordConfig)
v1.site.register(models.Student)