From c64ed43215f50c90104da110dcda21ed1683c85d Mon Sep 17 00:00:00 2001
From: Masen Furer <m_github@0x26.net>
Date: Fri, 24 Jan 2025 13:40:59 -0800
Subject: [PATCH] Move export checking logic to reflex CLI

---
 reflex/reflex.py       | 8 ++++++++
 reflex/utils/export.py | 7 -------
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/reflex/reflex.py b/reflex/reflex.py
index 249b4848c..a74024201 100644
--- a/reflex/reflex.py
+++ b/reflex/reflex.py
@@ -328,6 +328,14 @@ def export(
     if prerequisites.needs_reinit(frontend=True):
         _init(name=config.app_name, loglevel=loglevel)
 
+    if frontend and not config.show_built_with_reflex:
+        # The sticky badge may be disabled on export for team/enterprise tiers.
+        prerequisites.check_config_option_in_tier(
+            option_name="show_built_with_reflex",
+            allowed_tiers=["team", "enterprise"],
+            fallback_value=False,
+        )
+
     export_utils.export(
         zipping=zipping,
         frontend=frontend,
diff --git a/reflex/utils/export.py b/reflex/utils/export.py
index 7fc38190e..edb4a6e1a 100644
--- a/reflex/utils/export.py
+++ b/reflex/utils/export.py
@@ -55,13 +55,6 @@ def export(
     console.rule("[bold]Compiling production app and preparing for export.")
 
     if frontend:
-        if not config.show_built_with_reflex:
-            # The sticky badge may be disabled on export for team/enterprise tiers.
-            prerequisites.check_config_option_in_tier(
-                option_name="show_built_with_reflex",
-                allowed_tiers=["team", "enterprise"],
-                fallback_value=False,
-            )
         # Ensure module can be imported and app.compile() is called.
         prerequisites.get_compiled_app(export=True)
         # Set up .web directory and install frontend dependencies.