newspipe/controllers/icon.py


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

import base64

import requests

from .abstract import AbstractController
from newspipe.lib.utils import newspipe_get
from newspipe.models import Icon


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:
            try:
                resp = newspipe_get(attrs["url"], timeout=5)
                attrs.update(
                    {
                        "url": resp.url,
                        "mimetype": resp.headers.get("content-type", None),
                        "content": base64.b64encode(resp.content).decode("utf8"),
                    }
                )
            except requests.exceptions.ConnectionError:
                pass
        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))