detectweb/route.py

from flask import render_template, redirect, url_for, request, \
    abort, current_app, flash
from flask_login import login_user, current_user, logout_user, login_required
from sqlalchemy import and_

from detectweb.forms import LoginForm, RegisterForm, EditProfileForm, TweetForm, \
    PasswdResetRequestForm, PasswdResetForm
from detectweb.models.user import User, load_user
from detectweb.models.tweet import Tweet
from detectweb.models.predict import Predict
from detectweb import db, utils
from detectweb.email_ import send_email
from detectweb.model import *

# 加上了login_required，页面就能被保护，需要先登录
@login_required
def index():
    # # 获取page参数，或者等于1，即默认就是显示第一页
    # page_num = int(request.args.get('page') or 1)
    # # index的tweet是自己+关注人的
    # tweets = current_user.own_and_followed_tweets().paginate(
    #     page=page_num, per_page=current_app.config['TWEET_PER_PAGE'], error_out=False)
    #
    # # tweets.next_num获取下一页的number，前提是还有下一页
    # next_url = url_for('index', page=tweets.next_num) if tweets.has_next else None
    # prev_url = url_for('index', page=tweets.prev_num) if tweets.has_prev else None
    # return render_template(
    #     'index.html', tweets=tweets.items, next_url=next_url, prev_url=prev_url
    # )
    return render_template('index.html')


def login():
    # session管理
    if current_user.is_authenticated:
        return redirect(url_for('index'))
    form = LoginForm()
    # 表示如果是点击submit触发的该函数
    if form.validate_on_submit():
        u = User.query.filter_by(username=form.username.data).first()
        if u is None or not u.check_password(form.password.data):
            flash('非法用户名或密码', 'danger')
            return redirect(url_for('login'))
        # remember_me用于session管理，而不是记住用户名
        login_user(u, remember=form.remember_me.data)
        next_page = request.args.get('next')
        if next_page: # 是否来自于跳转登录，如果是，那就返回到用户一开始点的那个页面去
            return redirect(next_page)
        return redirect(url_for('index')) # url_for通过函数名来找
    return render_template('login.html', title="Sign In", form=form)

def visitor():
    random_username = "visitor"+''.join(random.choice(string.digits) for _ in range(8))
    ip = request.remote_addr
    addr = utils.getAddrFromIP(ip)
    user = User(
        username=random_username,
        email=utils.RandomEmail(),
        country=addr["ct"],
        city=addr["city"],
        province=addr["prov"]
    )
    user.is_visitor = True
    # login_user(user)
    return render_template('index.html', current_user=user)

def logout():
    logout_user()
    return redirect(url_for('login'))


def register():
    # 如果是已登录状态，则直接返回index页面，不让注册
    if current_user.is_authenticated:
        return redirect(url_for('index'))
    form = RegisterForm()
    if form.validate_on_submit():
        # 获取ip
        ip = request.remote_addr
        print(ip)
        addr = utils.getAddrFromIP(ip)
        print(addr)
        # 创建表
        user = User(
            username=form.username.data,
            email=form.email.data,
            country=addr["ct"],
            city=addr["city"],
            province=addr["prov"]
        )
        user.set_password(form.password.data)
        db.session.add(user)
        db.session.commit()
        return redirect(url_for('login'))
    return render_template('register.html', title='Registration', form=form)

@login_required
def user(username):
    u = User.query.filter_by(username=username).first()
    if u is None: # 用户不存在，返回404
        abort(404)

    if request.method == 'POST':
        if request.form['request_button'] == 'Follow':
            current_user.follow(u)
            db.session.commit()
        elif request.form['request_button'] == "Unfollow":
            current_user.unfollow(u)
            db.session.commit()
        else:
            flash("邮件已发送, 请查看!", 'info')
            send_email_for_user_activate(current_user)
    return render_template('user.html', title='Profile', user=u)


# 生成url并发送邮件
def send_email_for_user_activate(user):

    token = user.get_jwt()
    url_user_activate = url_for(
        'user_activate',
        token=token,
        _external=True
    )
    send_email(
        subject=current_app.config['MAIN_SUBJECT_USER_ACTIVATE'],
        recipients=[user.email],
        text_body= render_template(
            'email/user_activate.txt',
            username=user.username,
            url_user_activate=url_user_activate
        ),
        html_body=render_template(
            'email/user_activate.html',
            username=user.username,
            url_user_activate=url_user_activate
        )
    )

