forked from sch/Slixfeed
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Based on http_upload.py example from project slixmpp
|
|
https://codeberg.org/poezio/slixmpp/src/branch/master/examples/http_upload.py
|
|
"""
|
|
|
|
import logging
|
|
from slixmpp.exceptions import IqTimeout, IqError
|
|
from slixmpp.plugins.xep_0363.http_upload import HTTPError
|
|
# import sys
|
|
|
|
class XmppUpload:
|
|
|
|
async def start(self, jid, filename, domain=None):
|
|
logging.info('Uploading file %s...', filename)
|
|
try:
|
|
upload_file = self['xep_0363'].upload_file
|
|
# if self.encrypted and not self['xep_0454']:
|
|
# print(
|
|
# 'The xep_0454 module isn\'t available. '
|
|
# 'Ensure you have \'cryptography\' '
|
|
# 'from extras_require installed.',
|
|
# file=sys.stderr,
|
|
# )
|
|
# return
|
|
# elif self.encrypted:
|
|
# upload_file = self['xep_0454'].upload_file
|
|
try:
|
|
url = await upload_file(
|
|
filename, domain, timeout=10,
|
|
)
|
|
logging.info('Upload successful!')
|
|
logging.info('Sending file to %s', jid)
|
|
except HTTPError:
|
|
url = (
|
|
"Error: It appears that this server doesn't support "
|
|
"HTTP File Upload."
|
|
)
|
|
logging.error(
|
|
"It appears that this server doesn't support HTTP File Upload."
|
|
)
|
|
# raise HTTPError(
|
|
# "This server doesn't appear to support HTTP File Upload"
|
|
# )
|
|
except IqTimeout:
|
|
raise TimeoutError('Could not send message in time')
|
|
return url
|