From c1cdbdb24332b18c1e4eb844306fd2d94ed829dd Mon Sep 17 00:00:00 2001
From: "[A" <quantfreedom1022@gmail.com>
Date: Sun, 5 May 2024 10:34:53 -0700
Subject: [PATCH] add clear cache

---
 clear_cache.py                               | 29 ++++++++++++++++++--
 tests/backtest/vol_backtest/long_backtest.py |  2 +-
 tests/backtest/vol_backtest/testing.ipynb    |  2 +-
 3 files changed, 29 insertions(+), 4 deletions(-)

diff --git a/clear_cache.py b/clear_cache.py
index cc21ebc..a615097 100644
--- a/clear_cache.py
+++ b/clear_cache.py
@@ -1,3 +1,28 @@
-from quantfreedom.helpers.utils import clear_cache
+from pathlib import Path
+from IPython.paths import get_ipython_cache_dir
 
-clear_cache()
+
+def delete_dir(p: Path):
+    for sub in p.iterdir():
+        if sub.is_dir():
+            delete_dir(sub)
+        else:
+            sub.unlink()
+    p.rmdir()
+
+
+def clear_cache():
+    for p in Path(get_ipython_cache_dir() + "\\numba_cache").rglob("*.nb*"):
+        p.unlink()
+    for p in Path(__file__).parent.parent.rglob("numba_cache"):
+        delete_dir(p)
+    for p in Path(__file__).parent.parent.rglob("__pycache__"):
+        delete_dir(p)
+    for p in Path(__file__).parent.parent.rglob("cdk.out"):
+        delete_dir(p)
+    for p in Path(__file__).parent.parent.rglob("*.py[co]"):
+        p.unlink()
+
+
+if __name__ == "__main__":
+    clear_cache()
diff --git a/tests/backtest/vol_backtest/long_backtest.py b/tests/backtest/vol_backtest/long_backtest.py
index 570d901..4a0af57 100644
--- a/tests/backtest/vol_backtest/long_backtest.py
+++ b/tests/backtest/vol_backtest/long_backtest.py
@@ -14,7 +14,7 @@
         remove("backtest_results.h5")
 
     print("Downloading candles")
-    with open('thing.pkl', 'rb') as f:
+    with open('Volume_Candles_Tuple_mar_apr.pkl', 'rb') as f:
         candles = pickle.load(f) 
     backtest_results = run_df_backtest(
         backtest_settings_tuple=backtest_settings_tuple,
diff --git a/tests/backtest/vol_backtest/testing.ipynb b/tests/backtest/vol_backtest/testing.ipynb
index 4829dc9..faeb44d 100644
--- a/tests/backtest/vol_backtest/testing.ipynb
+++ b/tests/backtest/vol_backtest/testing.ipynb
@@ -49,7 +49,7 @@
     "import pickle\n",
     "\n",
     "\n",
-    "with open(\"thing.pkl\", \"rb\") as f:\n",
+    "with open(\"Volume_Candles_Tuple_mar_apr.pkl\", \"rb\") as f:\n",
     "    candles = pickle.load(f)\n",
     "\n",
     "\n",