discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

refactored

Browse Source
This commit is contained in:
tonyrewin 2022-09-19 18:24:43 +03:00
parent 4536370c79
commit c479b12e65
1 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
migration/tables/content_items.py
View File

@ -13,7 +13,6 @@ from orm.user import User
from orm.topic import TopicFollower from orm.topic import TopicFollower
from services.stat.reacted import ReactedStorage from services.stat.reacted import ReactedStorage
from services.stat.viewed import ViewedStorage from services.stat.viewed import ViewedStorage
from services.zine.topics import TopicStorage
OLD_DATE = "2016-03-05 22:22:00.350000" OLD_DATE = "2016-03-05 22:22:00.350000"
ts = datetime.now() ts = datetime.now()
@ -88,20 +87,6 @@ def create_author_from_app(app):
async def create_shout(shout_dict, userslug): async def create_shout(shout_dict, userslug):
s = Shout.create(**shout_dict) s = Shout.create(**shout_dict)
with local_session() as session: with local_session() as session:
topics = session.query(ShoutTopic).where(ShoutTopic.shout == s.slug).all()
for tpc in topics:
tf = session.query(
TopicFollower
).where(
TopicFollower.follower == userslug
).filter(
TopicFollower.topic == tpc.slug
).first()
if not tf:
tf = TopicFollower.create(topic=tpc.slug, follower=userslug, auto=True)
session.add(tf)
await TopicStorage.update_topic(tpc.slug)
srf = session.query(ShoutReactionsFollower).where( srf = session.query(ShoutReactionsFollower).where(
ShoutReactionsFollower.shout == s.slug ShoutReactionsFollower.shout == s.slug
).filter( ).filter(
@ -180,6 +165,24 @@ async def migrate(entry, storage):
print("[migration] unknown old topic id: " + oid) print("[migration] unknown old topic id: " + oid)
r["topics"] = list(r["topics"]) r["topics"] = list(r["topics"])
# add author as TopicFollower
with local_session() as session:
for tpc in r['topics']:
tf = session.query(
TopicFollower
).where(
TopicFollower.follower == userslug
).filter(
TopicFollower.topic == tpc
).first()
if not tf:
tf = TopicFollower.create(
topic=tpc,
follower=userslug,
auto=True
)
session.add(tf)
entry["topics"] = r["topics"] entry["topics"] = r["topics"]
entry["cover"] = r["cover"] entry["cover"] = r["cover"]