X-Git-Url: https://git.armaanb.net/?p=lightcards.git;a=blobdiff_plain;f=lightcards%2Fdisplay.py;h=d0dd9a779b2881f19c897b1d8dad153ce362fb26;hp=49840583eb27c1987a68bd0df13db85dfc07f3c6;hb=c812ea549b247e92fc32227be419061a94c318a9;hpb=cdfb48d9f8d7816da9e9c78a4bbe959efb1a9a36

diff --git a/lightcards/display.py b/lightcards/display.py
index 4984058..d0dd9a7 100644
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -27,7 +27,7 @@ class Display():
         if self.obj.getIdx() == len(self.stack):
             self.obj.setIdx(0)
 
-        progress.dump((self.obj, self.stack, self.headers), self.stack)
+        progress.dump(self.stack, lightcards.get_orig())
         sys.exit(0)
 
     def ntotal(self):
@@ -92,6 +92,9 @@ class Display():
             elif key == "y":
                 self.stack = lightcards.get_orig()[1]
                 self.menu_print("Stack reset!")
+            elif key == "a":
+                self.stack.sort()
+                self.menu_print("Stack alphabetized!")
             elif key == "u":
                 [x.unStar() for x in self.stack]
                 self.menu_print("All unstarred!")
@@ -146,6 +149,7 @@ class Display():
                         curses.A_BOLD)
         self.win.hline(1, 0, curses.ACS_HLINE, 15)
         self.win.addstr(2, 0, "[y]: reset stack to original state\n" +
+                        "[a]: alphabetize stack\n" +
                         "[z]: shuffle stack\n" +
                         "[f]: flip all cards in stack\n" +
                         "[t]: reverse stack order\n" +