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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
"""
This plugin sends a small image to the recipient of your choice, using XHTML-IM and Bits of Binary.
Usage
-----
/bob some/image.png
Configuration options
---------------------
.. glossary::
max_size
**Default:** ``2048``
The maximum acceptable size of a file, over which you will get an error instead.
max_age
**Default:** ``86400``
The time during which the file should stay in cache on the receiving side.
"""
from poezio.core.structs import Completion
from poezio.plugin import BasePlugin
from poezio import tabs
from pathlib import Path
from glob import glob
from os.path import expanduser
from mimetypes import guess_type
class Plugin(BasePlugin):
default_config = {'bob': {'max_size': 2048, 'max_age': 86400}}
def init(self):
for tab in tabs.DynamicConversationTab, tabs.StaticConversationTab, tabs.PrivateTab, tabs.MucTab:
self.api.add_tab_command(
tab,
'bob',
self.command_bob,
usage='<image>',
help='Send image <image> to the current discussion',
short='Send a short image',
completion=self.completion_bob)
async def command_bob(self, filename):
path = Path(expanduser(filename))
try:
size = path.stat().st_size
except OSError as exc:
self.api.information('Error sending “%s”: %s' % (path.name, exc),
'Error')
return
mime_type = guess_type(path.as_posix())[0]
if mime_type is None or not mime_type.startswith('image/'):
self.api.information(
'Error sending “%s”, not an image file.' % path.name, 'Error')
return
if size > self.config.get('max_size'):
self.api.information(
'Error sending “%s”, file too big.' % path.name, 'Error')
return
with open(path.as_posix(), 'rb') as file:
data = file.read()
max_age = self.config.get('max_age')
cid = await self.core.xmpp.plugin['xep_0231'].set_bob(
data, mime_type, max_age=max_age)
self.api.run_command(
'/xhtml <img src="cid:%s" alt="%s"/>' % (cid, path.name))
@staticmethod
def completion_bob(the_input):
txt = expanduser(the_input.get_text()[5:])
images = []
for filename in glob(txt + '*'):
mime_type = guess_type(filename)[0]
if mime_type is not None and mime_type.startswith('image/'):
images.append(filename)
return Completion(the_input.auto_completion, images, quotify=False)
|
