Merge pull request #79 from trb143/flex_config

Add Enviroment config option - Thanks @trb143
Orange-Cyberdefense · Aug 5, 2024 · 8e2d96b · 8e2d96b
2 parents 42bb31a + 3d941d7
commit 8e2d96b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -2,6 +2,7 @@
 .idea/
 __pycache__/
 *.pyc
+.arsenal.json
 arsenal/data/cheats/.obsidian/
 cheats/.obsidian/
 cheats/Arsenal/.obsidian/

diff --git a/arsenal/modules/config.py b/arsenal/modules/config.py
@@ -22,6 +22,9 @@
 os.environ.setdefault('ESCDELAY', '25')
 os.environ['TERM'] = 'xterm-256color'
 
-savevarfile = join(HOMEPATH, ".arsenal.json")
+if os.environ.get('ARSENAL_LOCAL'):
+    savevarfile = join(os.getcwd(), ".arsenal.json")
+else:
+    savevarfile = join(HOMEPATH, ".arsenal.json")
 
 PREFIX_GLOBALVAR_NAME = "arsenal_prefix_cmd"