From 284e58a90cd69a164438a3663830aa460b84cbcd Mon Sep 17 00:00:00 2001
From: Khaleel Al-Adhami <khaleel.aladhami@gmail.com>
Date: Tue, 21 Jan 2025 15:41:29 -0800
Subject: [PATCH] reflex export environment should default to prod

---
 reflex/reflex.py       | 4 ++++
 reflex/utils/export.py | 7 ++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/reflex/reflex.py b/reflex/reflex.py
index 2d6ebc30c..dc31f177e 100644
--- a/reflex/reflex.py
+++ b/reflex/reflex.py
@@ -306,6 +306,9 @@ def export(
         help="Whether to exclude sqlite db files when exporting backend.",
         hidden=True,
     ),
+    env: constants.Env = typer.Option(
+        constants.Env.PROD, help="The environment to run the app in."
+    ),
     loglevel: constants.LogLevel = typer.Option(
         config.loglevel, help="The log level to use."
     ),
@@ -323,6 +326,7 @@ def export(
         backend=backend,
         zip_dest_dir=zip_dest_dir,
         upload_db_file=upload_db_file,
+        env=env,
         loglevel=loglevel.subprocess_level(),
     )
 
diff --git a/reflex/utils/export.py b/reflex/utils/export.py
index 2fbf633f6..74c1a31cd 100644
--- a/reflex/utils/export.py
+++ b/reflex/utils/export.py
@@ -4,7 +4,7 @@ from pathlib import Path
 from typing import Optional
 
 from reflex import constants
-from reflex.config import get_config
+from reflex.config import environment, get_config
 from reflex.utils import build, console, exec, prerequisites, telemetry
 
 config = get_config()
@@ -18,6 +18,7 @@ def export(
     upload_db_file: bool = False,
     api_url: Optional[str] = None,
     deploy_url: Optional[str] = None,
+    env: constants.Env = constants.Env.PROD,
     loglevel: constants.LogLevel = console._LOG_LEVEL,
 ):
     """Export the app to a zip file.
@@ -30,11 +31,15 @@ def export(
         upload_db_file: Whether to upload the database file. Defaults to False.
         api_url: The API URL to use. Defaults to None.
         deploy_url: The deploy URL to use. Defaults to None.
+        env: The environment to use. Defaults to constants.Env.PROD.
         loglevel: The log level to use. Defaults to console._LOG_LEVEL.
     """
     # Set the log level.
     console.set_log_level(loglevel)
 
+    # Set env mode in the environment
+    environment.REFLEX_ENV_MODE.set(constants.Env.PROD)
+
     # Override the config url values if provided.
     if api_url is not None:
         config.api_url = str(api_url)