aboutsummaryrefslogtreecommitdiff
path: root/web/controllers/icon.py
blob: 07c4a4ef13e5837d4f367c82c4c50ad814043274 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import base64
import requests
from web.models import Icon
from .abstract import AbstractController


class IconController(AbstractController):
    _db_cls = Icon
    _user_id_key = None

    def _build_from_url(self, attrs):
        if 'url' in attrs and 'content' not in attrs:
            resp = requests.get(attrs['url'], verify=False)
            attrs.update({'url': resp.url,
                    'mimetype': resp.headers.get('content-type', None),
                    'content': base64.b64encode(resp.content).decode('utf8')})
        return attrs

    def create(self, **attrs):
        return super().create(**self._build_from_url(attrs))

    def update(self, filters, attrs):
        return super().update(filters, self._build_from_url(attrs))
bgstack15