From 0f18e702e80ab77d8c4daea2fe94dc5279944668 Mon Sep 17 00:00:00 2001
From: guanghongwei <hongwei.guang@yolu-inc.com>
Date: Tue, 16 Sep 2014 18:23:56 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=87=EE=86=BD=E6=95=BC=E6=B7=87=E6=BF=86?=
 =?UTF-8?q?=E7=93=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 webroot/AutoSa/AutoSa/views.py      | 11 ++++++++---
 webroot/AutoSa/UserManage/models.py |  4 ++--
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/webroot/AutoSa/AutoSa/views.py b/webroot/AutoSa/AutoSa/views.py
index 1a2100ae9..6cc228558 100644
--- a/webroot/AutoSa/AutoSa/views.py
+++ b/webroot/AutoSa/AutoSa/views.py
@@ -4,14 +4,13 @@ from django.http import HttpResponse
 from django.template import RequestContext
 from django.shortcuts import render_to_response
 from django.http import HttpResponseRedirect
-from UserManage.models import User
+from UserManage.models import User, Group
 from Assets.models import Assets, AssetsUser
 import subprocess
 from Crypto.Cipher import AES
 from binascii import b2a_hex, a2b_hex
 import random
 import ConfigParser
-import pam
 from UserManage.forms import UserAddForm, GroupAddForm
 
 
@@ -176,16 +175,22 @@ def addUser(request):
         if form.is_valid():
             user = form.cleaned_data
             ldap_password = keygen(16)
+            group_post = user['group']
+            groups = []
+            for group_name in group_post:
+                groups.append(Group.objects.get(name=group_name))
+
             u = User(
                 username=user['username'],
                 password=user['password'],
                 key_pass=user['key_pass'],
                 name=user['name'],
-                group=user['group'],
                 is_admin=user['is_admin'],
                 is_superuser=user['is_superuser'],
                 ldap_password=ldap_password)
             u.save()
+            u.group = groups
+            u.save()
 
         return render_to_response('addUser.html', {'msg': msg, 'user_menu': 'active'},
                                   context_instance=RequestContext(request))
diff --git a/webroot/AutoSa/UserManage/models.py b/webroot/AutoSa/UserManage/models.py
index eca612de2..d1bdd402b 100644
--- a/webroot/AutoSa/UserManage/models.py
+++ b/webroot/AutoSa/UserManage/models.py
@@ -2,14 +2,14 @@ from django.db import models
 
 
 class Group(models.Model):
-    name = models.CharField(max_length=50)
+    name = models.CharField(max_length=50, unique=True)
 
     def __unicode__(self):
         return self.name
 
 
 class User(models.Model):
-    username = models.CharField(max_length=50)
+    username = models.CharField(max_length=50, unique=True)
     password = models.CharField(max_length=100)
     key_pass = models.CharField(max_length=100)
     name = models.CharField(max_length=50)