dev: add models to dev environment (#452)

Ballsdex-Team · Oct 28, 2024 · 9d06f31 · 9d06f31
1 parent 6a9ca6d
commit 9d06f31
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/ballsdex/core/dev.py b/ballsdex/core/dev.py
@@ -21,8 +21,17 @@
     BallInstance,
     BlacklistedGuild,
     BlacklistedID,
+    BlacklistHistory,
+    Block,
+    DonationPolicy,
+    Economy,
+    FriendPolicy,
+    Friendship,
     GuildConfig,
+    MentionPolicy,
     Player,
+    PrivacyPolicy,
+    Regime,
     Special,
     Trade,
     TradeObject,
@@ -230,6 +239,15 @@ def get_environment(self, ctx: commands.Context) -> dict:
             "Special": Special,
             "Trade": Trade,
             "TradeObject": TradeObject,
+            "Regime": Regime,
+            "Economy": Economy,
+            "DonationPolicy": DonationPolicy,
+            "PrivacyPolicy": PrivacyPolicy,
+            "MentionPolicy": MentionPolicy,
+            "FriendPolicy": FriendPolicy,
+            "BlacklistHistory": BlacklistHistory,
+            "Friendship": Friendship,
+            "Block": Block,
             "text_to_file": text_to_file,
             "_": self._last_result,
             "__name__": "__main__",