From b1a79e70ed02ceeac0f2343067e0f58deff3b37e Mon Sep 17 00:00:00 2001
From: Armaan Bhojwani <me@armaanb.net>
Date: Mon, 1 Feb 2021 13:02:47 -0500
Subject: [PATCH] Comment and clean code

---
 lightcards/display.py  | 7 ++++++-
 lightcards/progress.py | 5 +----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/lightcards/display.py b/lightcards/display.py
index d0dd9a7..5f7276b 100644
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -16,6 +16,7 @@ class Display():
         self.obj = obj
 
     def run(self, stdscr):
+        """Set important options before beginning"""
         self.win = stdscr
         curses.curs_set(0)  # Hide cursor
         curses.init_pair(1, curses.COLOR_CYAN, 0)
@@ -24,6 +25,7 @@ class Display():
         self.get_key()
 
     def leave(self):
+        """Pickle stack before quitting"""
         if self.obj.getIdx() == len(self.stack):
             self.obj.setIdx(0)
 
@@ -71,6 +73,7 @@ class Display():
         self.win.insstr(bar_end, curses.color_pair(1))
 
     def menu_print(self, string, err=False):
+        """Print messages on the menu screen"""
         self.win.clear()
         if err:
             color = curses.color_pair(2)
@@ -81,6 +84,7 @@ class Display():
         self.menu_grab()
 
     def menu_grab(self):
+        """Grab keypresses for the menu screen"""
         while True:
             key = self.win.getkey()
             if key == "q":
@@ -163,6 +167,7 @@ class Display():
             self.menu_grab()
 
     def wrap_width(self):
+        """Calculate the width at which the body should wrap"""
         (_, mcols) = self.win.getmaxyx()
         wrap_width = mcols - 20
         if wrap_width > 80:
@@ -280,6 +285,7 @@ class Display():
                 self.get_key()
 
     def disp_sidebar(self):
+        """Display a sidebar with the starred terms"""
         (mlines, mcols) = self.win.getmaxyx()
         left = mcols - 19
 
@@ -303,4 +309,3 @@ class Display():
 
         if i == 0:
             self.win.addstr(2, left, "None starred")
-
diff --git a/lightcards/progress.py b/lightcards/progress.py
index 665ae31..2147bff 100644
--- a/lightcards/progress.py
+++ b/lightcards/progress.py
@@ -9,6 +9,7 @@ import shutil
 global dired
 dired = f"{os.path.expanduser('~')}/.cache/lightcards/"
 
+
 def name_gen(stra):
     hasher = hashlib.md5()
     hasher.update(str(stra).encode("utf-8"))
@@ -44,7 +45,3 @@ def purge(stra):
 def purge_all():
     if os.path.exists(dired):
         shutil.rmtree(dired)
-
-
-if __name__ == "__main__":
-    main()
-- 
2.39.2