# 点击链接后进入该函数激活
def user_activate(token):
    # # 这一句为作者误写，如果加了永远都激活不了
    # if current_user.is_authenticated:
    #     return redirect(url_for('index'))
    user = User.verify_jwt(token)
    if not user: # 如果没找到，其实不是expire了，因为没有判断是否expire，同一个链接永远都可以用，只能说明token是伪造的
        msg = "Token已过期d, 请尝试重新发送文件"
    else:
        user.is_activated = True
        db.session.commit()
        msg = '用户已激活!'
    return render_template(
        'user_activate.html', msg=msg
    )


def page_not_found(e):
    return render_template('404.html'), 404


@login_required
def edit_profile():
    form = EditProfileForm()
    if request.method == 'GET':
        form.about_me.data = current_user.about_me # 显示原来的
    if form.validate_on_submit():
        current_user.about_me = form.about_me.data # 赋值
        db.session.commit() # 提交
        # 跳转
        return redirect(url_for('profile', username=current_user.username))
    return render_template('edit_profile.html', form=form)


def reset_password_request():
    # 如果已处于登录状态，就不需要重置密码了
    if current_user.is_authenticated:
        return redirect(url_for('index'))
    # 初始化这个form
    form = PasswdResetRequestForm()
    if form.validate_on_submit():
        user = User.query.filter_by(email=form.email.data).first()
        if user:
            flash(
                "你将很快收到一封邮件，你能借此重置你的密码。 \
                 如果没有收到，请检查垃圾箱。", 'info'
            )
            # 用户请求的时候，服务器生成token
            token = user.get_jwt()
            # 生成reset密码的url
            url_password_reset = url_for(
                'password_reset',
                token=token,
                _external=True # 表示是完整的http链接：ip 端口 以及后面的url
            )
            url_password_reset_request = url_for(
                'reset_password_request',
                _external=True
            )
            send_email(
                subject=current_app.config['MAIL_SUBJECT_RESET_PASSWORD'], # 主题
                recipients=[user.email], # 接收者
                text_body= render_template(
                    'email/passwd_reset.txt',
                    url_password_reset=url_password_reset, # 传入url
                    url_password_reset_request=url_password_reset_request
                ),
                html_body=render_template(
                    'email/passwd_reset.html',
                    url_password_reset=url_password_reset,
                    url_password_reset_request=url_password_reset_request
                )
            )
            # 重置完了就返回登录页面
        return redirect(url_for('login'))
    return render_template('password_reset_request.html', form=form)


def password_reset(token):
    # 已经登录的情况下，同样不需要reset
    if current_user.is_authenticated:
        return redirect(url_for('index'))
    # 先验证，然后根据token返回一个user，然后对这个user的密码进行重置
    user = User.verify_jwt(token)
    if not user:
        # 重定向到login
        return redirect(url_for('login'))
    # 重置密码的form
    form = PasswdResetForm()
    # 提交后就覆盖之前的密码
    if form.validate_on_submit():
        user.set_password(form.password.data)
        db.session.commit()
        # 重定向login
        return redirect(url_for('login'))
    return render_template(
        'password_reset.html', title='Password Reset', form=form
    )


@login_required
def explore():
    # get all user and sort by followers
    page_num = request.args.get('page') or "1"
    if page_num.isdigit():
        page_num = int(page_num)
    else:
        abort(404)
    tweets = Tweet.query.order_by(Tweet.create_time.desc()).paginate(
        page=page_num, per_page=current_app.config['TWEET_PER_PAGE'], error_out=False)

    next_url = url_for('explore', page=tweets.next_num) if tweets.has_next else None
    prev_url = url_for('explore', page=tweets.prev_num) if tweets.has_prev else None
    return render_template(
        'explore.html', tweets=tweets.items, next_url=next_url, prev_url=prev_url
    )

@login_required
def predict_history():
    u = User.query.filter_by(username=current_user.username).first()

    # 默认第一页
    page_num = request.args.get('page') or "1"
    if page_num.isdigit():
        page_num = int(page_num)
    else:
        abort(404)
    predicts = u.predicts.order_by(Predict.predict_time.desc()).paginate(
        page=page_num,
        per_page=current_app.config['PREDICT_PER_PAGE'],
        error_out=False)

    # 下一页的url
    next_url = url_for('predict_history', page=predicts.next_num, username=current_user.username) if predicts.has_next else None
    prev_url = url_for('predict_history', page=predicts.prev_num, username=current_user.username) if predicts.has_prev else None

    return render_template(
        'predict_history.html',
        title='History',
        allpredicts=predicts.items,
        user=u,
        next_url=next_url,
        prev_url=prev_url,
        totol_num=u.predicts.count() - (page_num-1)*current_app.config['PREDICT_PER_PAGE'],
        en2ch=utils.en2ch
    )

