From 0837caeadee57cb5c74a3a91b6e2438ef3f3a245 Mon Sep 17 00:00:00 2001
From: Max Nordlund <max.nordlund@gmail.com>
Date: Mon, 9 Mar 2020 21:29:02 +0100
Subject: [PATCH] Add support for ptpython config and history files

This uses the official/upstream way of finding the config and history file,
since ~/.ptpython/config.py is deprecated. It also uses the global ones,
though it can be easily extended to support per project config and history.
But that's for a rainy day.

Fixes #7 and #13.
---
 flask_shell_ptpython.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/flask_shell_ptpython.py b/flask_shell_ptpython.py
index cc17f5c..15b5096 100644
--- a/flask_shell_ptpython.py
+++ b/flask_shell_ptpython.py
@@ -20,7 +20,8 @@ def shell_command():
     without having to manually configure the application.
     """
     from flask.globals import _app_ctx_stack
-    from ptpython.repl import embed
+    from ptpython.repl import embed, run_config
+    from ptpython.entry_points.run_ptpython import create_parser, get_config_and_history_file
 
     app = _app_ctx_stack.top.app
     ctx = {}
@@ -34,4 +35,15 @@ def shell_command():
 
     ctx.update(app.make_shell_context())
 
-    embed(globals=ctx)
+    config_file, history_filename = get_config_and_history_file(
+        create_parser().parse_args([])
+    )
+
+    def configure(repl):
+        run_config(repl, config_file=config_file)
+
+    embed(
+        globals=ctx,
+        history_filename=history_filename,
+        configure=configure
+    )