def mainIndex():
user = None
attempted = False
sessionUser = ['', '', '']
deactivated = False
#Log in user
if request.method == 'POST':
attempted = True
email = request.form['email']
pwd = request.form['pwd']
query = pg.logIn(email, pwd)
if query != None and len(query) > 0:
#Check for deactivated status.
if query[3] == 3:
deactivated = True
attempted = False
else:
user = User(query[0], query[1], query[2], query[3])
session['userName'] = user.firstname
session['email'] = user.email
session['role'] = user.role.value
#Session Check
if 'userName' in session:
sessionUser = [session['userName'], session['email'], session['role']]
else:
sessionUser = ['', '']
return render_template('index.html',
sessionUser=sessionUser,
attempted=attempted,
deactivated=deactivated)
def register_interface(name, password):
user = select_user(name)
if user:
return False, '用户已经存在'
else:
from lib.User import User
new_user = User(name, password)
save_user(new_user)
user_logger.info('%s 注册了' % name)
return True, '注册成功'
def thumbs_change():
print('thumbs change')
data = request.form
"""Changes queue when thumbs up/down."""
user = User(session['id'])
user.save_thumbs_change(data['track_id'], data['change'])
queue.thumbs_change(data['track_id'],
data['change'],
decrement=(data['decrement'] == 'true'))
queue_change()
return json.dumps({'success': 'acknowledged the upvote/downvote'})
def UserLogin(get_user, get_pass):
'''用户登录'''
#注册用户类
user = User(get_user)
# 检查用户是否注册
if user.getUserID() == None:
return "账号未注册，请注册"
# 验证用户密码
print(get_user, get_pass)
if user.getUserPasswd() == get_pass:
user.uploadUserinfo()
# 如果登录成功返回user对象
return user()
else:
raise ValueError("密码错误")
def registerUser(self, name):
"""Creates and registers user with the given name.
This is a weird implementation since users cannot exist outside
of this protocol.
But, eh ¯\\\\\_(ツ)\\_/¯
Parameters:
----------
name : String
Returns:
--------
None
"""
user = User(name)
self.__users.append(user)
def getUsers(self, verbose=False):
self._users = []
self._readFiles()
sudoFileObject = StringIO(self.files["sudoers"])
sp = SudoersParser()
sp.parseFile(sudoFileObject)
# TODO: clean up, use one variable
for ul in self.files["passwd"].split("\n"):
username = ""
username = ul.split(":")[0]
userID = ul.split(":")[2]
userGecos = ul.split(":")[4]
userPrimaryGroupID = ul.split(":")[3]
userShell = ul.split(":")[6]
userLocked = self.isUserLocked(username)
if verbose:
logging.warning("Working on user {0}".format(username))
# Get groups
groups = []
for gl in self.files["group"].split("\n"):
groupName = gl.split(":")[0]
groupID = gl.split(":")[2]
groupUsers = gl.split(":")[3]
if userPrimaryGroupID == groupID or groupUsers.find(
username) != -1:
groups.append(groupName)
# Get sudo permissions
userSudoPerms = sp.getCommands(username)
# TODO: put this in a separate function, to get all OS group privileges in sudoers file
for group in groups:
for sudoline in self.files["sudoers"].split("\n"):
if re.search("^%{0} \S*".format(group), sudoline):
userSudoPerms.append(sudoline)
user = User(userID, username, userGecos, userLocked, groups,
userSudoPerms)
self._users.append(user)
return self._users
import pandas as pd
from flask import Flask, request, session, redirect, url_for, escape, render_template, send_file
import xlrd, os, uuid, pdfkit
from openpyxl.workbook import Workbook
from lib.errors import *
from lib.User import User
from lib.tableKeys import keys as tableKeys
app = Flask(__name__,
static_url_path='',
static_folder='templates/',
template_folder='templates/')
app.secret_key = "some"
user = User()
@app.route('/', methods=['GET'])
def home():
if 'user' in session:
return user.index()
else:
return redirect(url_for('login'))
@app.route('/blank', methods=['GET'])
def blank():
try:
return user.blank()
except LinkError:
def connect():
try:
user = User(session['id'])
return json.dumps(user.get_data()['thumbs_tracks'])
except AttributeError:
return json.dumps({'error': 'no data for this user yet'})
key_type_dic = DZMon.find(userid=user.getUserID(), cloum=collo_DZ)
print(key_type_dic)
btnSta = dict()
keylist = [
keyid for keyid in item.keys() if keyid > 10100 and keyid < 10200
]
for keyid in keylist:
try:
if key_type_dic[collo_DZ][str(keyid)] > 1:
btnSta.update({
keyid:
(True,
key_type_dic[collo_DZ][str(keyid)] + 10 - int(time()))
})
else:
btnSta.update({keyid: False})
except KeyError:
btnSta.update({keyid: False})
except TypeError:
btnSta.update({keyid: False})
print(btnSta, ores)
return ores, btnSta
if __name__ == "__main__":
user = User(userid=2)
print(duanZao(user, key_type=10101))
# 查询 指定钥匙的是锻造情况