@login_required
def feedback_history():
    u = User.query.filter_by(username=current_user.username).first()

    # 默认第一页
    page_num = request.args.get('page') or "1"
    if page_num.isdigit():
        page_num = int(page_num)
    else:
        abort(404)
    feedbacks = u.tweets.order_by(Tweet.create_time.desc()).paginate(
        page=page_num,
        per_page=current_app.config['TWEET_PER_PAGE'],
        error_out=False)
    allfeedbacks = []
    for per in feedbacks.items:
        allfeedbacks.append("usr_predict_images/{}/{}.jpg".format(current_user.username, per.img))
    # 下一页的url
    next_url = url_for('feedback_history', page=feedbacks.next_num, username=current_user.username) if feedbacks.has_next else None
    prev_url = url_for('feedback_history', page=feedbacks.prev_num, username=current_user.username) if feedbacks.has_prev else None

    return render_template(
        'feedback_history.html',
        title='History',
        allfeedbacks=feedbacks.items,
        user=u,
        next_url=next_url,
        prev_url=prev_url,
        totol_num=u.tweets.count() - (page_num - 1) * current_app.config['TWEET_PER_PAGE']
    )

@login_required
def feedback():
    form = TweetForm()
    # 发文
    if form.validate_on_submit():
        t = Tweet(body=form.tweet.data, author=current_user, img=request.form['imgname'])
        db.session.add(t)
        db.session.commit()
        return redirect(url_for('index'))
    u = User.query.filter_by(username=current_user.username).first()

    value = str(request.args.get('value') or None)

    designated = False
    if value == "None": # 没指定，则返回最新的图片
        # predict图片
        predict_img = u.predicts.order_by(Predict.predict_time.desc()).paginate(
            page=1,
            per_page=current_app.config['TWEET_PER_PAGE'],
            error_out=False)
        predict_img = predict_img.items[0]
        value = predict_img.img_name
    else: # value不为空表示指定了哪张图片
        predict_img = u.predicts.filter_by(img_name=value).first()
        designated = True

    find_ = u.tweets.filter_by(img=value).first()
    feedbacked = False
    if find_ is not None:
        feedbacked = True
    img_path = predict_img.img_path
    predict_img.img_path = "".join([img_path[:-33], '/', img_path[-32:]])
    return render_template('feedback.html',
                           form=form,
                           find_=find_,
                           feedbacked=feedbacked,
                           designated=designated,
                           predict_img=predict_img,
                           en2ch=utils.en2ch)

@login_required
def predict():
    if request.method == 'GET':
        return render_template('predict.html')
    if request.method == 'POST':
        message = request.get_json(force=True)
        encoded = message["image"]
        decoded = base64.b64decode(encoded)
        img = Image.open(io.BytesIO(decoded))
        print(img.mode)
        now = datetime.datetime.now()

        destination = "static/usr_predict_images/{}".format(current_user.username)
        if not os.path.exists(destination):
            os.makedirs(destination)

        randomPost = False
        rand_str = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(8))
        if randomPost:
            ra1, ra2, ra3, ra4 = random.randint(15, 21), random.randint(1, 9), random.randint(1, 9), random.randint(1, 9)
            randomtime_file = '20{}-0{}-1{}-1{}-33-48-'.format(ra1, ra2, ra3, ra4)
            randomtime_db = datetime.datetime.strptime('20{}-0{}-1{} 1{}:33:48'.format(ra1, ra2, ra3, ra4), "%Y-%m-%d %H:%M:%S")
            file_name = os.path.join(destination, randomtime_file + rand_str + '.jpg')
        else:
            file_name = os.path.join(destination, str(now.strftime("%Y-%m-%d-%H-%M-%S-")) + rand_str + '.jpg')


        try:
            if img.mode != "RGB": # jpeg是RGB
                img = img.convert('RGB')
            img.save(os.path.join(file_name), "JPEG", quality=80, optimize=True, progressive=True)
        except IOError:
            ImageFile.MAXBLOCK = img.size[0] * img.size[1]
            img.save(file_name, "JPEG", quality=80, optimize=True, progressive=True)

        t = label_image.read_tensor_from_image_file(file_name,
                                                    input_height=input_height,
                                                    input_width=input_width,
                                                    input_mean=input_mean,
                                                    input_std=input_std)

        input_name = "import/" + input_layer
        output_name = "import/" + output_layer
        input_operation = graph.get_operation_by_name(input_name);
        output_operation = graph.get_operation_by_name(output_name);

        with tf.Session(graph=graph) as sess:
            start = time.time()
            results = sess.run(output_operation.outputs[0],
                               {input_operation.outputs[0]: t})
            end = time.time()

        results = np.squeeze(results)

        top_k = results.argsort()[-5:][::-1]
        labels = label_image.load_labels(label_file)

        print('\nEvaluation time (1-image): {:.3f}s\n'.format(end - start))

        # top_k [35 18 29 28 30]
        for i in top_k:
            if max(results) == results[i]:
                res = results[i]
                lab = labels[i]
                print(labels[i], results[i], "this is max")
            else:
                print(labels[i], results[i], "this is not max")
        # 入库
        if randomPost:
            t = Predict(img_name=file_name[-32:-4], img_path=file_name, size=str(img.size[0]) + "*" + str(img.size[1]),
                        predict_time=randomtime_db, predict_result=lab, predict_value=str(res), author=current_user)
        else:
            t = Predict(img_name=file_name[-32:-4], img_path=file_name, size=str(img.size[0]) + "*" + str(img.size[1]),
                        predict_result=lab, predict_value=str(res), author=current_user)
        ch = utils.en2ch[lab]
        db.session.add(t)
        db.session.commit()
        response = {
            'prediction': {
                'species': ch.split(' ')[0],
                'condition': ' '.join(ch.split(' ')[1:]),
                'value': str(res)
            }
        }
        return jsonify(response)

