github/pelican
1
0
Fork 1
mirror of https://github.com/getpelican/pelican.git synced 2025-10-15 20:28:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Parameterize get_attachments attachment resolution by id

Browse source
This commit is contained in:
Stuart Axon 2018-12-05 02:15:04 +00:00 • committed by Stuart Axon
commit f1580e19f0
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
pelican/tools/pelican_import.py
View file

@ -677,9 +677,10 @@ def get_out_filename(output_path, filename, ext, kind,
return out_filename return out_filename
def get_attachments(xml): def get_attachments(xml, resolve_by_id=False):
""" """
:param resolve_by_id: Lookup posts by id
:param xml: :param xml:
:return: attachment_urls :return: attachment_urls
""" """
@ -708,8 +709,9 @@ def get_attachments(xml):
if kind == 'attachment': if kind == 'attachment':
attachments.append((item.find('post_parent').string, attachments.append((item.find('post_parent').string,
item.find('attachment_url').string)) item.find('attachment_url').string))
attachment_ids[post_id] = item.find('attachment_url').string if resolve_by_id:
elif kind == 'post': attachment_ids[post_id] = item.find('attachment_url').string
elif resolve_by_id and kind == 'post':
content = item.find('encoded').string content = item.find('encoded').string
find_attachment = re.compile(r'({}/\?attachment_id=(\d+))'.format(server)) find_attachment = re.compile(r'({}/\?attachment_id=(\d+))'.format(server))
for url, attachment_id in find_attachment.findall(content): for url, attachment_id in find_attachment.findall(content):
@ -722,7 +724,7 @@ def get_attachments(xml):
attachedposts[parent_name].add(url) attachedposts[parent_name].add(url)
attachment_links = defaultdict(set) attachment_links = defaultdict(set)
if attachments_by_id: if resolve_by_id and attachments_by_id:
for filename, links in attachments_by_id.items(): for filename, links in attachments_by_id.items():
for url, attachment_id in links: for url, attachment_id in links:
destination = attachment_ids[attachment_id] destination = attachment_ids[attachment_id]
@ -1052,7 +1054,7 @@ def main():
fields = feed2fields(args.input) fields = feed2fields(args.input)
if args.wp_attach: if args.wp_attach:
attachments, attachment_links = get_attachments(args.input) attachments, attachment_links = get_attachments(args.input, args.wp_resolve)
else: else:
attachments, attachment_links = None, None attachments, attachment_links = None, None