Add attachment refs to the posts with saved content

boards/models/post/sync.py

@@ -61,8 +61,20 b' class SyncManager:'
 
             global_id = post.global_id
 
+            images = post.images.all()
+            attachments = post.attachments.all()
             if global_id.content:
                 model.append(et.fromstring(global_id.content))
+                if len(images) > 0 or len(attachments) > 0:
+                    attachment_refs = et.SubElement(model, TAG_ATTACHMENT_REFS)
+                    for image in images:
+                        SyncManager._attachment_to_xml(
+                            None, attachment_refs, image.image.file,
+                            image.hash, image.image.url)
+                    for file in attachments:
+                        SyncManager._attachment_to_xml(
+                            None, attachment_refs, file.file.file,
+                            file.hash, file.file.url)
             else:
                 content_tag = et.SubElement(model, TAG_CONTENT)
 
@@ -89,8 +101,6 b' class SyncManager:'
                 pub_time = et.SubElement(content_tag, TAG_PUB_TIME)
                 pub_time.text = str(post.get_pub_time_str())
 
-                images = post.images.all()
-                attachments = post.attachments.all()
                 if len(images) > 0 or len(attachments) > 0:
                     attachments_tag = et.SubElement(content_tag, TAG_ATTACHMENTS)
                     attachment_refs = et.SubElement(model, TAG_ATTACHMENT_REFS)
@@ -248,10 +258,11 b' class SyncManager:'
 
     @staticmethod
     def _attachment_to_xml(tag_attachments, tag_refs, file, hash, url):
-        mimetype = get_file_mimetype(file)
-        attachment = et.SubElement(tag_attachments, TAG_ATTACHMENT)
-        attachment.set(ATTR_MIMETYPE, mimetype)
-        attachment.text = hash
+        if tag_attachments is not None:
+            mimetype = get_file_mimetype(file)
+            attachment = et.SubElement(tag_attachments, TAG_ATTACHMENT)
+            attachment.set(ATTR_MIMETYPE, mimetype)
+            attachment.text = hash
 
         attachment_ref = et.SubElement(tag_refs, TAG_ATTACHMENT_REF)
         attachment_ref.set(ATTR_REF, hash)