mastodon-BDSM/BDSM/views.py

#!/usr/bin/env python3
import os

from flask import render_template, request, url_for, redirect, flash
from BDSM import app, db
from BDSM.models import Media, Settings, Toot, Emoji
from BDSM.toot import app_register, archive_toot
from mastodon import Mastodon
from types import SimpleNamespace

@app.route('/', methods=['GET', 'POST'])
def index():
    page = request.args.get('page', 1, type=int)
    toots_ = Toot.query.order_by(Toot.created_at.desc()).paginate(page, per_page=50)
    toots = []

    for toot_ in toots_.items:
        toot = SimpleNamespace(**toot_.__dict__)
        if toot.content == None:
            continue

        if toot.media_list != "":
            toot.medias = []
            #media_list "1111,2222,333,"
            media_list = toot.media_list[:-1].split(",")

            for media_id in media_list:
                media = Media.query.get(int(media_id))
                if media != None:
                    toot.medias.append(media)

        if toot.emoji_list != "":
            toot.emojis = []
            #emoji_list "blobfoxaaa,blobcatwww,fox_think,"
            emoji_list = toot.emoji_list[:-1].split(",")

            for emoji_shortcode in emoji_list:
                emoji = Emoji.query.filter_by(shortcode=emoji_shortcode, acct=toot.acct).first()

                if emoji != None:
                    emoji_shortcode = ':' + emoji_shortcode + ':'
                    emoji_url = emoji.url
                    emoji_html = f'''
                    <img class="emojione custom-emoji" alt="{emoji_shortcode}" title="{emoji_shortcode}" src="{emoji.url}" >
                    '''
                    toot.content = toot.content.replace(emoji_shortcode, emoji_html)

        toots.append(toot)

    return render_template('view.html', toots=toots, pagination=toots_)

@app.route('/settings', methods=['GET', 'POST'])
def settings():
    if request.method == 'POST':
        account = request.form['account']
        timezone = request.form['timezone']

        if not account or len(account) > 30:
            flash('无效输入')
            return redirect(url_for('settings'))

        settings = Settings.query.first()

        if settings == None:
            settings = Settings(account=account, timezone=timezone)
            db.session.add(settings)
        else:
            settings.account = account
            settings.timezone = timezone

        db.session.commit()
        flash('设置已修改')
        return redirect(url_for('settings'))

    settings = Settings.query.first()
    app_init = os.path.isfile('pyBDSM_clientcred.secret') and os.path.isfile('user.secret')
    if settings == None:
        flash('请输入相关设置！')

    return render_template('settings.html',settings=settings, app_init=app_init)

@app.route('/register', methods=['GET', 'POST'])
def register():
    settings = Settings.query.first()
    if settings == None:
        flash('请先输入用户名！')
        return redirect(url_for('settings'))
    else:
        account = settings.account[1:]
        username, domain = account.split("@")
        url = "https://" + domain

        if request.method == 'POST':
            token = request.form['token'].rstrip()
            mastodon = Mastodon(client_id='pyBDSM_clientcred.secret', api_base_url=url)
            mastodon.log_in(code=token, to_file='user.secret', scopes=['read'])

            if os.path.isfile('user.secret'):
                flash('应用已授权！')
                return redirect(url_for('settings'))

        if not os.path.isfile('pyBDSM_clientcred.secret'):
            app_register(url)
        if not os.path.isfile('user.secret'):
            mastodon = Mastodon(client_id='pyBDSM_clientcred.secret', api_base_url=url)
            url = mastodon.auth_request_url(client_id='pyBDSM_clientcred.secret', scopes=['read'])
            return render_template('register.html',url=url)

        flash('已授权过！')
        return redirect(url_for('settings'))

@app.route('/archive', methods=['GET', 'POST'])
def archive():
    settings = Settings.query.first()
    if settings == None:
        return redirect(url_for('settings'))
    elif len(Toot.query.all()) > 0:
        flash('现暂不支持重复存档！') #TODO
        return redirect(url_for('index'))
    else:
        account = settings.account[1:]
        username, domain = account.split("@")
        url = "https://" + domain

        archive_toot(url)

    flash('存档完成……大概！')
    return redirect(url_for('index'))