Merge pull request #335 from FirmlyReality/login_info

Login info

src/master/httprest.py

@@ -620,7 +620,7 @@ def user_quotainfo_monitor(user, beans, form, issue):
         return json.dumps({'success':'true', 'createdvnodes':res})
     elif issue == 'net_stats':
         logger.info("handle request: monitor/user/net_stats/")
-        res = G_historymgr.get_user_net_stats(user)
+        res = monitor.Container_Fetcher.get_user_net_stats(user)
         return json.dumps({'success':'true', 'net_stats':res})
     else:
         return json.dumps({'success':'false', 'message':"Unspported Method!"})

src/master/monitor.py

@@ -152,6 +152,18 @@ class Container_Fetcher:
             res = {}
         return res
 
+    # get users' net_stats
+    @staticmethod
+    def get_user_net_stats(owner):
+        global monitor_vnodes
+        try:
+            res = monitor_vnodes[owner]['net_stats']
+        except Exception as err:
+            logger.warning(traceback.format_exc())
+            logger.warning(err)
+            res = {}
+        return res
+
     def get_basic_info(self):
         global monitor_vnodes
         try:

src/master/userManager.py

@@ -7,7 +7,7 @@ Warning: in some early versions, "token" stand for the instance of class model.U
 Original author: Liu Peidong
 '''
 
-from utils.model import db, User, UserGroup, Notification, UserUsage
+from utils.model import db, User, UserGroup, Notification, UserUsage, LoginMsg
 from functools import wraps
 import os, subprocess, math
 import hashlib
@@ -144,6 +144,7 @@ class userManager:
         '''
         try:
             User.query.all()
+            LoginMsg.query.all()
         except:
             db.create_all()
             if password == None:
@@ -254,7 +255,7 @@ class userManager:
         }
         return result
 
-    def auth_external(self, form):
+    def auth_external(self, form, userip=""):
 
         if (env.getenv('EXTERNAL_LOGIN') != 'True'):
             failed_result = {'success': 'false', 'reason' : 'external auth disabled'}
@@ -267,6 +268,10 @@ class userManager:
             return failed_result
 
         username = result['username']
+        logger.info("External login success: username=%s, userip=%s" % (username, userip))
+        loginmsg = LoginMsg(username,userip)
+        db.session.add(loginmsg)
+        db.session.commit()
         user = User.query.filter_by(username = username).first()
         if (user != None and user.auth_method == result['auth_method']):
             result = {
@@ -316,18 +321,27 @@ class userManager:
         }
         return result
 
-    def auth(self, username, password):
+    def auth(self, username, password, userip=""):
         '''
         authenticate a user by username & password
         return a token as well as some user information
         '''
         user = User.query.filter_by(username = username).first()
+        result = {}
         if (user == None or user.auth_method =='pam'):
-            return self.auth_pam(username, password)
+            result = self.auth_pam(username, password)
         elif (user.auth_method == 'local'):
-            return self.auth_local(username, password)
+            result = self.auth_local(username, password)
         else:
             result  = {'success':'false', 'reason':'auth_method error'}
+
+        if result['success'] == 'true':
+            loginmsg = LoginMsg(result['data']['username'],userip)
+            db.session.add(loginmsg)
+            db.session.commit()
+            logger.info("Login success: username=%s, userip=%s" % (result['data']['username'], userip))
+        else:
+            logger.info("Login failed: userip=%s" % (userip))
         return result
 
     def auth_token(self, token):

src/utils/model.py

@@ -43,7 +43,8 @@ app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///'+fsdir+'/global/sys/UserTabl
 app.config['SQLALCHEMY_BINDS'] = {
     'history': 'sqlite:///'+fsdir+'/global/sys/HistoryTable.db',
     'beansapplication': 'sqlite:///'+fsdir+'/global/sys/BeansApplication.db',
-    'system': 'sqlite:///'+fsdir+'/global/sys/System.db'
+    'system': 'sqlite:///'+fsdir+'/global/sys/System.db',
+    'login': 'sqlite:///'+fsdir+'/global/sys/Login.db'
     }
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
 try:
@@ -205,6 +206,21 @@ class UserNotificationPair(db.Model):
     def __repr__(self):
         return '<UserName: %r, NotifyId: %r>' % (self.userName, self.notifyId)
 
+class LoginMsg(db.Model):
+    __bind_key__ = 'login'
+    id = db.Column(db.Integer, primary_key=True)
+    username = db.Column(db.String(10))
+    userip = db.Column(db.String(20))
+    time = db.Column(db.DateTime)
+
+    def __init__(self, username, userip):
+        self.username = username
+        self.userip = userip
+        self.time = datetime.now()
+
+    def __repr__(self):
+        return '<id=%d, username=%s, userip=%s, time=%s>' % (self.id,self.username,self.userip,self.time.strftime("%Y-%m-%d %H:%M:%S"))
+
 class VNode(db.Model):
     __bind_key__ = 'history'
     name = db.Column(db.String(100), primary_key=True)

src/worker/monitor.py

@@ -668,14 +668,3 @@ class History_Manager:
             tmp = {"name":vnode.name,"billing":vnode.billing}
             res.append(tmp)
         return res
-
-    # get users' net_stats
-    def get_user_net_stats(self,owner):
-        global monitor_vnodes
-        try:
-            res = monitor_vnodes[owner]['net_stats']
-        except Exception as err:
-            logger.warning(traceback.format_exc())
-            logger.warning(err)
-            res = {}
-        return res

user/user.py

@@ -87,9 +87,10 @@ def login():
     logger.info("handle request : user login")
     user = request.form.get("user", None)
     key = request.form.get("key", None)
+    userip = request.form.get("ip", "")
     if user == None or key == None:
         return json.dumps({'success': 'false', 'message':'user or key is null'})
-    auth_result = G_usermgr.auth(user,key)
+    auth_result = G_usermgr.auth(user,key,userip)
     if auth_result['success'] == 'false':
         logger.info("%s login failed" % user)
         return json.dumps({'success':'false', 'message':'auth failed'})
@@ -100,8 +101,9 @@ def login():
 def external_login():
     global G_usermgr
     logger.info("handle request : external user login")
+    userip = request.form.get("ip", "")
     try:
-        result = G_usermgr.auth_external(request.form)
+        result = G_usermgr.auth_external(request.form,userip)
         return json.dumps(result)
     except:
         result = {'success':'false', 'reason':'Something wrong happened when auth an external account'}

web/webViews/authenticate/login.py

@@ -52,7 +52,7 @@ class loginView(normalView):
     @classmethod
     def post(self):
         if (request.form['username']):
-            data = {"user": request.form['username'], "key": request.form['password']}
+            data = {"user": request.form['username'], "key": request.form['password'], 'ip': request.remote_addr}
             result = dockletRequest.unauthorizedpost('/login/', data)
             ok = result and result.get('success', None)
             if (ok and (ok == "true")):