from flask import Flask, flash, render_template, session, request, redirect, url_for, jsonify
from admin.userinfo import *
from admin.roleinfo import *
from flask import current_app as app

#-----------------系统用户登录------------------------------#
# 系统用户登录

@app.route('/adminlogin')
def adminlogin ():
    if 'username' in session:
        return render_template('admin/a_admindefault.html', username=session['username'])
    return render_template('admin/a_adminlogin.html')


# 系统用户登录处理
@app.route('/adminlogin.do', methods=['POST'])
def adminlogin_do ():
    username = request.form['txt_username']
    password = request.form['txt_password']
    # print(username, password)
    results = admin_login(username, password)
    if len(results) > 0:
        # print(results)  #(('ccc', '船长'),)
        session["username"] = results[0][0]
        session["truename"] = results[0][1]
        session["rolename"] = results[0][2]
        return redirect(url_for('admindefault'))
    else:
        flash('登录失败, 请重新登录')
        return redirect(url_for('adminlogin'))


@app.route('/admindefault')
def admindefault ():
    if 'username' in session and 'truename' in session and 'rolename' in session:
        return render_template('admin/a_admindefault.html', lbl_truename=session['truename'],
                               lbl_rolename=session['rolename'])
    else:
        flash('登录失败, 请重新登录')
        return redirect(url_for('adminlogin'))


# --------------------用户管理------------------------------#
@app.route('/usermanage')
def usermanage ():
    if 'username' in session:
        results = loadalluser()
        return render_template('admin/a_user_manage.html', results=results)
    else:
        flash('登录失败, 请重新登录')
        return redirect(url_for('adminlogin'))


#打开用户信息添加页面
@app.route('/newuserinfo')
def newuserinfo():
    if 'username' in session:
        return render_template('admin/a_user_edit.html', status="add")
    else:
        flash('登录失败, 请重新登录')
        return redirect(url_for('adminlogin'))


@app.route('/adduser', methods=['POST'])
def adduser ():
    if request.method == 'POST':
        rolename = request.form.get('txt_role')
        # print("用户身份", roleid)
        username = request.form['txt_username']
        truename = request.form['txt_truename']
        password = request.form['txt_password']
        telephone = request.form['txt_telephone']
        user = [username, truename, password, rolename, telephone]
        # print("用户身份", user)
        i, msg = add_user(user)
        if i > 0:
            flash('用户添加成功!')
        else:
            flash('用户添加失败!%s' % str(msg))
    return redirect(url_for('usermanage'))


@app.route('/deluser/<userid>', methods=['POST'])
def deluser (userid):
    # print("删除用户:", userid)
    i, msg = del_user(userid)
    if i > 0:
        flash('用户删除成功!')
    else:
        flash('用户删除失败!%s' % msg)
    return redirect(url_for('usermanage'))


@app.route('/loadusertoModal/<userid>', methods=['POST', 'GET'])
def loadUserforView (userid):
    result = getUserByID(userid)
    if len(result) > 0:
        print("加载成功")
        print(jsonify(result))
        return jsonify(result)
    else:
        return render_template('admin/a_user_manage.html')


# 修改用户信息
@app.route('/edituser.do', methods=['POST'])
def edituser ():
    if request.method == 'POST':
        userid = request.form['lbl_userid']
        rolename = request.form.get('txt_Role3')
        username = request.form['txt_UserName3']
        truename = request.form['txt_TrueName3']
        telephone = request.form['txt_Telephone3']
        ustatus = request.form.get('txt_UStatus3')
        data = [username, truename, rolename, telephone, ustatus]
        i, msg = update_userinfo(data, userid)
        if i > 0:
            flash('用户信息修改成功!')
        else:
            flash('用户信息修改失败!%s' % str(msg))
    return redirect(url_for('usermanage'))


# 修改用户信息
@app.route('/changepassword.do', methods=['POST'])
def changepassword ():
    if request.method == 'POST':
        userid = request.form['lbl_userid4']
        oldpassword = request.form['txt_oldpassword']
        newpassword1 = request.form['txt_newpassword1']
        newpassword2 = request.form['txt_newpassword2']
        if (newpassword1 != newpassword2):
            flash('两次输入的新密码不一致,请重新输入!')
        else:
            checkresult = checkoldpassword(userid, oldpassword)
            if len(checkresult) > 0:  # 旧密码正确
                i, msg = change_Password(userid, newpassword1)
                if i > 0:  # 密码修改成功
                    flash('用户密码修改成功!')
                else:
                    flash('用户密码修改失败!%s' % str(msg))
            else:
                flash('旧密码错误,请重新输入!')
    return redirect(url_for('usermanage'))