migration fixes

This commit is contained in:
tonyrewin 2022-06-25 17:31:37 +03:00
parent ddacbbb0d5
commit a075bfaf10
2 changed files with 1033 additions and 1003 deletions

View File

@ -87,7 +87,7 @@ class HTML2Text(html.parser.HTMLParser):
self.open_quote = config.OPEN_QUOTE # covered in cli self.open_quote = config.OPEN_QUOTE # covered in cli
self.close_quote = config.CLOSE_QUOTE # covered in cli self.close_quote = config.CLOSE_QUOTE # covered in cli
self.header_id = None self.header_id = None
self.span_hightlight = False self.span_highlight = False
self.span_lead = False self.span_lead = False
if out is None: if out is None:
@ -137,6 +137,7 @@ class HTML2Text(html.parser.HTMLParser):
self.preceding_stressed = False self.preceding_stressed = False
self.preceding_data = "" self.preceding_data = ""
self.current_tag = "" self.current_tag = ""
self.current_class = ""
config.UNIFIABLE["nbsp"] = "&nbsp_place_holder;" config.UNIFIABLE["nbsp"] = "&nbsp_place_holder;"
@ -340,7 +341,8 @@ class HTML2Text(html.parser.HTMLParser):
parent_style = self.tag_stack[-1][2] parent_style = self.tag_stack[-1][2]
if hn(tag): if hn(tag):
# check if nh is inside of an 'a' tag (incorrect but found in the wild) # check if nh is inside of an 'a' tag
# (incorrect but found in the wild)
if self.astack: if self.astack:
if start: if start:
self.inheader = True self.inheader = True
@ -362,19 +364,33 @@ class HTML2Text(html.parser.HTMLParser):
else: else:
self.inheader = False self.inheader = False
return # prevent redundant emphasis marks on headers return # prevent redundant emphasis marks on headers
if 'class' in attrs:
self.current_class = attrs.get('class')
# self.p()
if not start:
self.current_class = ''
if 'style' in attrs:
if attrs.get('style') == 'text-align: center':
self.current_class = 'center'
if not start:
self.current_class = ''
if tag == 'span': if tag == 'span':
if start and 'class' in attrs: if start:
if attrs['class'] == 'highlight': if self.current_class == 'highlight' and \
self.inheader == False and \
self.span_lead == False and \
self.astack == False:
self.o('`') # NOTE: same as <code> self.o('`') # NOTE: same as <code>
self.span_hightlight = True self.span_highlight = True
elif attrs['class'] == 'lead': elif self.current_class == 'lead' and \
self.inheader == False and \
self.span_highlight == False:
self.o('==') # NOTE: but CriticMarkup uses {== ==} self.o('==') # NOTE: but CriticMarkup uses {== ==}
self.span_lead = True self.span_lead = True
else: else:
if self.span_hightlight: if self.span_highlight:
self.o('`') self.o('`')
self.span_hightlight = False self.span_highlight = False
elif self.span_lead: elif self.span_lead:
self.o('==') self.o('==')
self.span_lead = False self.span_lead = False

View File

