alikia2x/OpenRewind
1
1
Fork 0
Code Issues Releases 1 Activity

added --storage-path argument to config

Browse Source
This commit is contained in:
Koen van Eijk 2024-06-09 23:23:24 +02:00
parent 07a33f1ee4
commit 110b0a3543
1 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
openrecall/config.py
View File

@ -1,6 +1,24 @@
import os import os
import sys import sys
import argparse
parser = argparse.ArgumentParser(
description="OpenRecall"
)
parser.add_argument(
"--storage-path",
type=str,
default=None,
help="Path to store the screenshots and database",
)
args = parser.parse_args()
if args.storage_path:
appdata_folder = args.storage_path
screenshots_path = os.path.join(appdata_folder, "screenshots")
db_path = os.path.join(appdata_folder, "recall.db")
def get_appdata_folder(app_name="openrecall"): def get_appdata_folder(app_name="openrecall"):
if sys.platform == "win32": if sys.platform == "win32":
@ -18,10 +36,14 @@ def get_appdata_folder(app_name="openrecall"):
os.makedirs(path) os.makedirs(path)
return path return path
if args.storage_path:
appdata_folder = get_appdata_folder() appdata_folder = args.storage_path
db_path = os.path.join(appdata_folder, "recall.db") screenshots_path = os.path.join(appdata_folder, "screenshots")
screenshots_path = os.path.join(appdata_folder, "screenshots") db_path = os.path.join(appdata_folder, "recall.db")
else:
appdata_folder = get_appdata_folder()
db_path = os.path.join(appdata_folder, "recall.db")
screenshots_path = os.path.join(appdata_folder, "screenshots")
if not os.path.exists(screenshots_path): if not os.path.exists(screenshots_path):
try: try: