newthuhole/hole_thu_backend
4
6
Fork 2
Code Issues 7 Pull Requests Projects Releases Wiki Activity

自定义头衔

Browse Source
This commit is contained in:
hole-thu
2021-12-24 00:07:39 +08:00
parent 57ca9f41d3
commit e8ef07cceb
4 changed files with 46 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
clear_redis.py Normal file
View File

@@ -0,0 +1,5 @@
# 每次重置时执行
from utils import rds, RDS_KEY_TITLE, RDS_KEY_BLOCKED_COUNT
rds.delete(RDS_KEY_BLOCKED_COUNT)
rds.delete(RDS_KEY_TITLE)

45
hole.py
View File

@@ -10,7 +10,7 @@ from sqlalchemy.sql.expression import func
from mastodon import Mastodon
from models import db, User, Post, Comment, Attention, TagRecord, Syslog
from utils import get_current_username, map_post, map_comment, map_syslog, check_attention, hash_name, look, get_num, tmp_token, is_admin, check_can_del, rds, RDS_KEY_POLL_OPTS, RDS_KEY_POLL_VOTES, gen_poll_dict, name_with_tmp_limit, RDS_KEY_BLOCK_SET, RDS_KEY_BLOCKED_COUNT, RDS_KEY_DANGEROUS_USERS
from utils import get_current_username, map_post, map_comment, map_syslog, check_attention, hash_name, look, get_num, tmp_token, is_admin, check_can_del, rds, RDS_KEY_POLL_OPTS, RDS_KEY_POLL_VOTES, gen_poll_dict, name_with_tmp_limit, RDS_KEY_BLOCK_SET, RDS_KEY_BLOCKED_COUNT, RDS_KEY_DANGEROUS_USERS, RDS_KEY_TITLE
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///hole.db'
@@ -24,7 +24,6 @@ app.config.from_pyfile('config.py')
db.init_app(app)
migrate = Migrate(app, db)
CS_LOGIN_URL = Mastodon(api_base_url=app.config['MASTODON_URL']) \
.auth_request_url(
client_id=app.config['CLIENT_ID'],
@@ -41,9 +40,6 @@ limiter = Limiter(
PER_PAGE = 50
DANGEROUS_USER_THRESHOLD = 10
# 重置后旧的被拉黑次数可以丢弃了，但其他仍需要保留
rds.delete(RDS_KEY_BLOCKED_COUNT)
class APIError(Exception):
msg = '未知错误'
@@ -138,6 +134,7 @@ def get_list():
return {
'code': 0,
'tmp_token': tmp_token(),
'custom_title': rds.hget(RDS_KEY_TITLE, hash_name(username)),
'count': len(data),
'data': data
}
@@ -185,8 +182,6 @@ def get_multi():
}
@app.route('/_api/v1/search')
def search():
username = get_current_username()
@@ -248,6 +243,7 @@ def do_post():
post_type = request.form.get('type')
cw = request.form.get('cw', '').strip()
poll_options = request.form.getlist('poll_options')
use_title = request.form.get('use_title')
if not content or len(content) > 4096 or len(cw) > 32:
raise APIError('无内容或超长')
@@ -263,8 +259,10 @@ def do_post():
if max(map(len, poll_options)) > 32:
raise APIError('选项过长')
name_hash = hash_name(username)
p = Post(
name_hash=hash_name(username),
name_hash=name_hash,
author_title=rds.hget(RDS_KEY_TITLE, name_hash) if use_title else None,
content=content,
search_text=search_text,
post_type=post_type,
@@ -351,14 +349,18 @@ def do_comment():
if post.deleted and not check_can_del(username, post.name_hash):
abort(451)
content = request.form.get('text')
content = content.strip() if content else None
content = '[tmp]\n' + content if username[:4] == 'tmp_' else content
content = request.form.get('text', '').strip()
if username.startswith('tmp_'):
content = '[tmp]\n' + content
if not content or len(content) > 4096:
abort(422)
use_title = request.form.get('use_title')
name_hash = hash_name(username)
c = Comment(
name_hash=hash_name(username),
name_hash=name_hash,
author_title=rds.hget(RDS_KEY_TITLE, name_hash) if use_title else None,
content=content,
)
post.comments.append(c)
@@ -518,6 +520,7 @@ def system_log():
'start_time': app.config['START_TIME'],
'salt': look(app.config['SALT']),
'tmp_token': tmp_token(),
'custom_title': rds.hget(RDS_KEY_TITLE, hash_name(username)),
'data': [map_syslog(s, username) for s in ss]
}
@@ -627,5 +630,23 @@ def block_user_by_target():
}
@app.route('/_api/v1/title', methods=['POST'])
@limiter.limit("10 / hour; 1 / 2 second")
def set_title():
username = get_current_username()
title = request.form.get('title')
if not title:
rds.hdel(RDS_KEY_TITLE, hash_name(username))
else:
if len(title) > 10:
raise APIError('自定义头衔太长')
if title in rds.hvals(RDS_KEY_TITLE): # 如果未来量大还是另外用个set维护
raise APIError('已经被使用了')
rds.hset(RDS_KEY_TITLE, hash_name(username), title)
return {'code': 0}
if __name__ == '__main__':
app.run(debug=True)

2
models.py
View File

@@ -18,6 +18,7 @@ class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
name_hash = db.Column(db.String(64))
author_title = db.Column(db.String(10))
content = db.Column(db.String(4096))
search_text = db.Column(db.String(4096), default='', index=True)
post_type = db.Column(db.String(8))
@@ -44,6 +45,7 @@ class Post(db.Model):
class Comment(db.Model):
id = db.Column(db.Integer, primary_key=True)
name_hash = db.Column(db.String(64))
author_title = db.Column(db.String(10))
content = db.Column(db.String(4096))
timestamp = db.Column(db.Integer)
deleted = db.Column(db.Boolean, default=False)

8
utils.py
View File

@@ -14,6 +14,8 @@ RDS_KEY_BLOCK_SET = 'hole_thu:block_list:%s' # key的参数是name而非namehas
RDS_KEY_BLOCKED_COUNT = 'hole_thu:blocked_count' # namehash -> 被拉黑次数
RDS_KEY_DANGEROUS_USERS = 'hole_thu:dangerous_users'
RDS_KEY_TITLE = 'hole_thu:title' # 用户自己设置的专属头衔, namehash -> 头衔
rds = redis.Redis(**RDS_CONFIG)
@@ -31,7 +33,7 @@ def tmp_token():
)[5:21]
def get_current_username():
def get_current_username() -> str:
token = request.headers.get('User-Token') or request.args.get('user_token')
if not token:
abort(401)
@@ -72,7 +74,8 @@ def map_post(p, name, mc=50):
'attention': check_attention(name, p.id),
'can_del': check_can_del(name, p.name_hash),
'allow_search': bool(p.search_text),
'poll': None if blocked else gen_poll_dict(p.id, name)
'poll': None if blocked else gen_poll_dict(p.id, name),
'author_title': p.author_title
}
if is_admin(name):
r['hot_score'] = p.hot_score
@@ -134,6 +137,7 @@ def map_comment(p, name):
'blocked': (blocked := is_blocked(c.name_hash, name)),
'cid': c.id,
'name_id': gen_name_id(c.name_hash),
'author_title': c.author_title,
'pid': p.id,
'text': '' if blocked else c.content,
'timestamp': c.timestamp,