from flask import render_template, redirect, url_for, request, flash, session
from front_back import front
from modles.dbmodels import User, db
import hashlib
from utils import login_check


@front.route('/login',methods=['GET','POST'])
def login():
    if request.method=='GET':
        return render_template('login.html')
    elif request.method=='POST':
        username=request.form.get('username')
        password=request.form.get('password')
        user=User.query.filter_by(username=username,password=password).first()
        print(user)
        if user:
            session['user']=username
            print(session['user'])
            flash('登錄成功')
            return redirect(url_for('front.index'))
        else:
            flash('登錄失敗')
            return redirect(url_for('front.login'))

@front.route('/register',methods=['GET','POST'])
def register():
        # 如果是get請(qǐng)求的話就返回頁面，post請(qǐng)求的話就接收表單數(shù)據(jù)
        if request.method == 'GET':
            return render_template("register.html")
        elif request.method == 'POST':
            username = request.form.get('username')
            password = request.form.get('password')
            check_password = request.form.get('check_password')

            if username and password and password == check_password:
                md5 = hashlib.md5()
                md5.update(password.encode('utf-8'))
                user = User()

                user.username = username
                # 使用hashlib加密密碼再存入數(shù)據(jù)庫,拿到md5.hexdigest()加密后的密碼
                user.password = md5.hexdigest()
                user.password=password
                print(username,password)
            try:
               db.session.add(user)
               db.session.commit()
               flash('注冊(cè)成功，歡迎訪問我的個(gè)人博客！')
               return redirect(url_for('front.register'))
            except Exception:
               flash('注冊(cè)失敗，請(qǐng)檢查密碼后重新注冊(cè)')
               return redirect(url_for('front.register'))
        else:
             flash('注冊(cè)失敗')
             return redirect(url_for('front.register'))

@front.route('/index')
@front.route('/')
# @login_required

def index():
    page=int(request.args.get('page',default=1))   #找到傳入的？page=？ 參數(shù)，沒有的話默認(rèn)就是1，因?yàn)閭骰貋淼氖亲址?，所以需要轉(zhuǎn)化成int
    #pre_page=?  這個(gè)是設(shè)置每一頁最多顯示幾條數(shù)據(jù)
    paginate=Article.query.order_by(Article.create_time.desc()).paginate(page=page,per_page=5)
    article=paginate.items
    groups=ArticleGroup.query.all()  #拿到模型所有的數(shù)據(jù)
    return render_template("index.html",groups=groups,article=article,paginate=paginate)

@front.route('/article_detail/<int:article_id>')
# @login_required
def article_detail(article_id):
    article=Article.query.get(article_id)
    groups=ArticleGroup.query.all()  #拿到模型所有的數(shù)據(jù)
    return render_template("article_detail.html",groups=groups,article=article)

@front.route('/add_article',methods=['GET','POST'])
# @login_required
def add_article():
    if request.method=='GET':
        groups=ArticleGroup.query.all()  #拿到模型所有的數(shù)據(jù)
        return render_template("add_article.html",groups=groups)
    elif request.method=='POST':
        title=request.form.get('title')
        content=request.form.get('content')
        gid=request.form.get('gid')

        user=User.query.filter_by(username=session.get('user')).first()
        uid=user.id
        # 獲取當(dāng)前時(shí)間，從datetime這個(gè)庫里面
        create_time=datetime.datetime.now()

        article=Article()
        article.title=title
        article.content=content
        article.gid=gid
        article.uid=uid
        article.create_time=create_time
        print(user,title,content,gid,uid,create_time)
        try:
            db.session.add(article)
            db.session.commit()
            flash('添加%s成功' % title)
            return redirect(url_for('front.add_article'))
        except Exception:
            flash('添加%s失敗' % title)
            return redirect(url_for('front.add_article'))

@front.route("/add_group",methods=['GET','POST'])
def add_group():
    if request.method=='GET':
        groups = ArticleGroup.query.all()
        colors=['default','primary','success','into','warning','danger']
        return render_template('add_group.html',colors=colors,groups=groups)
    elif request.method=='POST':
        name=request.form.get('name')
        color=request.form.get('color')
        group=ArticleGroup()
        group.name=name
        group.color=color
        try:
            db.session.add(group)
            db.session.commit()
            flash("添加分組成功")
            return redirect(url_for('front.manage'))
        except Exception:
            flash("添加失敗")
            return redirect(url_for('front.manage'))

@front.route("/article_group_manage")

def article_group_manage():
    groups = ArticleGroup.query.all()
    return render_template('article_group_mange.html',groups=groups)
