From 1f15f13aac38eebeb632c15d1bffd86ae37f0ce7 Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Wed, 24 Dec 2014 23:17:51 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9bUg?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 connect.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/connect.py b/connect.py
index 4e2f680ed..b3bc119c8 100644
--- a/connect.py
+++ b/connect.py
@@ -155,8 +155,8 @@ def get_user_host(username):
     hosts = {}
     try:
         user = User.objects.get(username=username)
-    except AttributeError, ObjectDoesNotExist:
-        red_print("Don't Use Root To Do That or User isn't Exist.")
+    except ObjectDoesNotExist:
+        return {'Error': 'username %s is not exist.' % username}
     else:
         perm_all = user.permission_set.all()
         for perm in perm_all: