linkding/bookmarks/services/bookmarks.py

48 lines
1.3 KiB
Python
Raw Normal View History

2019-06-29 00:01:26 +00:00
import requests
from bs4 import BeautifulSoup
2019-06-28 17:37:41 +00:00
from django.contrib.auth.models import User
from django.utils import timezone
from bookmarks.models import Bookmark
def create_bookmark(bookmark: Bookmark, current_user: User):
# 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()
2019-06-28 22:27:20 +00:00
bookmark.date_modified = timezone.now()
bookmark.save()
def update_bookmark(bookmark: Bookmark):
# Update website info
_update_website_metadata(bookmark)
# Update dates
bookmark.date_modified = timezone.now()
2019-06-28 17:37:41 +00:00
bookmark.save()
def _update_website_metadata(bookmark: Bookmark):
2019-06-29 00:01:26 +00:00
# noinspection PyBroadException
try:
page_text = load_page(bookmark.url)
soup = BeautifulSoup(page_text, 'html.parser')
title = soup.title.string if soup.title is not None else None
description_tag = soup.find('meta', attrs={'name': 'description'})
description = description_tag['content'] if description_tag is not None else None
bookmark.website_title = title
bookmark.website_description = description
except Exception:
bookmark.website_title = None
bookmark.website_description = None
def load_page(url: str):
r = requests.get(url)
return r.text