@front.route('/edit_group/<int:gid>',methods=['GET','POST'])
def edit_group(gid):
    if request.method=='GET':
        groups=ArticleGroup.query.all()
        #找到通過id找到分組
        group=ArticleGroup.query.get(gid)
        colors = ['default', 'primary', 'success', 'into', 'warning', 'danger']
        return render_template('edit_group.html',group=group,colors=colors,groups=groups)
    elif request.method=='POST':
        name=request.form.get("name")
        color=request.form.get("color")

        group=ArticleGroup.query.get(gid)
        group.name=name
        group.color=color
        try:
            db.session.add(group)
            db.session.commit()
            flash("修改成功")
            return redirect(url_for("front.article_group_manage"))
        except Exception:
            flash("修改失敗")
            return redirect(url_for("front.article_group_mange"))

# 刪除的 視圖函數(shù)
@front.route('/delete_group/<int:gid>')
def delete_group(gid):
    group=ArticleGroup.query.get(gid)
    try:
        db.session.delete(group)
        db.session.commit()
        flash('刪除%s成功' % group.name)
        return redirect(url_for("front.article_group_manage"))
    except Exception:
        flash('刪除%s失敗' % group.name)
        return redirect(url_for("front.article_group_manage"))

@front.route('/edit_article/<int:article_id>',methods=['GET','POST'])
def edit_article(article_id):
    if request.method=='GET':
        # 拿到文章內(nèi)容
        article=Article.query.get(article_id)
        groups=ArticleGroup.query.all()
        return render_template('edit_article.html',groups=groups,article=article)
    elif request.method=='POST':
        title=request.form.get("title")
        content=request.form.get("content")
        gid=request.form.get("gid")
        update_time=datetime.datetime.now()

        article=Article.query.get(article_id)
        article.title=title
        article.content=content
        print(content)
        article.gid=gid
        article.update_time=update_time

        try:
            db.session.commit()
            return redirect(url_for("front.article_manage"))
        except Exception:
            flash("修改%s失敗" % title)
            return redirect(url_for("front.article_manage"))
@front.route('/delete_article/<int:article_id>')
def delete_article(article_id):
    article=Article.query.get(article_id)
    try:
        db.session.delete(article)
        db.session.commit()
        flash("刪除文章%s成功"%article.title)
        return redirect(url_for("front.article_manage"))
    except Exception:
        flash("刪除文章%s失敗"%article.title)
        return redirect(url_for("front.article_manage"))

# 注銷登錄，在base寫上
@front.route('/logout')
def logout():
    session.clear()
    flash("注銷登錄成功")
    return redirect(url_for('front.login'))

login_manager = LoginManager()

db=SQLAlchemy()    #實(shí)例化sqlalchemy對(duì)象

class User(db.Model,UserMixin):
    # __tablename__='login_register'
    id = Column(db.Integer,autoincrement=True,primary_key=True)
    #autoincrement自增長(zhǎng)
    username=Column(db.String(50),nullable=False)
    password=Column(db.String(128),nullable=True)


    # 沒有這個(gè)的話，使用{{users}}顯示的就是一個(gè)對(duì)象
    def __repr__(self):
        return self.username

    def get_id(self):
        return True

class Article(db.Model):
    __tablename__='article'
    id = db.Column(db.Integer, autoincrement=True, primary_key=True)
    title = db.Column(db.String(50), nullable=False,unique=True)
    content=db.Column(db.String(512),nullable=True)
    uid=db.Column(db.Integer,db.ForeignKey('user.id'))
    gid=db.Column(db.Integer,db.ForeignKey('article_group.id'))
    create_time=db.Column(db.DateTime)
    update_time=db.Column(db.DateTime)
    #下面相當(dāng)于做一個(gè)聯(lián)表查詢。
    users=db.relationship('User',backref=db.backref('articles'))
    groups=db.relationship('ArticleGroup',backref=db.backref('articles'))


    def __repr__(self):
        return '<Article %s>'%self.title

class ArticleGroup(db.Model):
    __tablename__='article_group'
    id = db.Column(db.Integer, autoincrement=True, primary_key=True)
    name = db.Column(db.String(50), nullable=False,unique=True)
    color= db.Column(db.String(50), nullable=False)

    def __repr__(self):
        return '<ArticleGroup %s>'% self.name

@login_manager.user_loader
def get_user(user_id):
    #對(duì)主鍵的查詢是沒有必要使用filler_by的
    user = db.session.query(User).get(user_id)
    return user

def create_app():
    app=Flask(__name__)
    app.config.from_object('config')
    register_blueprint(app)
    db.init_app(app)

    login_manager.init_app(app)
    login_manager.login_view="front.login"
    login_manager.login_message_category='請(qǐng)先登錄或注冊(cè)'

    db.create_all(app=app)
    return app


def register_blueprint(app):
    from front_back import front
    app.register_blueprint(front)

from current import create_app

app=create_app()
if __name__ == '__main__':
    app.run(host='0.0.0.0',debug=True,port=83)