# coding = utf-8
"""
@author:
@time:2019/11/7 11:22
@File: app.py
"""
from flask import Flask, render_template, request, abort, session, redirect, url_for, flash
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, EqualTo, ValidationError
from pyecharts import options as opts
from pyecharts.charts import Kline, Line, Bar, Grid
import tushare as ts
import pandas as pd
from flask_bootstrap import Bootstrap
from flask_sqlalchemy import SQLAlchemy
from werkzeug.security import generate_password_hash, check_password_hash
from flask_login import UserMixin, login_user, LoginManager, current_user, logout_user, login_required
import time
from flask_migrate import Migrate


app = Flask(__name__)
app.secret_key = 'A Hard String'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + 'myweb.sqlite'
app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True
bootstrap = Bootstrap(app)
db = SQLAlchemy(app)
login_manager = LoginManager(app)
login_manager.session_protection = 'strong'
migrate = Migrate(app, db, render_as_batch=True)

#Flask-Login通过装饰器@login_required来检查访问视图函数的用户是否已登录
@login_manager.user_loader
def load_user(user_id):
    return WebUser.query.get(int(user_id))


# 用户表结构
class WebUser(UserMixin, db.Model):
    __tablename__ = 'webuser'
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.String(64), unique=True, index=True)
    email = db.Column(db.String(64), unique=True, index=True)
    username = db.Column(db.String(64), unique=True, index=True)
    password_hash = db.Column(db.String(128))
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'), default=1)

    @staticmethod
    def insert_user():     #插入用户
        users = {
            'user1': ['user1@luobo.com', 'test1', 1],
            'user2': ['user2@luobo.com', 'test2', 1],
            'admin1': ['admin1@luobo.com', 'admin1', 2],
            'admin2': ['admin2@luobo.com', 'admin2', 2]
        }
        for u in users:
            user = WebUser.query.filter_by(username=u[0]).first()  #先查询再插入
            if user is None:
                user = WebUser(user_id=time.time(), username=u, email=users[u][0])
                user.password = users[u][1]
                db.session.add(user)
            db.session.commit()

    @property  #@property装饰器会将方法转换为相同名称的只读属性,可以与所定义的属性配合使用，这样可以防止属性被修改。
    def password(self):
        raise AttributeError('You can not read the password')

    @password.setter    #修改用户的password 字段   (修改密码)
    def password(self, password):
        self.password_hash = generate_password_hash(password)

    def verify_password(self, password):   #验证密码
        if self.password_hash is not None:
            return check_password_hash(self.password_hash, password)

    def is_admin(self):  #判断是否为管理员用户
        if self.role_id is 2:
            return True
        else:
            return False




class Role(db.Model):
    __tablename__ = 'roles'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True)
    users = db.relationship('WebUser', backref='role')

    @staticmethod
    def init_roles():
        roles = ['User', 'Admin']
        for r in roles:
            role = Role.query.filter_by(name=r).first()
            if role is None:
                role = Role(name=r)
                db.session.add(role)
        db.session.commit()

#####表单
class LoginForm(FlaskForm):
    email = StringField('email', validators=[DataRequired()])
    password = PasswordField('password', validators=[DataRequired()])
    submit = SubmitField('Submit')


class RegisterForm(FlaskForm):
    email = StringField('email', validators=[DataRequired()])
    password = PasswordField('password', validators=[DataRequired(),
                                                     EqualTo('confirm_pw', message='两次输入的密码需要一致!')])
    confirm_pw = PasswordField('confirm password', validators=[DataRequired()])
    submit = SubmitField('Submit')

    def validate_email(self, field):  #验证邮箱是否存在
        if WebUser.query.filter_by(email=field.data).first():
            raise ValidationError('该邮箱已经存在!')

#注册的视图函数
@app.route('/register/', methods=['GET', 'POST'])
def register():
    form = RegisterForm()
    if form.validate_on_submit():
        email = form.email.data
        password = form.password.data
        user = WebUser.query.filter_by(email=email).first()
        if user is None:
            newuser = WebUser(email=email, username=email, password=password, user_id=time.time())
            db.session.add(newuser)
            flash("你可以登陆啦！")
            return redirect(url_for('login'))
        flash("邮箱已经存在！")
    return render_template('register.html', form=form)






