Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add example with rich text; fix generation of system type field #13

Merged
merged 1 commit into from
May 12, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion atproto/codegen/models/generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -319,8 +319,11 @@ def _generate_def_model(nsid: NSID, def_name: str, def_model: models.LexObject,
_get_model(nsid, def_model),
]

def_type = f'{nsid}#{def_name}'
if def_name == 'main':
lines.append(f"{_(1)}_type: str = '{nsid}'")
def_type = str(nsid)

lines.append(f"{_(1)}_type: str = '{def_type}'")

lines.append('')

Expand Down
8 changes: 8 additions & 0 deletions atproto/xrpc_client/models/app/bsky/actor/defs.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,8 @@ class ProfileViewBasic(base.ModelBase):
labels: Optional[List['models.ComAtprotoLabelDefs.Label']] = None
viewer: Optional['models.AppBskyActorDefs.ViewerState'] = None

_type: str = 'app.bsky.actor.defs#profileViewBasic'


@dataclass
class ProfileView(base.ModelBase):
Expand All @@ -59,6 +61,8 @@ class ProfileView(base.ModelBase):
labels: Optional[List['models.ComAtprotoLabelDefs.Label']] = None
viewer: Optional['models.AppBskyActorDefs.ViewerState'] = None

_type: str = 'app.bsky.actor.defs#profileView'


@dataclass
class ProfileViewDetailed(base.ModelBase):
Expand Down Expand Up @@ -93,6 +97,8 @@ class ProfileViewDetailed(base.ModelBase):
postsCount: Optional[int] = None
viewer: Optional['models.AppBskyActorDefs.ViewerState'] = None

_type: str = 'app.bsky.actor.defs#profileViewDetailed'


@dataclass
class ViewerState(base.ModelBase):
Expand All @@ -112,3 +118,5 @@ class ViewerState(base.ModelBase):
followedBy: Optional[str] = None
following: Optional[str] = None
muted: Optional[bool] = None

_type: str = 'app.bsky.actor.defs#viewerState'
6 changes: 6 additions & 0 deletions atproto/xrpc_client/models/app/bsky/embed/external.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,8 @@ class External(base.ModelBase):
uri: str
thumb: Optional[BlobRef] = None

_type: str = 'app.bsky.embed.external#external'


@dataclass
class View(base.ModelBase):
Expand All @@ -56,6 +58,8 @@ class View(base.ModelBase):

external: 'models.AppBskyEmbedExternal.ViewExternal'

_type: str = 'app.bsky.embed.external#view'


@dataclass
class ViewExternal(base.ModelBase):
Expand All @@ -73,3 +77,5 @@ class ViewExternal(base.ModelBase):
title: str
uri: str
thumb: Optional[str] = None

_type: str = 'app.bsky.embed.external#viewExternal'
6 changes: 6 additions & 0 deletions atproto/xrpc_client/models/app/bsky/embed/images.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,8 @@ class Image(base.ModelBase):
alt: str
image: BlobRef

_type: str = 'app.bsky.embed.images#image'


@dataclass
class View(base.ModelBase):
Expand All @@ -52,6 +54,8 @@ class View(base.ModelBase):

images: List['models.AppBskyEmbedImages.ViewImage']

_type: str = 'app.bsky.embed.images#view'


@dataclass
class ViewImage(base.ModelBase):
Expand All @@ -67,3 +71,5 @@ class ViewImage(base.ModelBase):
alt: str
fullsize: str
thumb: str

_type: str = 'app.bsky.embed.images#viewImage'
8 changes: 8 additions & 0 deletions atproto/xrpc_client/models/app/bsky/embed/record.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,8 @@ class View(base.ModelBase):
'Dict[str, Any]',
]

_type: str = 'app.bsky.embed.record#view'


@dataclass
class ViewRecord(base.ModelBase):
Expand Down Expand Up @@ -76,6 +78,8 @@ class ViewRecord(base.ModelBase):
] = None
labels: Optional[List['models.ComAtprotoLabelDefs.Label']] = None

_type: str = 'app.bsky.embed.record#viewRecord'


@dataclass
class ViewNotFound(base.ModelBase):
Expand All @@ -88,6 +92,8 @@ class ViewNotFound(base.ModelBase):

uri: str

_type: str = 'app.bsky.embed.record#viewNotFound'


