aboutsummaryrefslogtreecommitdiff
path: root/src/web/controllers/bookmark.py
blob: b5413243efc4fe6124198e034a66d8fa47a76da9 (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
import logging
import itertools
from datetime import datetime, timedelta

from bootstrap import db
from web.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