Django 扩展User 编写自己的认证后台

草帽路飞UU

前言

今天用django写web平台，需要用到账号管理，当然第一时间就想到Django的强大的User模型，各种权限的控制,session的管理都速度解决了。但是Django的管理系统与自己的后台数据库User对象是紧密相连的，而我又不希望用Django User数据库作为我的后台数据库，查了相关资料，发现可以编写自己的认证后台解决。

实现

实现方法就是，编写自己的认证后台，每次登陆的时候在 authenticate 方法中把自己的后台数据中的用户都插入一个相应的Django User对象。这样就可以无缝结合到Django的认证中,享受Django强大的认证后台功能

1.创建自己的后台模块

1. myauth/
   
2. ├── admin.py
   
3. ├── auth.py
   
4. ├── __init__.py
   
5. └── models.py

复制代码

2.models.py

1. # -*- coding: utf-8 -*-
   
2. from django.db import models
   
3. import hashlib
   
4. #自己的后台数据库表.account
   
5. class Account(models.Model):
   
6.     username = models.CharField(u"用户名",blank=True,max_length=32)
   
7.     password = models.CharField(u"密码",blank=True,max_length=50)
   
8.     domain = models.CharField(u"可操作域名",blank=True,max_length=256,help_text='填写多个域名，以,号分隔')
   
9.     is_active = models.IntegerField(u"is_active",blank=True)
   
10.     phone = models.CharField(u"电话",max_length=50)
    
11.     mail = models.CharField(u"邮箱",max_length=50)
    
12. 
    
13.     def __unicode__(self):
    
14.         return self.username
    
15. 
    
16.     def is_authenticated(self):
    
17.         return True
    
18. 
    
19.     def hashed_password(self, password=None):
    
20.         if not password:
    
21.             return self.password
    
22.         else:
    
23.             return hashlib.md5(password).hexdigest()
    
24.     def check_password(self, password):
    
25.         if self.hashed_password(password) == self.password:
    
26.         #if password == self.password:
    
27.             return True
    
28.         return False
    
29.     class Meta:
    
30.         db_table = "account"

复制代码

3.auth.py

一个认证后台其实就是一个实现了如下两个方法的类： get_user(id) 和 authenticate(**credentials)，我也就是在authenticate中动手脚

1. # -*- coding: utf-8 -*-
   
2. from django.contrib.auth.models import User
   
3. from myauth.models import Account
   
4. class MyCustomBackend:
   
5. 
   
6.     def authenticate(self, username=None, password=None):
   
7.         try:
   
8.             user = Account.objects.get(username=username)
   
9.         except Account.DoesNotExist:
   
10.             return None
    
11.         else:
    
12.             if user.check_password(password):
    
13.                 try:
    
14.                     django_user = User.objects.get(username=user.username)
    
15.                 except User.DoesNotExist:
    
16.                     #当在django中找不到此用户，便创建这个用户
    
17.                     django_user = User(username=user.username,password=user.password)
    
18.                     django_user.is_staff = True
    
19.                     django_user.save()
    
20.                 return django_user
    
21.             else:
    
22.                 return None
    
23.     def get_user(self, user_id):
    
24.         try:
    
25.             return User.objects.get(pk=user_id)
    
26.         except User.DoesNotExist:
    
27.             return None
    

复制代码

4.admin.py

把自己的后台数据库表也加到django的管理系统里面

1. from myauth.models import Account
   
2. from django.contrib import admin
   
3. 
   
4. admin.site.register(Account)

复制代码

5.最后在settings.py中添加自己的认证后台，完成

1. AUTHENTICATION_BACKENDS = (
   
2.     'myauth.auth.MyCustomBackend' ,
   
3. )

复制代码

Miss_love

感谢分享