@login_required
def predictbykind():
    if request.method == 'POST':
        value = request.get_json(force=True)["value"]
        encoded = request.get_json(force=True)["image"]
        decoded = base64.b64decode(encoded)
        image = Image.open(io.BytesIO(decoded))

        # 保存图片
        now = datetime.datetime.now()
        destination = "static/usr_predict_images/{}".format(current_user.username)
        if not os.path.exists(destination):
            os.makedirs(destination)
        rand_str = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(8))
        file_name = os.path.join(destination, str(now.strftime("%Y-%m-%d-%H-%M-%S-")) + rand_str + '.jpg')
        try:
            if image.mode != "RGB": # jpeg是RGB
                image = image.convert('RGB')
            image.save(os.path.join(file_name), "JPEG", quality=80, optimize=True, progressive=True)
        except IOError:
            ImageFile.MAXBLOCK = image.size[0] * image.size[1]
            image.save(file_name, "JPEG", quality=80, optimize=True, progressive=True)

        response = utils.infer_special(value, image)

        # 入库
        t = Predict(img_name=file_name[-32:-4], img_path=file_name, size=str(image.size[0]) + "*" + str(image.size[1]),
                    predict_result=" ".join([value, response['conditionen']]),
                    predict_value=response['prediction']['value'], author=current_user)
        db.session.add(t)
        db.session.commit()
    return jsonify(response)

@login_required
def dashboard():
    time_now = datetime.datetime.now()

    # join实现复杂查询https://www.kancloud.cn/tokimeki/flask_notebook/775173
    # 方法一，查看有病的
    same_city_count = 0
    for u, a in db.session.query(User, Predict).filter(User.id == Predict.user_id).all():
        if u.city == current_user.city and 'healthy' not in a.predict_result:
            same_city_count+=1

    # 方法二，查看有病的，只要包含healthy的label就不算有病
    all_of_same_city = Predict.query.join(User, (User.id == Predict.user_id)).filter_by(city=current_user.city).filter(and_(Predict.predict_result.notlike('%healthy%'), Predict.predict_result.notlike('%healthy')))

    # 以后按这样来，以免算法有问题，如果两个都一致，结果就没问题
    assert same_city_count == all_of_same_city.count()

    timeline = request.args.get('timeline') or "week"
    type_ = request.args.get('type') or "allkinds"

    return utils.getStatistics(timeline, type_, all_of_same_city, time_now, current_user)


def predict_for_visitor():
    if request.method == 'GET':
        try:
            username = request.args.get('username')
        except:
            print("重造")
            username = "visitor" + ''.join(random.choice(string.digits) for _ in range(8))
        ip = request.remote_addr
        addr = utils.getAddrFromIP(ip)
        user = User(
            username=username,
            email=utils.RandomEmail(),
            country=addr["ct"],
            city=addr["city"],
            province=addr["prov"]
        )
        user.is_visitor = True
        return render_template('predict.html', current_user=user)
    if request.method == 'POST':
        value = request.get_json(force=True)["value"]
        encoded = request.get_json(force=True)["image"]
        decoded = base64.b64decode(encoded)
        image = Image.open(io.BytesIO(decoded))
        if image.mode != "RGB":  # jpeg是RGB
            image = image.convert('RGB')
        response = utils.infer_special(value, image)
    return jsonify(response)


def intro():
    value = str(request.args.get('value') or "apple")
    if value in utils.classname:
        return render_template("intros/intro_{}.html".format(value))
    abort(404)


def article():
    value = request.args.get('value') or "1"
    if value.isdigit():
        value = int(value)
    else:
        abort(404)
    if value == 1:
        return render_template("articles/article{}.html".format(value))
    if value == 2:
        return render_template("articles/article{}.html".format(value))
    abort(404)