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

db/change db model name

Browse source
This commit is contained in:
SouthFox 2022-11-07 22:42:20 +08:00
parent 327bb0d732
commit 7fcff2566a
3 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
BDSM/models.py
View file

@ -26,7 +26,7 @@ class Toot(db.Model):
favourites_count = db.Column(db.Integer)
language = db.Column(db.Text)
class Reblog(db.Model):
class Other(db.Model):
id = db.Column(db.Integer, primary_key=True)
acct = db.Column(db.Text)
url = db.Column(db.Text)

4
BDSM/toot.py
View file

@ -2,7 +2,7 @@
from mastodon import Mastodon
from BDSM import db
from BDSM.models import Reblog, Toot, Tag, Media, Emoji, Poll
from BDSM.models import Other, Toot, Tag, Media, Emoji, Poll
import sys
def app_register(url):
@ -160,7 +160,7 @@ def archive_toot(url):
favourites_count = status['favourites_count']
language = status['language']
table = Reblog() if is_reblog else Toot()
table = Other() if is_reblog else Toot()
table.id=id
table.acct = acct
table.url=url

4
BDSM/views.py
View file

@ -5,7 +5,7 @@ import pytz
from flask import render_template, request, url_for, redirect, flash
from flask_sqlalchemy import Pagination
from BDSM import app, db
from BDSM.models import Media, Settings, Toot, Emoji, Reblog
from BDSM.models import Media, Settings, Toot, Emoji, Other
from BDSM.toot import app_register, archive_toot
from mastodon import Mastodon
from types import SimpleNamespace
@ -179,7 +179,7 @@ def process_toot(toots_):
toot = SimpleNamespace(**toot.__dict__)
toot.is_reblog = True
else:
toot = Reblog.query.get(toot.reblog_id)
toot = Other.query.get(toot.reblog_id)
toot = SimpleNamespace(**toot.__dict__)
toot.is_reblog = True