From 067c62a5ebbdfe415e3d18ffd76e04bc87299a85 Mon Sep 17 00:00:00 2001
From: Adrian Heine <mail@adrianheine.de>
Date: Wed, 19 Jun 2024 13:45:50 +0200
Subject: [PATCH] Add --otra-vez

---
 debian/changelog |  6 ++++++
 main.py          | 20 ++++++++++++++++++--
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/debian/changelog b/debian/changelog
index 1337b37..1e57dae 100644
--- a/debian/changelog
+++ b/debian/changelog
@@ -1,3 +1,9 @@
+klette (0.3) UNRELEASED; urgency=low
+
+  * Add --otra-vez
+
+ -- Adrian Heine <debian@adrianheine.de>  Sat, 19 Jun 2024 13:46:00 +0200
+
 klette (0.2) UNRELEASED; urgency=low
 
   * Various improvements.
diff --git a/main.py b/main.py
index f46329d..bfe6d88 100644
--- a/main.py
+++ b/main.py
@@ -164,6 +164,19 @@ class Sesion:
         self.bien = 0
         self.mal = 0
 
+    def otra_vez(self, quiero_unidad):
+        unidad = False
+        unidad_no = -1
+        paso = False
+        cur_palabras = False
+        for palabra in self.palabras:
+            if unidad != palabra['unidad']:
+                unidad_no += 1
+                unidad = palabra['unidad']
+            if quiero_unidad == None or quiero_unidad == unidad_no:
+                if self.status.get(palabra['id'], None) == 5:
+                    self.status[palabra['id']] = 4
+
     def empezar(self, quiero_unidad):
         unidad = False
         unidad_no = -1
@@ -222,7 +235,7 @@ class Sesion:
                 pass
             os.replace(STATUS_FILE + '.new', STATUS_FILE)
 
-def abfrage(parser, quiero_unidad):
+def abfrage(parser, quiero_unidad, otra_vez):
     random.seed()
     try:
         with open(VOKABELN_FILE, 'r') as f:
@@ -236,6 +249,8 @@ def abfrage(parser, quiero_unidad):
         return
 
     sesion = Sesion(palabras, status)
+    if otra_vez:
+        sesion.otra_vez(quiero_unidad)
     sesion.empezar(quiero_unidad)
     print(f'{bcolors.OKGREEN}+{sesion.bien}{bcolors.ENDC} / {bcolors.FAIL}-{sesion.mal}{bcolors.ENDC}')
 
@@ -245,6 +260,7 @@ parser.add_argument('--import-data', type=str, help="Path to assets", metavar="D
 default_status_file = os.environ['HOME'] + '/klett/1266/vokabeltrainerData147'
 parser.add_argument('--import-status', type=str, help="Path to AMF File, defaults to " + default_status_file, metavar="FILE", nargs='?', const=default_status_file)
 parser.add_argument('--unidad', type=int)
+parser.add_argument('--otra-vez', action='store_true')
 args = parser.parse_args()
 
 if args.import_data:
@@ -253,4 +269,4 @@ if args.import_data:
 elif args.import_status:
     import_status(args.import_status)
 else:
-    abfrage(parser, args.unidad)
+    abfrage(parser, args.unidad, args.otra_vez)