@dataclass
class ViewBlocked(base.ModelBase):
Expand All @@ -99,3 +105,5 @@ class ViewBlocked(base.ModelBase):
"""

uri: str

_type: str = 'app.bsky.embed.record#viewBlocked'
Original file line number Diff line number Diff line change
Expand Up @@ -40,3 +40,5 @@ class View(base.ModelBase):

media: Union['models.AppBskyEmbedImages.View', 'models.AppBskyEmbedExternal.View', 'Dict[str, Any]']
record: 'models.AppBskyEmbedRecord.View'

_type: str = 'app.bsky.embed.recordWithMedia#view'
16 changes: 16 additions & 0 deletions atproto/xrpc_client/models/app/bsky/feed/defs.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,8 @@ class PostView(base.ModelBase):
repostCount: Optional[int] = None
viewer: Optional['models.AppBskyFeedDefs.ViewerState'] = None

_type: str = 'app.bsky.feed.defs#postView'


@dataclass
class ViewerState(base.ModelBase):
Expand All @@ -65,6 +67,8 @@ class ViewerState(base.ModelBase):
like: Optional[str] = None
repost: Optional[str] = None

_type: str = 'app.bsky.feed.defs#viewerState'


@dataclass
class FeedViewPost(base.ModelBase):
Expand All @@ -81,6 +85,8 @@ class FeedViewPost(base.ModelBase):
reason: Optional[Union['models.AppBskyFeedDefs.ReasonRepost', 'Dict[str, Any]']] = None
reply: Optional['models.AppBskyFeedDefs.ReplyRef'] = None

_type: str = 'app.bsky.feed.defs#feedViewPost'


@dataclass
class ReplyRef(base.ModelBase):
Expand All @@ -95,6 +101,8 @@ class ReplyRef(base.ModelBase):
parent: 'models.AppBskyFeedDefs.PostView'
root: 'models.AppBskyFeedDefs.PostView'

_type: str = 'app.bsky.feed.defs#replyRef'


@dataclass
class ReasonRepost(base.ModelBase):
Expand All @@ -109,6 +117,8 @@ class ReasonRepost(base.ModelBase):
by: 'models.AppBskyActorDefs.ProfileViewBasic'
indexedAt: str

_type: str = 'app.bsky.feed.defs#reasonRepost'


@dataclass
class ThreadViewPost(base.ModelBase):
Expand Down Expand Up @@ -141,6 +151,8 @@ class ThreadViewPost(base.ModelBase):
]
] = None

_type: str = 'app.bsky.feed.defs#threadViewPost'


@dataclass
class NotFoundPost(base.ModelBase):
Expand All @@ -155,6 +167,8 @@ class NotFoundPost(base.ModelBase):
notFound: bool
uri: str

_type: str = 'app.bsky.feed.defs#notFoundPost'


@dataclass
class BlockedPost(base.ModelBase):
Expand All @@ -168,3 +182,5 @@ class BlockedPost(base.ModelBase):

blocked: bool
uri: str

_type: str = 'app.bsky.feed.defs#blockedPost'
2 changes: 2 additions & 0 deletions atproto/xrpc_client/models/app/bsky/feed/get_likes.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,3 +62,5 @@ class Like(base.ModelBase):
actor: 'models.AppBskyActorDefs.ProfileView'
createdAt: str
indexedAt: str

_type: str = 'app.bsky.feed.getLikes#like'
6 changes: 6 additions & 0 deletions atproto/xrpc_client/models/app/bsky/feed/post.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,8 @@ class ReplyRef(base.ModelBase):
parent: 'models.ComAtprotoRepoStrongRef.Main'
root: 'models.ComAtprotoRepoStrongRef.Main'

_type: str = 'app.bsky.feed.post#replyRef'


@dataclass
class Entity(base.ModelBase):
Expand All @@ -41,6 +43,8 @@ class Entity(base.ModelBase):
type: str
value: str

_type: str = 'app.bsky.feed.post#entity'


@dataclass
class TextSlice(base.ModelBase):
Expand All @@ -55,6 +59,8 @@ class TextSlice(base.ModelBase):
end: int
start: int

_type: str = 'app.bsky.feed.post#textSlice'


@dataclass
class Main(base.RecordModelBase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -68,3 +68,5 @@ class Notification(base.ModelBase):
uri: str
labels: Optional[List['models.ComAtprotoLabelDefs.Label']] = None
reasonSubject: Optional[str] = None

_type: str = 'app.bsky.notification.listNotifications#notification'
6 changes: 6 additions & 0 deletions atproto/xrpc_client/models/app/bsky/richtext/facet.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,8 @@ class Mention(base.ModelBase):

did: str

_type: str = 'app.bsky.richtext.facet#mention'


@dataclass
class Link(base.ModelBase):
Expand All @@ -51,6 +53,8 @@ class Link(base.ModelBase):

uri: str

_type: str = 'app.bsky.richtext.facet#link'


@dataclass
class ByteSlice(base.ModelBase):
Expand All @@ -64,3 +68,5 @@ class ByteSlice(base.ModelBase):

byteEnd: int
byteStart: int

_type: str = 'app.bsky.richtext.facet#byteSlice'
Loading