#登录验证
@app.route('/login/', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        user = WebUser.query.filter_by(email=form.email.data).first()
        if user is not None and user.verify_password(form.password.data):
            login_user(user)
            flash('欢迎回来！')
            return redirect(request.args.get('next') or url_for('index'))
        flash('用户名或密码不正确！')
    return render_template('login.html', form=form)


@app.route('/logout/')
@login_required
def logout():
    logout_user()
    return redirect(url_for('index'))


@app.errorhandler(404)
def page_not_found(e):
    return render_template("404.html"), 404


def get_stock_data(code, ctime):
    df = ts.get_hist_data(code)
    df_time = df[:ctime]
    mydate = df_time.index.tolist()
    kdata = df_time[['open', 'close', 'low', 'high']].values.tolist()
    madata_5 = df_time['ma5'].values.tolist()
    madata_10 = df_time['ma10'].values.tolist()
    madata_20 = df_time['ma20'].values.tolist()
    volume_rise = [df_time.volume[x] if df_time.close[x] > df_time.open[x] else "0" for x in range(0, len(df_time.index))]
    volume_drop = [df_time.volume[x] if df_time.close[x] <= df_time.open[x] else "0" for x in range(0, len(df_time.index))]
    return [mydate, kdata, madata_5, madata_10, madata_20, volume_rise, volume_drop]


# 移动平均线
def moving_average_chart(mydate, data_5, data_10, data_20, name) -> Line:
    moving_average = (
        Line()
        .add_xaxis(mydate)
        .add_yaxis("ma5", data_5, is_smooth=True)
        .add_yaxis("ma10", data_10, is_smooth=True)
        .add_yaxis("ma20", data_20, is_smooth=True)
        .set_global_opts(title_opts=opts.TitleOpts(title="%s-移动平均线" % name),
                         datazoom_opts=[opts.DataZoomOpts()],
                         )
        .set_series_opts(
            label_opts=opts.LabelOpts(is_show=False),
        )
    )
    return moving_average


# 成交量
def volume_chart(mydate, volume_rise, volume_drop, name) -> Bar:
    bar = (
        Bar()
        .add_xaxis(mydate)
        .add_yaxis("volume_rise", volume_rise, stack=True, color=["#ec0000"])
        .add_yaxis("volume_drop", volume_drop, stack=True, color=["#00da3c"])
        .set_global_opts(title_opts=opts.TitleOpts(title="%s-成交量" % name),
                        datazoom_opts=[opts.DataZoomOpts()],)
        .set_series_opts(
            label_opts=opts.LabelOpts(is_show=False),
        )
    )
    return bar


# K 线图
def kline_base(mydate, data, name) -> Kline:
    kline = (
        Kline()
        .add_xaxis(mydate)
        .add_yaxis("%s" % name, data, markpoint_opts=opts.MarkLineOpts(
            data=[opts.MarkLineItem(type_="max", value_dim="close")]
        ), markline_opts=opts.MarkLineOpts(
            data=[opts.MarkLineItem(type_="max", value_dim="close")]
        ),
                   itemstyle_opts=opts.ItemStyleOpts(
                       color="#ec0000",
                       color0="#00da3c",
                       border_color="#8A0000",
                       border_color0="#008F28",
                   ),
                   )
        .set_global_opts(
            yaxis_opts=opts.AxisOpts(is_scale=True,
                                    splitarea_opts=opts.SplitAreaOpts(
                    is_show=True, areastyle_opts=opts.AreaStyleOpts(opacity=1)
                ),
            ),
            xaxis_opts=opts.AxisOpts(is_scale=True,
                                    axislabel_opts=opts.LabelOpts(rotate=-30)),
            title_opts=opts.TitleOpts(title="股票走势"),
            datazoom_opts=[opts.DataZoomOpts()],
            toolbox_opts=opts.ToolboxOpts(is_show=True),
        )
    )
    return kline


# full chart
def full_chart(mydate, kdata, data_5, data_10, data_20, volume_rise, volume_drop, name):
    kline = (
        Kline()
            .add_xaxis(mydate)
            .add_yaxis("%s" % name, kdata, markpoint_opts=opts.MarkLineOpts(
            data=[opts.MarkLineItem(type_="max", value_dim="close")]
        ), markline_opts=opts.MarkLineOpts(
            data=[opts.MarkLineItem(type_="max", value_dim="close")]
        ),
                       itemstyle_opts=opts.ItemStyleOpts(
                           color="#ec0000",
                           color0="#00da3c",
                           border_color="#8A0000",
                           border_color0="#008F28",
                       ),
                       )
            .set_global_opts(
            yaxis_opts=opts.AxisOpts(is_scale=True,
                                     splitarea_opts=opts.SplitAreaOpts(
                                         is_show=True, areastyle_opts=opts.AreaStyleOpts(opacity=1)
                                     ),
                                     ),
            xaxis_opts=opts.AxisOpts(is_scale=True,
                                     axislabel_opts=opts.LabelOpts(rotate=-30)),
            title_opts=opts.TitleOpts(title="股票走势"),
            datazoom_opts=[opts.DataZoomOpts(xaxis_index=[0, 1])],
            toolbox_opts=opts.ToolboxOpts(is_show=True),
            legend_opts=opts.LegendOpts(pos_left="20%")
        )
    )
    line = (
        Line()
            .add_xaxis(mydate)
            .add_yaxis("Ma5", data_5, is_smooth=True)
            .add_yaxis("Ma10", data_10, is_smooth=True)
            .add_yaxis("Ma20", data_20, is_smooth=True)
            .set_global_opts(title_opts=opts.TitleOpts(title="移动平均线"))
            .set_series_opts(
            label_opts=opts.LabelOpts(is_show=False),
        )
    )
    bar = (
        Bar()
            .add_xaxis(mydate)
            .add_yaxis("volume_rise", volume_rise, stack=True, color=["#ec0000"], )
            .add_yaxis("volume_drop", volume_drop, stack=True, color=["#00da3c"], )
            .set_global_opts(title_opts=opts.TitleOpts(),
                             legend_opts=opts.LegendOpts(pos_right="20%"))
            .set_series_opts(
            label_opts=opts.LabelOpts(is_show=False),
        )
    )

    overlap_kline_line = kline.overlap(line)
    grid = Grid()
    grid.add(
        overlap_kline_line,
        grid_opts=opts.GridOpts(pos_left="10%", pos_right="8%", height="50%"),
    )
    grid.add(
        bar,
        grid_opts=opts.GridOpts(
            pos_left="10%", pos_right="8%", pos_top="70%", height="16%"
        ),
    )
    return grid


@app.route("/")
def index():
    return render_template("index.html")


def check_stock(code):
    n = 0
    l = []
    stock_code = pd.read_csv("stock_code_name.csv", dtype=object)
    stock_code.drop('Unnamed: 0', axis=1, inplace=True)
    stock_list = stock_code.values.tolist()
    for i in stock_list:
        if code in i:
            n += 1
            l = i
        else:
            continue
    return n, l



#######全图
@app.route('/fullchart/', methods=['GET', 'POST'])
@login_required
def fullchart():
    if current_user.is_admin():
        return render_template('fullchart.html')
    flash('You have not permission to access this page')
    return redirect(url_for('index'))


@app.route("/Kline", methods=['GET', 'POST'])
def get_kline_chart():
    stock_name = request.form.get('stockName')
    query_time = request.form.get('queryTime')
    if not stock_name:
        stock_name = '平安银行'
    if not query_time:
        query_time = 30
    if int(query_time) > 30:
        if current_user.is_authenticated:
            pass
        else:
            abort(403)

    status, stock_code = check_stock(stock_name)
    if status == 0:
        return 'error stock code or name'
    mydate, kdata, madata = get_stock_data(stock_code[0], int(query_time))
    c = kline_base(mydate, kdata, stock_code[1])
    return c.dump_options()


@app.route("/mavg", methods=['GET', 'POST'])
def moving_average():
    return render_template("mavg.html")


@app.route("/volume", methods=['GET', 'POST'])
def volume():
    return render_template("volume.html")


#移动平均线
@app.route("/Line", methods=['GET', 'POST'])
def get_moving_average():
    stock_name = request.form.get('stockName')
    query_time = request.form.get('queryTime')
    if not stock_name:
        stock_name = '平安银行'
    if not query_time:
        query_time = 30
    if int(query_time) > 30:
        if current_user.is_authenticated:
            pass
        else:
            abort(403)

    status, stock_code = check_stock(stock_name)
    if status == 0:
        return 'error stock code or name'
    mydate, kdata, madata_5, madata_10, madata_20, volume_rise, volume_drop = get_stock_data(stock_code[0], int(query_time))
    c = moving_average_chart(mydate, madata_5, madata_10, madata_20, stock_code[1])
    return c.dump_options()

##成交量吧
@app.route("/Bar", methods=['GET', 'POST'])
def get_volume():
    stock_name = request.form.get('stockName')
    query_time = request.form.get('queryTime')
    if not stock_name:
        stock_name = '平安银行'
    if not query_time:
        query_time = 30
    if int(query_time) > 30:
        if current_user.is_authenticated:
            pass
        else:
            abort(403)
    status, stock_code = check_stock(stock_name)
    if status == 0:
        return 'error stock code or name'
    mydate, kdata, madata_5, madata_10, madata_20, volume_rise, volume_drop = get_stock_data(stock_code[0], int(query_time))
    c = volume_chart(mydate, volume_rise, volume_drop, stock_code[1])
    return c.dump_options()


##全图 移动平均线和成交量图
@app.route("/FullChart", methods=['GET', 'POST'])
def get_fullcharte():
    stock_name = request.form.get('stockName')
    query_time = request.form.get('queryTime')
    if not stock_name:
        stock_name = '平安银行'
    if not query_time:
        query_time = 30
    if int(query_time) > 30:
        if current_user.is_authenticated:
            pass
        else:
            abort(403)

    status, stock_code = check_stock(stock_name)
    if status == 0:
        return 'error stock code or name'
    mydate, kdata, madata_5, madata_10, madata_20, volume_rise, volume_drop = get_stock_data(stock_code[0], int(query_time))
    c = full_chart(mydate, kdata, madata_5, madata_10, madata_20, volume_rise, volume_drop, stock_code[1])
    return c.dump_options()


if __name__ == '__main__':
    app.run(debug=True)