Esempio n. 1
0
def serialize_post(post, auth_user, options=None):
    return util.serialize_entity(
        post,
        {
            'id': lambda: post.post_id,
            'version': lambda: post.version,
            'creationTime': lambda: post.creation_time,
            'lastEditTime': lambda: post.last_edit_time,
            'safety': lambda: SAFETY_MAP[post.safety],
            'source': lambda: post.source,
            'type': lambda: TYPE_MAP[post.type],
            'mimeType': lambda: post.mime_type,
            'checksum': lambda: post.checksum,
            'fileSize': lambda: post.file_size,
            'canvasWidth': lambda: post.canvas_width,
            'canvasHeight': lambda: post.canvas_height,
            'contentUrl': lambda: get_post_content_url(post),
            'thumbnailUrl': lambda: get_post_thumbnail_url(post),
            'flags': lambda: post.flags,
            'tags': lambda: [
                tag.names[0].name for tag in tags.sort_tags(post.tags)],
            'relations': lambda: sorted(
                {
                    post['id']:
                        post for post in [
                            serialize_micro_post(rel, auth_user)
                            for rel in post.relations]
                }.values(),
                key=lambda post: post['id']),
            'user': lambda: users.serialize_micro_user(post.user, auth_user),
            'score': lambda: post.score,
            'ownScore': lambda: scores.get_score(post, auth_user),
            'ownFavorite': lambda: len([
                user for user in post.favorited_by
                if user.user_id == auth_user.user_id]
            ) > 0,
            'tagCount': lambda: post.tag_count,
            'favoriteCount': lambda: post.favorite_count,
            'commentCount': lambda: post.comment_count,
            'noteCount': lambda: post.note_count,
            'relationCount': lambda: post.relation_count,
            'featureCount': lambda: post.feature_count,
            'lastFeatureTime': lambda: post.last_feature_time,
            'favoritedBy': lambda: [
                users.serialize_micro_user(rel.user, auth_user)
                for rel in post.favorited_by
            ],
            'hasCustomThumbnail':
                lambda: files.has(get_post_thumbnail_backup_path(post)),
            'notes': lambda: sorted(
                [serialize_note(note) for note in post.notes],
                key=lambda x: x['polygon']),
            'comments': lambda: [
                comments.serialize_comment(comment, auth_user)
                for comment in sorted(
                    post.comments,
                    key=lambda comment: comment.creation_time)],
        },
        options)
Esempio n. 2
0
def serialize_snapshot(snapshot, auth_user):
    assert snapshot
    return {
        'operation': snapshot.operation,
        'type': snapshot.resource_type,
        'id': snapshot.resource_name,
        'user': users.serialize_micro_user(snapshot.user, auth_user),
        'data': snapshot.data,
        'time': snapshot.creation_time,
    }
Esempio n. 3
0
def test_serialize_micro_user(user_factory):
    with patch("szurubooru.func.users.get_avatar_url"):
        users.get_avatar_url.return_value = "https://example.com/avatar.png"
        auth_user = user_factory()
        user = user_factory(name="dummy user")
        db.session.add(user)
        db.session.flush()
        assert users.serialize_micro_user(user, auth_user) == {
            "name": "dummy user",
            "avatarUrl": "https://example.com/avatar.png",
        }
Esempio n. 4
0
def test_serialize_micro_user(user_factory):
    with patch('szurubooru.func.users.get_avatar_url'):
        users.get_avatar_url.return_value = 'https://example.com/avatar.png'
        auth_user = user_factory()
        user = user_factory(name='dummy user')
        db.session.add(user)
        db.session.flush()
        assert users.serialize_micro_user(user, auth_user) == {
            'name': 'dummy user',
            'avatarUrl': 'https://example.com/avatar.png',
        }
Esempio n. 5
0
def serialize_snapshot(snapshot: model.Snapshot,
                       auth_user: model.User) -> Dict[str, Any]:
    assert snapshot
    return {
        'operation': snapshot.operation,
        'type': snapshot.resource_type,
        'id': snapshot.resource_name,
        'user': users.serialize_micro_user(snapshot.user, auth_user),
        'data': snapshot.data,
        'time': snapshot.creation_time,
    }
Esempio n. 6
0
def test_serialize_micro_user(user_factory):
    with patch('szurubooru.func.users.get_avatar_url'):
        users.get_avatar_url.return_value = 'https://example.com/avatar.png'
        auth_user = user_factory()
        user = user_factory(name='dummy user')
        db.session.add(user)
        db.session.flush()
        assert users.serialize_micro_user(user, auth_user) == {
            'name': 'dummy user',
            'avatarUrl': 'https://example.com/avatar.png',
        }
Esempio n. 7
0
def serialize_snapshot(snapshot: model.Snapshot,
                       auth_user: model.User) -> Dict[str, Any]:
    assert snapshot
    return {
        "operation": snapshot.operation,
        "type": snapshot.resource_type,
        "id": snapshot.resource_name,
        "user": users.serialize_micro_user(snapshot.user, auth_user),
        "data": snapshot.data,
        "time": snapshot.creation_time,
    }
Esempio n. 8
0
def serialize_comment(comment, auth_user, options=None):
    return util.serialize_entity(
        comment,
        {
            'id': lambda: comment.comment_id,
            'user':
                lambda: users.serialize_micro_user(comment.user, auth_user),
            'postId': lambda: comment.post.post_id,
            'version': lambda: comment.version,
            'text': lambda: comment.text,
            'creationTime': lambda: comment.creation_time,
            'lastEditTime': lambda: comment.last_edit_time,
            'score': lambda: comment.score,
            'ownScore': lambda: scores.get_score(comment, auth_user),
        },
        options)
Esempio n. 9
0
 def serialize_favorited_by(self) -> Any:
     return [
         users.serialize_micro_user(rel.user, self.auth_user)
         for rel in self.post.favorited_by
     ]
Esempio n. 10
0
 def serialize_user(self) -> Any:
     return users.serialize_micro_user(self.post.user, self.auth_user)
Esempio n. 11
0
 def serialize_user(self) -> Any:
     return users.serialize_micro_user(self.comment.user, self.auth_user)
Esempio n. 12
0
 def serialize_user(self) -> Any:
     return users.serialize_micro_user(self.user_token.user, self.auth_user)
Esempio n. 13
0
 def serialize_favorited_by(self) -> Any:
     return [
         users.serialize_micro_user(rel.user, self.auth_user)
         for rel in self.post.favorited_by
     ]