From 64dd28fe2430312094df38380080f58d5a17e2f4 Mon Sep 17 00:00:00 2001
From: David Lord <davidism@gmail.com>
Date: Thu, 28 Apr 2022 11:08:17 -0700
Subject: [PATCH] work

---
 src/flask/cli.py  | 40 ++++++++++++++++++++++++++++++++++++----
 src/flask/cli2.py | 17 +++++++++++++++++
 2 files changed, 53 insertions(+), 4 deletions(-)
 create mode 100644 src/flask/cli2.py

diff --git a/src/flask/cli.py b/src/flask/cli.py
index 36c4f1b6..c60619f5 100644
--- a/src/flask/cli.py
+++ b/src/flask/cli.py
@@ -358,7 +358,7 @@ class ScriptInfo:
 
     def __init__(self, app_import_path=None, create_app=None, set_debug_flag=True):
         #: Optionally the import path for the Flask application.
-        self.app_import_path = app_import_path or os.environ.get("FLASK_APP")
+        self.app_import_path = app_import_path
         #: Optionally a function that is passed the script info to create
         #: the instance of the application.
         self.create_app = create_app
@@ -423,8 +423,11 @@ def with_appcontext(f):
 
     @click.pass_context
     def decorator(__ctx, *args, **kwargs):
-        with __ctx.ensure_object(ScriptInfo).load_app().app_context():
-            return __ctx.invoke(f, *args, **kwargs)
+        if not current_app:
+            app = __ctx.ensure_object(ScriptInfo).load_app()
+            __ctx.with_resource(app.app_context())
+
+        return __ctx.invoke(f, *args, **kwargs)
 
     return update_wrapper(decorator, f)
 
@@ -497,7 +500,36 @@ class FlaskGroup(AppGroup):
         if add_version_option:
             params.append(version_option)
 
-        AppGroup.__init__(self, params=params, **extra)
+        def set_app(ctx, param, value):
+            info: ScriptInfo = ctx.ensure_object(ScriptInfo)
+            info.app_import_path = value
+
+        def set_env(ctx, param, value):
+            info: ScriptInfo = ctx.ensure_object(ScriptInfo)
+            info.app_env = value
+
+        params.append(
+            click.Option(
+                ["--app"],
+                metavar="IMPORT",
+                help="Import string for a Flask app object or factory function.",
+                callback=set_app,
+                is_eager=True,
+            )
+        )
+        params.append(
+            click.Option(
+                ["--env"],
+                metavar="ENV",
+                help="Runtime environment name.",
+                default="production",
+                show_default=True,
+                callback=set_env,
+                is_eager=True,
+            )
+        )
+        super().__init__(params=params, **extra)
+
         self.create_app = create_app
         self.load_dotenv = load_dotenv
         self.set_debug_flag = set_debug_flag
diff --git a/src/flask/cli2.py b/src/flask/cli2.py
new file mode 100644
index 00000000..3f337412
--- /dev/null
+++ b/src/flask/cli2.py
@@ -0,0 +1,17 @@
+import os
+
+import click
+
+
+@click.group("flask")
+@click.option("--app", "app")
+@click.option("--env")
+@click.pass_context
+def flask_group(ctx: click.Context, app: str, env: str) -> None:
+    os.environ["FLASK_RUN_FROM_CLI"] = "true"
+    # info = ctx.ensure_object(ScriptInfo)
+
+
+@flask_group.command("run")
+def run_command():
+    click.echo("run")