aboutsummaryrefslogtreecommitdiff
path: root/newspipe/controllers/bookmark.py
blob: 3c9dcce9843405d802c2234cc81c76cbd323d851 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import logging
import itertools
from datetime import datetime, timedelta

from newspipe.bootstrap import db
from newspipe.models import Bookmark
from .abstract import AbstractController
from .tag import BookmarkTagController

logger = logging.getLogger(__name__)


class BookmarkController(AbstractController):
    _db_cls = Bookmark

    def count_by_href(self, **filters):
        return self._count_by(Bookmark.href, filters)

    def update(self, filters, attrs):
        BookmarkTagController(self.user_id).read(
            **{"bookmark_id": filters["id"]}
        ).delete()

        for tag in attrs["tags"]:
            BookmarkTagController(self.user_id).create(
                **{
                    "text": tag.text,
                    "id": tag.id,
                    "bookmark_id": tag.bookmark_id,
                    "user_id": tag.user_id,
                }
            )

        del attrs["tags"]
        return super().update(filters, attrs)
bgstack15