from django.contrib.auth.models import User from django.utils import timezone from bookmarks.models import Bookmark, BookmarkForm, parse_tag_string from bookmarks.services.tags import get_or_create_tags from bookmarks.services.website_loader import load_website_metadata def create_bookmark(form: BookmarkForm, current_user: User): bookmark = form.save(commit=False) # Update website info _update_website_metadata(bookmark) # Set currently logged in user as owner bookmark.owner = current_user # Set dates bookmark.date_added = timezone.now() bookmark.date_modified = timezone.now() bookmark.save() # Update tag list _update_bookmark_tags(bookmark, form.data['tag_string'], current_user) bookmark.save() def update_bookmark(form: BookmarkForm, current_user: User): bookmark = form.save(commit=False) # Update website info _update_website_metadata(bookmark) # Update tag list _update_bookmark_tags(bookmark, form.data['tag_string'], current_user) # Update dates bookmark.date_modified = timezone.now() bookmark.save() def _update_website_metadata(bookmark: Bookmark): metadata = load_website_metadata(bookmark.url) bookmark.website_title = metadata.title bookmark.website_description = metadata.description def _update_bookmark_tags(bookmark: Bookmark, tag_string: str, user: User): tag_names = parse_tag_string(tag_string, ' ') tags = get_or_create_tags(tag_names, user) bookmark.tags.set(tags)