Comment and clean code

diff --git a/lightcards/display.py b/lightcards/display.py
index d0dd9a779b2881f19c897b1d8dad153ce362fb26..5f7276bac43fe977c0f66fef24213c2bc2062c0a 100644 (file)
--- 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 665ae315485cba08b742073af68e188608f031d3..2147bff8b0e31b47441ef3e631ba0a0f93b65419 100644 (file)
--- 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()