From ecb5c0e8621bd36a13a92d9512424e972db92369 Mon Sep 17 00:00:00 2001
From: Luc Mcgrady <lucmcgrady@gmail.com>
Date: Sun, 23 Jul 2023 15:39:52 +0100
Subject: [PATCH] Add wildcard support (#381)

* Added: Wildcard support

* Fix: forgot to change variable name
---
 README.md                               |  2 --
 package/fsrs4anki_optimizer/__main__.py | 22 ++++++++++------------
 2 files changed, 10 insertions(+), 14 deletions(-)

diff --git a/README.md b/README.md
index f273c5d..170a68c 100644
--- a/README.md
+++ b/README.md
@@ -166,8 +166,6 @@ options:
   -o OUT, --out OUT    File to APPEND the automatically generated profile to.
 ```
 
-Please note that this does not support multiple files in one command / wildcards so for that functionality use a for loop.
-
 #### Expected Functionality:
 
 ![image](https://github.com/Luc-mcgrady/fsrs4anki/assets/63685643/ac2e8ae0-726c-46fd-b110-0701fa87cb66)
diff --git a/package/fsrs4anki_optimizer/__main__.py b/package/fsrs4anki_optimizer/__main__.py
index 6f15d87..e276a28 100644
--- a/package/fsrs4anki_optimizer/__main__.py
+++ b/package/fsrs4anki_optimizer/__main__.py
@@ -1,4 +1,3 @@
-# python fsrs4anki_optimizer_cmd.py <filename.apkg|filename.colpkg>
 import fsrs4anki_optimizer
 import argparse
 import json
@@ -86,7 +85,7 @@ def remembered_fallback_prompt(key: str, pretty: str = None):
     profile = \
     f"""{{
     // Generated, Optimized anki deck settings
-    "deckName": "{args.filename}",// PLEASE CHANGE THIS TO THE DECKS PROPER NAME
+    "deckName": "{filename}",// PLEASE CHANGE THIS TO THE DECKS PROPER NAME
     "w": {optimizer.w},
     "requestRetention": {optimizer.optimal_retention},
     "maximumInterval": 36500,
@@ -112,7 +111,7 @@ def remembered_fallback_prompt(key: str, pretty: str = None):
     config_save = os.path.expanduser(".fsrs4anki_optimizer")
 
     parser = argparse.ArgumentParser()
-    parser.add_argument("filename")
+    parser.add_argument("filenames", nargs='+')
     parser.add_argument("-y","--yes",
                         action=argparse.BooleanOptionalAction,
                         help="If set automatically defaults on all stdin settings."
@@ -122,13 +121,12 @@ def remembered_fallback_prompt(key: str, pretty: str = None):
                         )
     args = parser.parse_args()
 
-
-    if os.path.isdir(args.filename):
-        files = [f for f in os.listdir(args.filename) if f.lower().endswith('.apkg')]
-        files = [os.path.join(args.filename, f) for f in files]
-        for file_path in files:
-            args.filename = file_path
-            process(file_path)
-    else:
-        process(args.filename)
+    for filename in args.filenames:
+        if os.path.isdir(filename):
+            files = [f for f in os.listdir(filename) if f.lower().endswith('.apkg')]
+            files = [os.path.join(filename, f) for f in files]
+            for file_path in files:
+                process(file_path)
+        else:
+            process(filename)