33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from migration.extract import extract_md, html2text
|
|
from base.orm import local_session
|
|
from orm import Topic, Community
|
|
|
|
|
|
def migrate(entry):
|
|
body_orig = entry.get("description", "").replace(" ", " ")
|
|
topic_dict = {
|
|
"slug": entry["slug"],
|
|
"oid": entry["_id"],
|
|
"title": entry["title"].replace(" ", " "),
|
|
"children": [],
|
|
"community": Community.default_community.slug,
|
|
}
|
|
topic_dict["body"] = extract_md(html2text(body_orig), entry["_id"])
|
|
with local_session() as session:
|
|
slug = topic_dict["slug"]
|
|
topic = session.query(Topic).filter(Topic.slug == slug).first() or Topic.create(
|
|
**topic_dict
|
|
)
|
|
if not topic:
|
|
raise Exception("no topic!")
|
|
if topic:
|
|
if len(topic.title) > len(topic_dict["title"]):
|
|
Topic.update(topic, {"title": topic_dict["title"]})
|
|
if len(topic.body) < len(topic_dict["body"]):
|
|
Topic.update(topic, {"body": topic_dict["body"]})
|
|
session.commit()
|
|
# print(topic.__dict__)
|
|
rt = topic.__dict__.copy()
|
|
del rt["_sa_instance_state"]
|
|
return rt
|