mirror of
https://github.com/makeplane/plane
synced 2024-06-14 14:31:34 +00:00
cb593538e4
* dev: pages migrations * dev: page models * dev: api migrations * chore: apis for pages migrations * chore: dropped project id from page label and logs * dev: pages logger exception --------- Co-authored-by: NarayanBavisetti <narayan3119@gmail.com> Co-authored-by: sriram veeraghanta <veeraghanta.sriram@gmail.com>
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
# Python imports
|
|
import json
|
|
|
|
# Django imports
|
|
from django.utils import timezone
|
|
|
|
# Third-party imports
|
|
from bs4 import BeautifulSoup
|
|
|
|
# Module imports
|
|
from plane.db.models import Page, PageLog
|
|
from celery import shared_task
|
|
|
|
|
|
def extract_components(value, tag):
|
|
try:
|
|
mentions = []
|
|
html = value.get("description_html")
|
|
soup = BeautifulSoup(html, "html.parser")
|
|
mention_tags = soup.find_all(tag)
|
|
|
|
for mention_tag in mention_tags:
|
|
mention = {
|
|
"id": mention_tag.get("id"),
|
|
"entity_identifier": mention_tag.get("entity_identifier"),
|
|
"entity_name": mention_tag.get("entity_name"),
|
|
}
|
|
mentions.append(mention)
|
|
|
|
return mentions
|
|
except Exception:
|
|
return []
|
|
|
|
|
|
@shared_task
|
|
def page_transaction(new_value, old_value, page_id):
|
|
page = Page.objects.get(pk=page_id)
|
|
new_page_mention = PageLog.objects.filter(page_id=page_id).exists()
|
|
|
|
old_value = json.loads(old_value) if old_value else {}
|
|
|
|
new_transactions = []
|
|
deleted_transaction_ids = set()
|
|
|
|
# TODO - Add "issue-embed-component", "img", "todo" components
|
|
components = ["mention-component"]
|
|
for component in components:
|
|
old_mentions = extract_components(old_value, component)
|
|
new_mentions = extract_components(new_value, component)
|
|
|
|
new_mentions_ids = {mention["id"] for mention in new_mentions}
|
|
old_mention_ids = {mention["id"] for mention in old_mentions}
|
|
deleted_transaction_ids.update(old_mention_ids - new_mentions_ids)
|
|
|
|
new_transactions.extend(
|
|
PageLog(
|
|
transaction=mention["id"],
|
|
page_id=page_id,
|
|
entity_identifier=mention["entity_identifier"],
|
|
entity_name=mention["entity_name"],
|
|
workspace_id=page.workspace_id,
|
|
created_at=timezone.now(),
|
|
updated_at=timezone.now(),
|
|
)
|
|
for mention in new_mentions
|
|
if mention["id"] not in old_mention_ids or not new_page_mention
|
|
)
|
|
|
|
# Create new PageLog objects for new transactions
|
|
PageLog.objects.bulk_create(
|
|
new_transactions, batch_size=10, ignore_conflicts=True
|
|
)
|
|
|
|
# Delete the removed transactions
|
|
PageLog.objects.filter(transaction__in=deleted_transaction_ids).delete()
|