fix: improve post migration script

This commit is contained in:
Vasily Zubarev 2023-01-12 15:44:08 +01:00
parent 03b706ef7b
commit ad11a1fd48
1 changed files with 18 additions and 1 deletions

View File

@ -1,5 +1,6 @@
import logging
import json
from datetime import datetime
from django.core.management import BaseCommand
from django.db import connections
@ -48,7 +49,7 @@ class Command(BaseCommand):
word_count=parse_word_count(row),
comment_count=row["comments_count"],
view_count=row["views_count"],
is_raw_html=bool(not row["text"] and row["html"]),
is_raw_html=parse_is_raw_html(row),
is_visible=row["is_visible"],
is_members_only=row["is_members_only"],
is_commentable=row["is_commentable"],
@ -80,6 +81,12 @@ def parse_text(row):
else:
text = row["html"]
if row["type"] == "world" and row["created_at"] < datetime(2017, 3, 1):
text = row["html"]
if row["slug"] == "389":
text = row["html"]
if text:
text = text.replace("i.vas3k.ru", "i.vas3k.blog")
text = text.replace("http://vas3k.ru", "https://vas3k.blog")
@ -88,6 +95,16 @@ def parse_text(row):
return text
def parse_is_raw_html(row):
if row["type"] == "world" and row["created_at"] < datetime(2017, 2, 20):
return True
if row["slug"] == "389":
return True
return bool(not row["text"] and row["html"])
def parse_image(row):
if row["image"]:
return row["image"].replace("i.vas3k.ru", "i.vas3k.blog").replace("http://", "https://")