kopia lustrzana https://gitlab.com/jaywink/federation
WIP process AP payloads with a json-ld processor (calamus)
Alain St-Denis
rodzic
01357aacb0
commit
a0282eec11
|
|
@ -3,7 +3,7 @@ CONTEXT_DIASPORA = {"diaspora": "https://diasporafoundation.org/ns/"}
|
|||
CONTEXT_HASHTAG = {"Hashtag": "as:Hashtag"}
|
||||
CONTEXT_LD_SIGNATURES = "https://w3id.org/security/v1"
|
||||
CONTEXT_MANUALLY_APPROVES_FOLLOWERS = {"manuallyApprovesFollowers": "as:manuallyApprovesFollowers"}
|
||||
CONTEXT_PYTHON_FEDERATION = {"pyfed": "https://docs.jasonrobinson.me/ns/python-federation"}
|
||||
CONTEXT_PYTHON_FEDERATION = {"pyfed": "https://docs.jasonrobinson.me/ns/python-federation#"}
|
||||
CONTEXT_SENSITIVE = {"sensitive": "as:sensitive"}
|
||||
|
||||
CONTEXTS_DEFAULT = [
|
||||
|
|
|
|||
|
|
@ -62,6 +62,16 @@ class CleanContentMixin(RawContentMixin):
|
|||
return
|
||||
return attrs
|
||||
|
||||
# payload went through the json-ld processor
|
||||
if hasattr(self, 'content_map'):
|
||||
self._rendered_content = self.content_map['orig'].strip()
|
||||
if hasattr(self, 'source') and self.source.get('mediaType') == 'text/markdown':
|
||||
self._media_type = self.source['mediaType']
|
||||
self.raw_content = self.source.get('content').strip()
|
||||
else:
|
||||
self._media_type = 'text/html'
|
||||
self.raw_content = self.content_map['orig']
|
||||
|
||||
if self._media_type == "text/markdown":
|
||||
# Skip when markdown
|
||||
return
|
||||
|
|
@ -301,6 +311,20 @@ class ActivitypubProfile(ActivitypubEntityMixin, Profile):
|
|||
_type = ActorType.PERSON.value
|
||||
public = True
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
super().__init__(*args, **kwargs)
|
||||
|
||||
if hasattr(self, 'schema'):
|
||||
self.inboxes = {'private': self.inbox, 'public':self.endpoints['sharedInbox']}
|
||||
self.public_key = self.publicKey['publicKeyPem']
|
||||
self.image_urls = {}
|
||||
if hasattr(self, 'icon'):
|
||||
self.image_urls = {
|
||||
'small': self.icon[0].url,
|
||||
'medium': self.icon[0].url,
|
||||
'large': self.icon[0].url
|
||||
}
|
||||
|
||||
def to_as2(self) -> Dict:
|
||||
as2 = {
|
||||
"@context": CONTEXTS_DEFAULT + [
|
||||
|
|
|
|||
|
|
@ -8,6 +8,7 @@ from federation.entities.activitypub.entities import (
|
|||
from federation.entities.base import Follow, Profile, Accept, Post, Comment, Retraction, Share, Image
|
||||
from federation.entities.mixins import BaseEntity
|
||||
from federation.types import UserType, ReceiverVariant
|
||||
from federation.entities.activitypub.schemas import schema_to_objects
|
||||
|
||||
logger = logging.getLogger("federation")
|
||||
|
||||
|
|
@ -52,6 +53,19 @@ def element_to_objects(payload: Dict) -> List:
|
|||
"""
|
||||
cls = None
|
||||
entities = []
|
||||
|
||||
# Initial attempt at handling json-ld with calamus
|
||||
# Fall back to legacy if AP payload is not supported yet
|
||||
entity = schema_to_objects(payload)
|
||||
if entity:
|
||||
logger.warning("Entity %s handled through the json-ld processor", entity)
|
||||
entity._source_protocol = "activitypub"
|
||||
entity._source_payload = payload
|
||||
entity._receivers = extract_receivers(payload)
|
||||
if hasattr(entity, "post_receive"):
|
||||
entity.post_receive()
|
||||
return [entity]
|
||||
|
||||
is_object = True if payload.get('type') in OBJECTS else False
|
||||
if payload.get('type') == "Delete":
|
||||
cls = ActivitypubRetraction
|
||||
|
|
|
|||
|
|
@ -0,0 +1,496 @@
|
|||
from calamus import fields
|
||||
from calamus.schema import JsonLDSchema
|
||||
from calamus.utils import normalize_value
|
||||
from marshmallow import pre_load, post_load, pre_dump, post_dump
|
||||
from marshmallow.fields import Integer
|
||||
from pyld import jsonld, documentloader
|
||||
import json
|
||||
import requests_cache
|
||||
|
||||
from federation.entities.mixins import BaseEntity
|
||||
from federation.entities.activitypub.entities import ActivitypubAccept, ActivitypubPost, ActivitypubComment, ActivitypubProfile, ActivitypubImage, ActivitypubFollow
|
||||
|
||||
from pprint import pprint
|
||||
|
||||
# This is required to workaround a bug in pyld that has the Accept header
|
||||
# accept other content types. From what I understand, precedence handling
|
||||
# is broken
|
||||
def myloader(*args, **kwargs):
|
||||
requests_cache.install_cache('ld_cache', backend='redis') # this will require some configuration mechanism
|
||||
requests_loader = documentloader.requests.requests_document_loader(*args, **kwargs)
|
||||
|
||||
def loader(url, options={}):
|
||||
options['headers']['Accept'] = 'application/ld+json'
|
||||
return requests_loader(url, options)
|
||||
|
||||
return loader
|
||||
jsonld.set_document_loader(myloader())
|
||||
|
||||
|
||||
# Not sure how exhaustive this needs to be...
|
||||
as2 = fields.Namespace("https://www.w3.org/ns/activitystreams#")
|
||||
toot = fields.Namespace("http://joinmastodon.org/ns#")
|
||||
ostatus = fields.Namespace("http://ostatus.org#")
|
||||
schema = fields.Namespace("http://schema.org#")
|
||||
sec = fields.Namespace("https://w3id.org/security#")
|
||||
dc = fields.Namespace("http://purl.org/dc/terms/")
|
||||
xsd = fields.Namespace("http://www.w3.org/2001/XMLSchema#")
|
||||
ldp = fields.Namespace("http://www.w3.org/ns/ldp#")
|
||||
vcard = fields.Namespace("http://www.w3.org/2006/vcard/ns#")
|
||||
pt = fields.Namespace("https://joinpeertube.org/ns#")
|
||||
pyfed = fields.Namespace("https://docs.jasonrobinson.me/ns/python-federation#")
|
||||
diaspora = fields.Namespace("https://diasporafoundation.org/ns/")
|
||||
|
||||
|
||||
# Maybe this is food for an issue with calamus. pyld expands IRIs in an array,
|
||||
# marshmallow then barfs with an invalid string value.
|
||||
# Workaround: get rid of the array.
|
||||
class IRI(fields.IRI):
|
||||
def _deserialize(self, value, attr, data, **kwargs):
|
||||
if isinstance(value, list) and len(value) == 0: return None
|
||||
value = normalize_value(value)
|
||||
if isinstance(value, list):
|
||||
# no call to super() in list comprehensions...
|
||||
ret = []
|
||||
for val in value:
|
||||
v = super()._deserialize(val, attr, data, **kwargs)
|
||||
ret.append(v)
|
||||
return ret
|
||||
|
||||
return super()._deserialize(value, attr, data, **kwargs)
|
||||
|
||||
# calamus sets a XMLSchema#integer type, but different definitions
|
||||
# maybe used, hence the flavor property
|
||||
# TODO: handle non negative types
|
||||
class Integer(fields._JsonLDField, Integer):
|
||||
flavor = None # add fields.IRIReference type hint
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
super().__init__(*args, **kwargs)
|
||||
self.flavor = kwargs.get('flavor')
|
||||
|
||||
def _serialize(self, value, attr, obj, **kwargs):
|
||||
value = super()._serialize(value, attr, obj, **kwargs)
|
||||
flavor = str(self.flavor) if self.flavor else "http://www.w3.org/2001/XMLSchema#integer"
|
||||
if self.parent.opts.add_value_types or self.add_value_types:
|
||||
value = {"@value": value, "@type": flavor}
|
||||
return value
|
||||
|
||||
# calamus doesn't implement json-ld langage maps
|
||||
class LanguageMap(fields.Dict):
|
||||
def _serialize(self, value, attr, obj, **kwargs):
|
||||
ret = super()._serialize(value, attr, obj, **kwargs)
|
||||
if not ret: return ret
|
||||
value = []
|
||||
for k,v in ret.items():
|
||||
if k == 'orig':
|
||||
value.append({'@value':v})
|
||||
else:
|
||||
value.append({'@language': k, '@value':v})
|
||||
|
||||
return value
|
||||
|
||||
def _deserialize(self, value, attr, data, **kwargs):
|
||||
ret = {}
|
||||
for i,c in enumerate(value):
|
||||
lang = c.pop('@language', None)
|
||||
lang = '_:'+lang if lang else '_:orig'
|
||||
ret[lang] = [c]
|
||||
return super()._deserialize(ret, attr, data, **kwargs)
|
||||
|
||||
class Entity:
|
||||
def __init__(self, *args, **kwargs):
|
||||
for k, v in kwargs.items():
|
||||
setattr(self, k, v)
|
||||
|
||||
'''
|
||||
Handle cases where an AP object type matches multiple
|
||||
classes depending on the existence/value of specific
|
||||
propertie(s).
|
||||
calamus Nested field can't handle using the same model
|
||||
or the same type in multiple schemas
|
||||
'''
|
||||
def copy(self):
|
||||
if self.__class__.__name__ in ('Note', 'Page', 'Article'):
|
||||
return ActivitypubComment(**self.__dict__) if hasattr(self, 'target_id') else ActivitypubPost(**self.__dict__)
|
||||
|
||||
return self
|
||||
|
||||
class Link(Entity):
|
||||
pass
|
||||
|
||||
class Note(Entity, BaseEntity):
|
||||
pass
|
||||
|
||||
class Article(Entity, BaseEntity):
|
||||
pass
|
||||
|
||||
class Page(Entity, BaseEntity):
|
||||
pass
|
||||
|
||||
class Video(Entity):
|
||||
pass
|
||||
|
||||
class Create(Entity):
|
||||
pass
|
||||
|
||||
class Update(Entity):
|
||||
pass
|
||||
|
||||
|
||||
# A node without an id isn't true json-ld, but many payloads have
|
||||
# id-less nodes. Since calamus forces random ids on such nodes,
|
||||
# this class removes it.
|
||||
class NoIdMixin:
|
||||
def dump(self, obj):
|
||||
ret = super().dump(obj)
|
||||
ret.pop('@id', None)
|
||||
return ret
|
||||
|
||||
class ObjectMixin:
|
||||
#_children = fields.Nested(as2.attachment, nested=[PropertyValueSchema], many=True)
|
||||
#audience
|
||||
content_map = LanguageMap(as2.content) # language maps are not implemented in calamus
|
||||
#contentMap = fields.Dict(as2.contentMap, many=True)
|
||||
context = IRI(as2.context)
|
||||
guid = fields.String(diaspora.guid)
|
||||
name = fields.String(as2.name)
|
||||
#endtime
|
||||
generator = fields.Dict(as2.generator)
|
||||
#location
|
||||
#preview
|
||||
created_at = fields.DateTime(as2.published, add_value_types=True)
|
||||
replies = fields.Dict(as2.replies) # todo: define Collection schema
|
||||
startTime = fields.DateTime(as2.startTime, add_value_types=True)
|
||||
summary = fields.String(as2.summary)
|
||||
updated = fields.DateTime(as2.updated, add_value_types=True)
|
||||
#to = fields.List(as2.to, cls_or_instance=IRI(as2.to))
|
||||
to = IRI(as2.to, many=True)
|
||||
#bto
|
||||
#cc = fields.List(as2.cc, cls_or_instance=IRI(as2.cc))
|
||||
cc = IRI(as2.cc, many=True)
|
||||
#bcc
|
||||
media_type = fields.String(as2.mediaType)
|
||||
#duration
|
||||
sensitive = fields.Boolean(as2.sensitive)
|
||||
source = fields.Dict(as2.source)
|
||||
|
||||
@pre_load
|
||||
def pre_load(self, data, **kwargs):
|
||||
# add a # at the end of the python-federation string
|
||||
# for socialhome payloads
|
||||
s = json.dumps(data.get('@context'))
|
||||
if 'python-federation"' in s:
|
||||
data['@context'] = json.loads(s.replace('python-federation', 'python-federation#', 1))
|
||||
return data
|
||||
|
||||
@post_load
|
||||
def make_instance(self, data, **kwargs):
|
||||
for k, v in data.items():
|
||||
if isinstance(v, dict):
|
||||
# don't want expanded IRIs to be exposed as dict keys
|
||||
data[k] = jsonld.compact(v, self.context)
|
||||
data[k].pop('@context')
|
||||
data['schema'] = self
|
||||
return super().make_instance(data, **kwargs)
|
||||
|
||||
# This mimics that federation currently handles AP Document as AP Image
|
||||
# May need to be exanded
|
||||
class DocumentMixin(ObjectMixin):
|
||||
inline = fields.Boolean(pyfed.inlineImage)
|
||||
height = Integer(as2.height, flavor=xsd.nonNegativeInteger, add_value_types=True)
|
||||
width = Integer(as2.width, flavor=xsd.nonNegativeInteger, add_value_types=True)
|
||||
url = IRI(as2.url)
|
||||
blurhash = fields.String(toot.blurhash)
|
||||
|
||||
class DocumentSchema(DocumentMixin, NoIdMixin, JsonLDSchema):
|
||||
|
||||
class Meta:
|
||||
#fields = ('inline', 'url', 'media_type', 'name')
|
||||
unknown = 'INCLUDE'
|
||||
rdf_type = as2.Document
|
||||
model = ActivitypubImage
|
||||
|
||||
class ImageSchema(DocumentMixin, NoIdMixin, JsonLDSchema):
|
||||
|
||||
class Meta:
|
||||
fields = ('inline', 'url', 'media_type', 'name')
|
||||
unknown = 'INCLUDE'
|
||||
rdf_type = as2.Image
|
||||
model = ActivitypubImage
|
||||
|
||||
class Infohash(Entity):
|
||||
pass
|
||||
|
||||
class InfohashSchema(NoIdMixin, JsonLDSchema):
|
||||
name = fields.String(as2.name)
|
||||
|
||||
class Meta:
|
||||
rdf_type = pt.Infohash
|
||||
model = Infohash
|
||||
|
||||
class LinkMixin:
|
||||
href = IRI(as2.href)
|
||||
rel = fields.List(as2.rel, cls_or_instance=fields.String(as2.rel))
|
||||
mediaType = fields.String(as2.mediaType)
|
||||
name = fields.String(as2.name)
|
||||
hrefLang = fields.String(as2.hrefLang)
|
||||
height = Integer(as2.height, flavor=xsd.nonNegativeInteger, add_value_types=True)
|
||||
width = Integer(as2.width, flavor=xsd.nonNegativeInteger, add_value_types=True)
|
||||
fps = Integer(pt.fps, flavor=schema.Number, add_value_types=True)
|
||||
size = Integer(pt.size, flavor=schema.Number, add_value_types=True)
|
||||
#preview : variable type?
|
||||
|
||||
class TaglinkSchema(LinkMixin, NoIdMixin, JsonLDSchema):
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Link
|
||||
model = Link
|
||||
|
||||
class LinkSchema(LinkMixin, NoIdMixin, JsonLDSchema):
|
||||
tag = fields.Nested(as2.tag, nested=[InfohashSchema, TaglinkSchema], many=True)
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Link
|
||||
model = Link
|
||||
|
||||
class Hashtag(Entity):
|
||||
pass
|
||||
|
||||
class HashtagSchema(LinkMixin, NoIdMixin, JsonLDSchema):
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Hashtag
|
||||
model = Hashtag
|
||||
|
||||
class Mention(Entity):
|
||||
pass
|
||||
|
||||
class MentionSchema(LinkMixin, NoIdMixin, JsonLDSchema):
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Mention
|
||||
model = Mention
|
||||
|
||||
class ObjectSchema(ObjectMixin, JsonLDSchema):
|
||||
id = fields.Id()
|
||||
icon = fields.Nested(as2.icon, nested=ImageSchema, many=True)
|
||||
image = fields.Nested(as2.image, nested=ImageSchema, many=True)
|
||||
tag_list = fields.Nested(as2.tag, nested=[HashtagSchema,MentionSchema], many=True)
|
||||
_children = fields.Nested(as2.attachment, nested=[ImageSchema, DocumentSchema], many=True)
|
||||
|
||||
class PropertyValue(Entity):
|
||||
pass
|
||||
|
||||
class PropertyValueSchema(NoIdMixin, JsonLDSchema):
|
||||
name = fields.String(as2.name)
|
||||
value = fields.String(schema.value)
|
||||
|
||||
class Meta:
|
||||
rdf_type = schema.PropertyValue
|
||||
model = PropertyValue
|
||||
|
||||
class ActorSchema(ObjectMixin, JsonLDSchema):
|
||||
attachment = fields.Nested(as2.attachment, nested=[PropertyValueSchema], many=True)
|
||||
inbox = IRI(ldp.inbox)
|
||||
outbox = IRI(as2.outbox)
|
||||
following = IRI(as2.following)
|
||||
followers = IRI(as2.followers)
|
||||
#liked is a collection
|
||||
#streams
|
||||
username = fields.String(as2.preferredUsername)
|
||||
endpoints = fields.Dict(as2.endpoints)
|
||||
#proxyUrl
|
||||
#oauthAuthorizationEndpoint
|
||||
#oauthTokenEndpoint
|
||||
#provideClientKey
|
||||
#signClientKey
|
||||
url = IRI(as2.url)
|
||||
icon = fields.Nested(as2.icon, nested=ImageSchema, many=True)
|
||||
image = fields.Nested(as2.image, nested=ImageSchema, many=True)
|
||||
tag_list = fields.Nested(as2.tag, nested=[HashtagSchema], many=True)
|
||||
|
||||
|
||||
class ProfileSchema(ActorSchema): # why isn't the as2 Profile object used by the various platforms?
|
||||
playlists = IRI(pt.playlists)
|
||||
featured = IRI(toot.featured)
|
||||
featuredTags = IRI(toot.featuredTags)
|
||||
manuallyApprovesFollowers = fields.Boolean(as2.manuallyApprovesFollowers)
|
||||
discoverable = fields.Boolean(toot.discoverable)
|
||||
devices = IRI(toot.devices)
|
||||
publicKey = fields.Dict(sec.publicKey)
|
||||
#guid = fields.String(diaspora.guid)
|
||||
handle = fields.String(diaspora.handle)
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Person
|
||||
model = ActivitypubProfile
|
||||
|
||||
class Person(Entity):
|
||||
pass
|
||||
|
||||
class PersonSchema(ActorSchema):
|
||||
class Meta:
|
||||
rdf_type = as2.Person
|
||||
model = Person
|
||||
|
||||
class Group(Entity):
|
||||
pass
|
||||
|
||||
class GroupSchema(ActorSchema):
|
||||
class Meta:
|
||||
rdf_type = as2.Group
|
||||
model = Group
|
||||
|
||||
class NoteMixin:
|
||||
actor_id = IRI(as2.attributedTo, many=True)
|
||||
target_id = IRI(as2.inReplyTo)
|
||||
atom_url = IRI(ostatus.atomUri)
|
||||
conversation = fields.String(ostatus.conversation)
|
||||
inReplyToAtomUri = IRI(ostatus.inReplyToAtomUri)
|
||||
url = IRI(as2.url)
|
||||
|
||||
class NoteSchema(NoteMixin, ObjectSchema):
|
||||
class Meta:
|
||||
rdf_type = as2.Note
|
||||
model = Note
|
||||
|
||||
class PageSchema(NoteMixin, ObjectSchema):
|
||||
class Meta:
|
||||
rdf_type = as2.Page
|
||||
model = Page
|
||||
|
||||
class ArticleSchema(NoteMixin, ObjectSchema):
|
||||
class Meta:
|
||||
rdf_type = as2.Article
|
||||
model = Article
|
||||
|
||||
# peertube uses a lot of properties differently...
|
||||
class VideoSchema(ObjectSchema):
|
||||
urls = fields.Nested(as2.url, nested=LinkSchema, many=True)
|
||||
actor_id = fields.Nested(as2.attributedTo, nested=[PersonSchema, GroupSchema], many=True)
|
||||
|
||||
class Meta:
|
||||
unknown = 'EXCLUDE' # required until all the pt fields are defined
|
||||
rdf_type = as2.Video
|
||||
model = Video
|
||||
|
||||
class Signature(Entity):
|
||||
pass
|
||||
|
||||
class SignatureSchema(NoIdMixin, JsonLDSchema):
|
||||
created = fields.DateTime(dc.created, add_value_types=True)
|
||||
creator = IRI(dc.creator)
|
||||
key = fields.String(sec.signatureValue)
|
||||
nonce = fields.String(sec.nonce)
|
||||
|
||||
class Meta:
|
||||
rdf_type = sec.RsaSignature2017
|
||||
model = Signature
|
||||
|
||||
class ActivityMixin(ObjectMixin):
|
||||
actor_id = IRI(as2.actor)
|
||||
#object will be defined in pre_load
|
||||
#target_id = IRI(as2.target)
|
||||
#result
|
||||
#origin
|
||||
instrument = fields.Dict(as2.instrument)
|
||||
signature = fields.Nested(sec.signature, nested = SignatureSchema)
|
||||
|
||||
@pre_load
|
||||
def pre_load(self, data, **kwargs):
|
||||
data = super().pre_load(data, **kwargs)
|
||||
|
||||
# AP activities may be signed, but some platforms don't
|
||||
# define RsaSignature2017. add it to the context
|
||||
ctx = data.get('@context')
|
||||
if ctx:
|
||||
w3id = 'https://w3id.org/security/v1'
|
||||
if w3id not in ctx: ctx.insert(0,w3id)
|
||||
idx = [i for i,v in enumerate(ctx) if isinstance(v, dict)]
|
||||
found = False
|
||||
for i in idx:
|
||||
if ctx[i].get('RsaSignature2017'):
|
||||
found = True
|
||||
break
|
||||
if not found: ctx[idx[0]]['RsaSignature2017'] = 'sec:RsaSignature2017'
|
||||
self.context = data['@context'] = ctx
|
||||
|
||||
return data
|
||||
|
||||
class FollowSchema(ActivityMixin, JsonLDSchema):
|
||||
activity_id = fields.Id()
|
||||
target_id = IRI(as2.object)
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Follow
|
||||
model = ActivitypubFollow
|
||||
|
||||
OBJECTS = [
|
||||
ArticleSchema,
|
||||
FollowSchema,
|
||||
# "Like": LikeSchema
|
||||
# "View": ViewSchema
|
||||
NoteSchema,
|
||||
PageSchema,
|
||||
# "Tombstone": TombstoneSchema
|
||||
VideoSchema
|
||||
]
|
||||
|
||||
class ActivitySchema(ActivityMixin, JsonLDSchema):
|
||||
object_ = fields.Nested(as2.object, nested=OBJECTS)
|
||||
|
||||
|
||||
class AcceptSchema(ActivitySchema):
|
||||
target_id = fields.Id()
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Accept
|
||||
model = ActivitypubAccept
|
||||
|
||||
class CreateSchema(ActivitySchema):
|
||||
activity_id = fields.Id()
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Create
|
||||
model = Create
|
||||
|
||||
class UpdateSchema(ActivitySchema):
|
||||
activity_id = fields.Id()
|
||||
|
||||
class Meta:
|
||||
rdf_type = as2.Update
|
||||
model = Update
|
||||
|
||||
SCHEMAMAP = {
|
||||
"Accept": AcceptSchema,
|
||||
# "Announce": AnnounceSchema
|
||||
"Article": ArticleSchema,
|
||||
"Create": CreateSchema,
|
||||
# "Delete": DeleteSchema
|
||||
"Follow": FollowSchema,
|
||||
# "Like": LikeSchema
|
||||
"Note": NoteSchema,
|
||||
"Page": PageSchema,
|
||||
"Person": ProfileSchema,
|
||||
# "Tombstone": TombstoneSchema
|
||||
# "Undo": UndoSchema
|
||||
"Update": UpdateSchema,
|
||||
# "View": ViewSchema
|
||||
}
|
||||
|
||||
def schema_to_objects(payload):
|
||||
entity = None
|
||||
schema = SCHEMAMAP.get(payload['type'])
|
||||
if schema:
|
||||
schema_instance = schema(context=payload['@context'])
|
||||
entity = schema_instance.load(payload)
|
||||
|
||||
if hasattr(entity, 'object_') and isinstance(entity.object_, BaseEntity):
|
||||
entity.object_.activity = entity
|
||||
entity = entity.object_.copy() if hasattr(entity.object_, 'copy') else entity.object_
|
||||
elif not isinstance(entity, BaseEntity):
|
||||
# payload not supported yet
|
||||
entity = None
|
||||
|
||||
return entity
|
||||
|
|
@ -1,4 +1,5 @@
|
|||
from typing import Dict, Tuple
|
||||
from mimetypes import guess_type
|
||||
|
||||
from dirty_validators.basic import Email
|
||||
|
||||
|
|
@ -43,7 +44,7 @@ class Image(OptionalRawContentMixin, CreatedAtMixin, BaseEntity):
|
|||
self.media_type = self.get_media_type()
|
||||
|
||||
def get_media_type(self) -> str:
|
||||
media_type = fetch_content_type(self.url)
|
||||
media_type = guess_type(self.url)[0] or fetch_content_type(self.url)
|
||||
if media_type in self._valid_media_types:
|
||||
return media_type
|
||||
return ""
|
||||
|
|
|
|||
|
|
@ -37,13 +37,20 @@ class BaseEntity:
|
|||
self._children = []
|
||||
self._mentions = set()
|
||||
self._receivers = []
|
||||
for key, value in kwargs.items():
|
||||
if hasattr(self, key):
|
||||
|
||||
# make the assumption that if a schema is being used, the payload
|
||||
# is deserialized and validated properly
|
||||
if kwargs.get('schema'):
|
||||
for key, value in kwargs.items():
|
||||
setattr(self, key, value)
|
||||
else:
|
||||
warnings.warn("%s.__init__ got parameter %s which this class does not support - ignoring." % (
|
||||
self.__class__.__name__, key
|
||||
))
|
||||
else:
|
||||
for key, value in kwargs.items():
|
||||
if hasattr(self, key):
|
||||
setattr(self, key, value)
|
||||
else:
|
||||
warnings.warn("%s.__init__ got parameter %s which this class does not support - ignoring." % (
|
||||
self.__class__.__name__, key
|
||||
))
|
||||
if not self.activity:
|
||||
# Fill a default activity if not given and type of entity class has one
|
||||
self.activity = getattr(self, "_default_activity", None)
|
||||
|
|
|
|||
Ładowanie…
Reference in New Issue