blob: 0fa7a39eb5041cf7a530a98da7c28a55d71e0041 (
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
|
import base64
import requests
from newspipe.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))
|