Skip to content
This repository was archived by the owner on Dec 20, 2021. It is now read-only.

Commit 07a07b8

Browse files
committed
Use FavoriteType enum
1 parent d303fd7 commit 07a07b8

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

vrcpy/favorite.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def __init__(self, client, obj, loop=None):
1616
},
1717
"type": {
1818
"dict_key": "type",
19-
"type": str
19+
"type": FavoriteType
2020
}
2121
})
2222

@@ -25,14 +25,14 @@ def __init__(self, client, obj, loop=None):
2525
@staticmethod
2626
def build_favorite(client, obj, loop=None):
2727
switch = {
28-
"world": WorldFavorite,
29-
"friend": FriendFavorite,
30-
"avatar": AvatarFavorite
28+
FavoriteType.WORLD: WorldFavorite,
29+
FavoriteType.FRIEND: FriendFavorite,
30+
FavoriteType.AVATAR: AvatarFavorite
3131
}
3232

3333
logging.debug("Building favorite of type " + obj["type"])
3434

35-
return switch[obj["type"]](client, obj, loop)
35+
return switch[FavoriteType(obj["type"].upper())](client, obj, loop)
3636

3737
async def unfavorite(self):
3838
"""Unfavorites the favorite object"""
@@ -75,6 +75,7 @@ def __init__(self, client, obj, loop=None):
7575

7676
del self.unfavorite
7777
self._assign(obj)
78+
self.favorites = []
7879

7980
@staticmethod
8081
def build_favorite_group(client, obj, loop=None):

0 commit comments

Comments
 (0)