81b37cb7d2
refactor: clean up code to py39+ supported syntax - f-strings instead of format - latest typing support instead of pre 3.9 TitleCase - remove UTF-8 declarations. - many more changes Powered by https://github.com/asottile/pyupgrade/ + manual cleanups
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
import re
|
|
|
|
import frappe
|
|
from frappe import _
|
|
from frappe.rate_limiter import rate_limit
|
|
from frappe.utils import add_to_date, now
|
|
from frappe.utils.html_utils import clean_html
|
|
from frappe.website.doctype.blog_settings.blog_settings import get_comment_limit
|
|
from frappe.website.utils import clear_cache
|
|
|
|
URLS_COMMENT_PATTERN = re.compile(
|
|
r"http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+", re.IGNORECASE
|
|
)
|
|
EMAIL_PATTERN = re.compile(r"(^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$)", re.IGNORECASE)
|
|
|
|
|
|
@frappe.whitelist(allow_guest=True)
|
|
@rate_limit(key="reference_name", limit=get_comment_limit, seconds=60 * 60)
|
|
def add_comment(comment, comment_email, comment_by, reference_doctype, reference_name, route):
|
|
doc = frappe.get_doc(reference_doctype, reference_name)
|
|
|
|
if frappe.session.user == "Guest" and doc.doctype not in ["Blog Post", "Web Page"]:
|
|
return
|
|
|
|
if not comment.strip():
|
|
frappe.msgprint(_("The comment cannot be empty"))
|
|
return False
|
|
|
|
if URLS_COMMENT_PATTERN.search(comment) or EMAIL_PATTERN.search(comment):
|
|
frappe.msgprint(_("Comments cannot have links or email addresses"))
|
|
return False
|
|
|
|
comment = doc.add_comment(
|
|
text=clean_html(comment), comment_email=comment_email, comment_by=comment_by
|
|
)
|
|
|
|
comment.db_set("published", 1)
|
|
|
|
# since comments are embedded in the page, clear the web cache
|
|
if route:
|
|
clear_cache(route)
|
|
|
|
content = (
|
|
comment.content
|
|
+ "<p><a href='{}/app/Form/Comment/{}' style='font-size: 80%'>{}</a></p>".format(
|
|
frappe.utils.get_request_site_address(), comment.name, _("View Comment")
|
|
)
|
|
)
|
|
|
|
if doc.doctype == "Blog Post" and not doc.enable_email_notification:
|
|
pass
|
|
else:
|
|
# notify creator
|
|
frappe.sendmail(
|
|
recipients=frappe.db.get_value("User", doc.owner, "email") or doc.owner,
|
|
subject=_("New Comment on {0}: {1}").format(doc.doctype, doc.name),
|
|
message=content,
|
|
reference_doctype=doc.doctype,
|
|
reference_name=doc.name,
|
|
)
|
|
|
|
# revert with template if all clear (no backlinks)
|
|
template = frappe.get_template("templates/includes/comments/comment.html")
|
|
return template.render({"comment": comment.as_dict()})
|