diff --git a/.gitignore b/.gitignore index f8b73e7..41810c6 100644 --- a/.gitignore +++ b/.gitignore @@ -1,4 +1,11 @@ # ---> Python + +#sqlite +*.db + +#config +config.py + # Byte-compiled / optimized / DLL files __pycache__/ *.py[cod] diff --git a/app.py b/app.py new file mode 100644 index 0000000..3027399 --- /dev/null +++ b/app.py @@ -0,0 +1,114 @@ +from flask import Flask, request, render_template, send_from_directory, abort, redirect +from flask_sqlalchemy import SQLAlchemy +from flask_limiter import Limiter +from flask_limiter.util import get_remote_address +from mastodon import Mastodon +import re +from datetime import datetime +from dateutil.tz import tzlocal +import html2text +from config import C + +app = Flask(__name__) +app.config.from_object('config.C') + +th = Mastodon( + access_token = C.token, + api_base_url = 'https://' + C.domain +) + +limiter = Limiter( + app, + key_func=get_remote_address, + default_limits=["50 / minute"], +) + +h2t = html2text.HTML2Text() +h2t.ignore_links = True + +db = SQLAlchemy(app) + +class Candidate(db.Model): + id = db.Column(db.Integer, primary_key=True) + content = db.Column(db.String(400)) + url = db.Column(db.String(50)) + time = db.Column(db.DateTime) + toot = db.Column(db.BigInteger) + +db.create_all() + +@app.route('/js/') +def send_js(path): + return send_from_directory('static/js', path) +@app.route('/img/') +def send_img(path): + return send_from_directory('static/img', path) + +@app.route('/ordinary/') +def inbox(): + + cans = [{ + 'content': c.content, + 'url' : c.url, + 'toot': c.toot, + 'time': c.time.replace(tzinfo=tzlocal()) + } for c in Candidate.query.all() + ] + + return render_template('inbox.html', cans=cans) + +@app.route('/ordinary/new', methods=['POST']) +@limiter.limit("5 / hour; 1 / 2 second") +def new_one(): + + content = request.form.get('text') + print(content) + if not content or len(content)>400: + abort(422) + + if not Candidate.query.filter_by(content=content).first(): + + toot = th.status_post( + f"叮~ 有新的自荐报名（大家可以直接在此处评论）：\n\n{content}", + visibility='unlisted' + ) + + c = Candidate(content=content, toot=toot.id, time = datetime.now()) + db.session.add(c) + db.session.commit() + + return redirect(".") + +@limiter.limit("100 / hour; 2 / second") +@app.route('/ordinary/') +def get_replies(toot): + c = Candidate.query.filter_by(toot=toot).first() + if not c: + abort(404) + + context = th.status_context(toot) + replies = [ + { + 'disp': (t.account.display_name or t.account.acct), + 'url': t.account.url, + 'content': h2t.handle(t.content).replace(C.bot_name,'').strip(), + 'time': str(t.created_at) + } + for t in context.descendants + ] + d = list(filter( + lambda r: r['content'] == '删除' and r['url'].split('/@')[1] in C.admins, + replies + )) + print(d) + if d: + db.session.delete(c) + db.session.commit() + th.status_delete(toot) + return '该内容已被删除', 404 + + return {'replies': replies} + +if __name__ == '__main__': + app.run(debug=True) + diff --git a/config.sample.py b/config.sample.py new file mode 100644 index 0000000..d455be4 --- /dev/null +++ b/config.sample.py @@ -0,0 +1,8 @@ +class C(object): + SQLALCHEMY_DATABASE_URI = 'sqlite:///ord.db' + SQLALCHEMY_TRACK_MODIFICATIONS = False + JSON_AS_ASCII = True + domain = 'thu.closed.social' + bot_name = '@ordinary_bot' + token = 'token' + admins = [] diff --git a/templates/inbox.html b/templates/inbox.html new file mode 100644 index 0000000..0ad17b2 --- /dev/null +++ b/templates/inbox.html @@ -0,0 +1,114 @@ + + + + + + + + + + + + + 华清特将报名 + + + + +
+
+ +
+

华清大学
特普通奖学金

+
+
+ +
+
+ + + +
+
+ +
+

已有的报名

+ + {% for c in cans|reverse %} +
+ 
{{c.content}}
+ +
+
+ 加载中... +
+
+
+ {% endfor %} + + + +
+ + +