southfox/mastodon-BDSM
1
0
Fork 0
Code Issues 6 Pull requests Projects Releases Packages Wiki Activity

feat/support show poll

Browse source
This commit is contained in:
SouthFox 2022-11-15 18:08:03 +08:00
parent b3e36c7b07
commit b496c95f10
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
BDSM/views.py
View file

@ -1,11 +1,12 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import os import os
import pytz import pytz
import json
from flask import render_template, request, url_for, redirect, flash, abort from flask import render_template, request, url_for, redirect, flash, abort
from flask_sqlalchemy import Pagination from flask_sqlalchemy import Pagination
from BDSM import app, db from BDSM import app, db
from BDSM.models import Media, Settings, Toot, Emoji, Other from BDSM.models import Media, Poll, Settings, Toot, Emoji, Other
from BDSM.toot import app_register, archive_toot, get_context from BDSM.toot import app_register, archive_toot, get_context
from mastodon import Mastodon from mastodon import Mastodon
from types import SimpleNamespace from types import SimpleNamespace
@ -235,5 +236,16 @@ def process_toot(toots_):
''' '''
toot.content = toot.content.replace(emoji_shortcode, emoji_html) toot.content = toot.content.replace(emoji_shortcode, emoji_html)
if toot.poll_id != None:
poll = Poll.query.get(toot.poll_id)
poll_options = json.loads(poll.options.replace("\'", "\""))
poll_count = str(poll.votes_count)
poll_content = '<strong>总票数： ' + poll_count + '</strong><br>'
for i in poll_options:
poll_content += i['title'] + " / " + str(i['votes_count']) + '<br>'
toot.content += poll_content
toots.append(toot) toots.append(toot)
return toots return toots