Merge branch 'master' of https://github.com/unias/docklet into batch

This commit is contained in:
zhuyj17 2018-09-23 23:17:37 +08:00
commit 371ec5e535
9 changed files with 68 additions and 27 deletions

View File

@ -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!"})

View File

@ -2,6 +2,7 @@ import threading, time, traceback
from utils import env
from utils.log import logger
from httplib2 import Http
from urllib.parse import urlencode
# major dict to store the monitoring data
# only use on Master
@ -152,6 +153,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:

View File

@ -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):

View File

@ -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:
@ -97,7 +98,7 @@ class User(db.Model):
self.department = department
self.truename = truename
self.tel = tel
self.beans = 1000
self.beans = 150
if (date != None):
self.register_date = date
else:
@ -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)

View File

@ -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

View File

@ -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'}
@ -517,6 +519,9 @@ def billing_beans():
def beans_apply(cur_user,user,form,issue):
global G_applicationmgr
if issue == 'apply':
if not cur_user.status == 'normal':
return json.dumps({'success':'false', 'message':'Fail to apply for beans because your account is locked/not activated. Please:'+
'\n 1. Complete your information and activate your account. \n Or: \n 2.Contact administor for further information'})
number = form.get("number",None)
reason = form.get("reason",None)
if number is None or reason is None:

View File

@ -50,7 +50,7 @@
<form action="/beans/apply/" method="POST" id="beansapplyForm">
<div class="form-group">
<label>Number</label><small class="font-bold"> How many beans do you need?</small>
<input type="number" class="form-control" placeholder="100-5000" name="number" id="number" min="100" max="5000"/>
<input type="number" class="form-control" placeholder="100-5000" name="number" id="number" min="100" max="5000" required />
</div>
<div class="form-group">
<label>Reason</label>
@ -104,11 +104,15 @@
<script src="http://cdn.bootcss.com/datatables/1.10.11/js/jquery.dataTables.js"></script>
<script src="http://cdn.bootcss.com/datatables/1.10.11/js/dataTables.bootstrap.js"></script>
<script src="http://cdn.bootcss.com/datatables-tabletools/2.1.5/js/TableTools.min.js"></script>
<script src="https://cdn.bootcss.com/jquery-validate/1.17.0/jquery.validate.js"></script>
<script>
$(document).ready(function() {
$(".table-image").DataTable();
});
$().ready(function() {
$("#beansapplyForm").validate();
});
function textCounter(field,countfield,maxlimit)
{
if(field.value.length > maxlimit)
@ -116,7 +120,7 @@ function textCounter(field,countfield,maxlimit)
else
countfield.value = maxlimit - field.value.length;
}
</script>
{% endblock %}

View File

@ -144,7 +144,7 @@ def state_batch_job():
return stateBatchJobView().as_view()
@app.route("/workspace/create/", methods=['GET'])
@activated_required
#@activated_required
def addCluster():
return addClusterView.as_view()

View File

@ -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")):