39 lines
905 B
Python
39 lines
905 B
Python
from datetime import datetime
|
|
|
|
from slixmpp.plugins.xep_0082 import format_datetime, parse
|
|
from slixmpp.xmlstream import ElementBase
|
|
|
|
NS = "urn:xmpp:file:metadata:0"
|
|
|
|
|
|
class File(ElementBase):
|
|
name = "file"
|
|
namespace = NS
|
|
plugin_attrib = "file"
|
|
interfaces = sub_interfaces = {"media-type", "name", "date", "size", "hash", "desc"}
|
|
|
|
def set_size(self, size: int):
|
|
self._set_sub_text("size", str(size))
|
|
|
|
def get_size(self):
|
|
return _int_or_none(self._get_sub_text("size"))
|
|
|
|
def get_date(self):
|
|
try:
|
|
return parse(self._get_sub_text("date"))
|
|
except ValueError:
|
|
return
|
|
|
|
def set_date(self, stamp: datetime):
|
|
try:
|
|
self._set_sub_text("date", format_datetime(stamp))
|
|
except ValueError:
|
|
pass
|
|
|
|
|
|
def _int_or_none(v):
|
|
try:
|
|
return int(v)
|
|
except ValueError:
|
|
return None
|