Zergling_man/moment
Zergling_man/moment
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement Media download

Browse Source
This commit is contained in:
miruka 2019-11-04 10:29:18 -04:00
parent 484eefe86d
commit 02c6ec4500
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/python/media_cache.py
View File

@ -79,7 +79,17 @@ class Media:
async def _get_remote_data(self) -> bytes: async def _get_remote_data(self) -> bytes:
raise NotImplementedError() parsed = urlparse(self.mxc)
resp = await self.cache.client.download(
server_name = parsed.netloc,
media_id = parsed.path.lstrip("/"),
)
if isinstance(resp, nio.DownloadError):
raise DownloadFailed(resp.message, resp.status_code)
return resp.body
@dataclass @dataclass
@ -180,5 +190,9 @@ class MediaCache:
self.downloads_dir.mkdir(parents=True, exist_ok=True) self.downloads_dir.mkdir(parents=True, exist_ok=True)
async def get_media(self, mxc: str) -> str:
return str(await Media(self, mxc, data=None).get())
async def get_thumbnail(self, mxc: str, width: int, height: int) -> str: async def get_thumbnail(self, mxc: str, width: int, height: int) -> str:
return str(await Thumbnail(self, mxc, None, (width, height)).get()) return str(await Thumbnail(self, mxc, None, (width, height)).get())