@ -39,6 +39,7 @@ def get_metadata(r):
metadata['createdAt'] = r.get('createdAt', ts) metadata['createdAt'] = r.get('createdAt', ts)
metadata['layout'] = r['layout'] metadata['layout'] = r['layout']
metadata['topics'] = [topic['slug'] for topic in r['topics']] metadata['topics'] = [topic['slug'] for topic in r['topics']]
metadata['topics'].sort()
if r.get('cover', False): if r.get('cover', False):
metadata['cover'] = r.get('cover') metadata['cover'] = r.get('cover')
return metadata return metadata
@ -80,7 +81,6 @@ def migrate(entry, users_by_oid, topics_by_oid):
'createdAt': entry.get('createdAt', '2016-03-05 22:22:00.350000') 'createdAt': entry.get('createdAt', '2016-03-05 22:22:00.350000')
} }
r['slug'] = entry.get('slug', '') r['slug'] = entry.get('slug', '')
body_orig = entry.get('body', '')
if not r['slug'] and entry.get('friendlySlugs') is not None: if not r['slug'] and entry.get('friendlySlugs') is not None:
r['slug'] = entry['friendlySlugs']['slug'][0]['slug'] r['slug'] = entry['friendlySlugs']['slug'][0]['slug']
if(r['slug'] is None): if(r['slug'] is None):
@ -94,12 +94,12 @@ def migrate(entry, users_by_oid, topics_by_oid):
mainTopic = topics_by_oid.get(category) mainTopic = topics_by_oid.get(category)
if mainTopic: if mainTopic:
r['mainTopic'] = mainTopic["slug"] r['mainTopic'] = mainTopic["slug"]
topic_oids = set([category]) topic_oids = [category, ]
topic_oids.update(entry.get("tags", [])) taglist = entry.get("tags", [])
topic_oids.extend(taglist)
for oid in topic_oids: for oid in topic_oids:
if oid in topics_by_oid: if oid in topics_by_oid:
r['topics'].append(topics_by_oid[oid]) r['topics'].append(topics_by_oid[oid])
if entry.get('image') is not None: if entry.get('image') is not None:
r['cover'] = entry['image']['url'] r['cover'] = entry['image']['url']
if entry.get('thumborId') is not None: if entry.get('thumborId') is not None:
@ -116,7 +116,7 @@ def migrate(entry, users_by_oid, topics_by_oid):
else: else:
body_html = str(BeautifulSoup( body_html = str(BeautifulSoup(
body_orig, features="html.parser")) body_orig, features="html.parser"))
r['body'] = body_html # html2text(body_html) r['body'] = html2text(body_html)
else: else:
print(r['slug'] + ': literature has no media') print(r['slug'] + ': literature has no media')
elif entry.get('type') == 'Video': elif entry.get('type') == 'Video':
@ -127,17 +127,31 @@ def migrate(entry, users_by_oid, topics_by_oid):
if video_url == '#': if video_url == '#':
video_url = 'https://vimeo.com/' + vm if vm else '#' video_url = 'https://vimeo.com/' + vm if vm else '#'
if video_url == '#': if video_url == '#':
print(entry.get('media', 'NO MEDIA!')) print(entry.get('media', 'UNKNOWN MEDIA PROVIDER!'))
# raise Exception # raise Exception
r['body'] = '<ShoutVideo src=\"' + video_url + \ therestof = html2text(m.get('body', ''))
'\" />' + html2text(m.get('body', '')) # FIXME r['body'] = 'import VideoPlayer from \"src/components/Article/VideoPlayer\"\n' + \
'<VideoPlayer src=\"''' + video_url + '\" />\n\n' + therestof
elif entry.get('type') == 'Music': elif entry.get('type') == 'Music':
r['body'] = '<ShoutMusic media={\"' + \ r['body'] = 'import MusicPlayer from \"src/components/MusicPlayer\"\n'
json.dumps(entry['media']) + '\"} />' # FIXME for m in entry['media']:
if m == { 'main': 'true' } or m == { 'main': True } or m == {}:
continue
# TODO: mark highlighted track isMain == True
try: r['body'] += '<MusicPlayer src=\"' + m['fileUrl'] + '\"'
except: print(m)
try: r['body'] += ' title=\"' + m['title'] + '\"'
except: print(m)
r['body'] += ' />\n\n'
r['body'] += html2text(m.get('body', ''))
elif entry.get('type') == 'Image':
m = r.get('media')
try: r['body'] = '<img src=\"' + r['cover'] + '\" />'
except: print(entry)
if r.get('body') is None: if r.get('body') is None:
body_orig = entry.get('body', '') body_orig = entry.get('body', '')
body_html = str(BeautifulSoup(body_orig, features="html.parser")) body_html = str(BeautifulSoup(body_orig, features="html.parser"))
r['body'] = body_html # html2text(body_html) r['body'] = html2text(body_html)
body = r.get('body', '') body = r.get('body', '')
# get author data # get author data
@ -174,10 +188,10 @@ def migrate(entry, users_by_oid, topics_by_oid):
shout_dict['authors'] = [ author, ] shout_dict['authors'] = [ author, ]
if entry['published']: if entry['published']:
metadata = get_metadata(r) metadata = get_metadata(shout_dict)
content = frontmatter.dumps(frontmatter.Post(body, **metadata)) content = frontmatter.dumps(frontmatter.Post(body, **metadata))
ext = 'md' ext = 'mdx'
open('migration/content/' + r['layout'] + '/' + r['slug'] + '.' + ext, 'w').write(content) open('../discoursio-web/content/' + r['layout'] + '/' + r['slug'] + '.' + ext, 'w').write(content)
try: try:
shout_dict['createdAt'] = date_parse(r.get('createdAt')) if entry.get('createdAt') else ts shout_dict['createdAt'] = date_parse(r.get('createdAt')) if entry.get('createdAt') else ts
shout_dict['publishedAt'] = date_parse(entry.get('publishedAt')) if entry.get('published') else None shout_dict['publishedAt'] = date_parse(entry.get('publishedAt')) if entry.get('published') else None