- stdscr.insch(mlines, mcols+2, '2')
- stdscr.insch(mlines, mcols+3, '/')
- stdscr.insch(mlines, mcols+4, str(len(stack)))
- except Exception:
- pass
-
-
-def get_key(stdscr, stack, headers):
- idx = Status()
- curses.curs_set(0)
- disp_card(stdscr, stack, headers, idx)
-
- while True:
- key = stdscr.getkey()
- try:
- if key == "q" or key == os.linesep:
- exit(0)
- elif key in ["l", "KEY_LEFT"]:
- idx.forward(stack)
- idx.setSide(0)
- disp_card(stdscr, stack, headers, idx)
- elif key in ["h", "KEY_RIGHT"]:
- idx.back()
- idx.setSide(0)
- disp_card(stdscr, stack, headers, idx)
- elif key in ["j", "k", "KEY_UP", "KEY_DOWN"]:
- idx.flip()
- disp_card(stdscr, stack, headers, idx)
- except Exception:
- pass
+ self.disp_card()
+
+ def leave(self):
+ """Pickle stack and confirm before quitting"""
+ self.quit_obj.disp()
+ if self.obj.index + 1 == len(self.stack):
+ self.obj.index = 0
+
+ progress.dump(self.stack, runner.get_orig()[1])
+ sys.exit(0)
+
+ def nstarred(self):
+ """Get total number of starred cards"""
+ return [card for card in self.stack if card.starred]
+
+ def disp_bar(self):
+ """
+ Display the statusbar at the bottom of the screen with progress, star
+ status, and card side.
+ """
+ (mlines, mcols) = self.win.getmaxyx()
+ self.win.hline(mlines - 3, 0, 0, mcols)
+
+ # Calculate percent done
+ if len(self.stack) <= 1:
+ percent = "100"
+ else:
+ percent = str(
+ round(self.obj.index / (len(self.stack) - 1) * 100)
+ ).zfill(2)
+
+ # Print yellow if starred
+ if self.current_card().starred:
+ star_color = curses.color_pair(3)
+ else:
+ star_color = curses.color_pair(1)
+
+ # Compose bar text
+ bar_start = "["
+ bar_middle = self.current_card().printStar()
+ bar_end = f"] [{len(self.nstarred())}/{str(len(self.stack))} starred] "
+ if self.view != 3:
+ bar_end += (
+ f" [{self.get_side()} ("
+ f"{str(int(self.current_card().side) + 1)})]"
+ )
+ bar_end += f" [View {str(self.view)}]"
+
+ # Put it all togethor
+ height = mlines - 2
+ self.win.addstr(height, 0, bar_start, curses.color_pair(1))
+ self.win.addstr(height, len(bar_start), bar_middle, star_color)
+ self.win.addstr(
+ height,
+ len(bar_start + bar_middle),
+ textwrap.shorten(bar_end, width=mcols - 20, placeholder="…"),
+ curses.color_pair(1),
+ )
+
+ progress = (
+ f"[{percent}% ("
+ f"{str(self.obj.index + 1).zfill(len(str(len(self.stack))))}"
+ f"/{str(len(self.stack))})] "
+ )
+
+ self.win.addstr(
+ height + 1,
+ 0,
+ progress,
+ curses.color_pair(1),
+ )
+
+ for i in range(
+ int(
+ self.obj.index
+ / (len(self.stack) - 1)
+ * (mcols - len(progress))
+ - 1
+ )
+ ):
+ self.win.addch(
+ height + 1,
+ i + len(progress),
+ self.config["progress_char"],
+ curses.color_pair(1),
+ )
+
+ def wrap_width(self):
+ """Calculate the width at which the body text should wrap"""
+ (_, mcols) = self.win.getmaxyx()
+ wrap_width = mcols - 20
+ if wrap_width > 80:
+ wrap_width = 80
+ return wrap_width
+
+ def get_side(self):
+ if self.obj.side == 0:
+ return self.headers[self.current_card().side]
+ else:
+ return self.headers[self.current_card().get_reverse()]
+
+ def disp_card(self):
+ (_, mcols) = self.win.getmaxyx()
+ self.main_panel.bottom()
+ self.win.clear()
+ num_done = str(self.obj.index + 1).zfill(len(str(len(self.stack))))
+
+ if self.view in [1, 2, 4]:
+ """
+ Display the contents of the card.
+ Shows a header, a horizontal line, and the contents of the current
+ side.
+ """
+ # If on the back of the card, show the content of the front side in
+ # the header
+ if self.view == 1:
+ self.obj.side = 0
+ elif self.view == 2:
+ self.obj.side = 1
+
+ if self.current_card().side == 0:
+ top = num_done + " | " + self.get_side()
+ else:
+ top = (
+ num_done
+ + " | "
+ + self.get_side()
+ + ' | "'
+ + str(self.current_card().get()[self.obj.get_reverse()])
+ + '"'
+ )
+
+ self.win.addstr(
+ 0,
+ 0,
+ textwrap.shorten(top, width=mcols - 20, placeholder="…"),
+ curses.A_BOLD,
+ )
+
+ # Show current side
+ self.win.addstr(
+ 2,
+ 0,
+ textwrap.fill(
+ self.current_card().get()[self.obj.side],
+ width=self.wrap_width(),
+ ),
+ )
+
+ elif self.view == 3:
+ """
+ Display the contents of the card with both the front and back sides.
+ """
+ (_, mcols) = self.win.getmaxyx()
+ self.main_panel.bottom()
+ self.win.clear()
+
+ self.win.addstr(
+ 0,
+ 0,
+ textwrap.shorten(
+ num_done,
+ width=mcols - 20,
+ placeholder="…",
+ ),
+ curses.A_BOLD,
+ )
+
+ # Show card content
+ self.win.addstr(
+ 2,
+ 0,
+ textwrap.fill(
+ self.headers[0] + ": " + self.current_card().front,
+ width=self.wrap_width(),
+ )
+ + "\n\n"
+ + textwrap.fill(
+ self.headers[1] + ": " + self.current_card().back,
+ width=self.wrap_width(),
+ ),
+ )
+
+ self.win.hline(1, 0, curses.ACS_HLINE, mcols)
+ self.disp_bar()
+ self.disp_sidebar()
+
+ def current_card(self):
+ """Get current card object"""
+ return self.stack[self.obj.index]
+
+ def get_key(self):
+ """
+ Display a card and wait for the input.
+ Used as a general way of getting back into the card flow from a menu
+ """
+ while True:
+ self.check_size()
+ key = self.win.getkey()
+ if key == self.config["quit_key"]:
+ self.leave()
+ elif key in self.config["card_prev"]:
+ self.obj.back()
+ self.current_card().side = 0
+ self.disp_card()
+ elif key in self.config["card_next"]:
+ if self.obj.index + 1 == len(self.stack):
+ if self.config["show_menu_at_end"]:
+ self.menu_obj.disp()
+ else:
+ self.obj.forward(self.stack)
+ self.current_card().side = 0
+ self.disp_card()
+ elif key in self.config["card_flip"] and self.view != 3:
+ self.current_card().flip()
+ self.disp_card()
+ elif key in self.config["card_star"]:
+ self.current_card().toggleStar()
+ self.disp_card()
+ elif key in self.config["card_first"]:
+ self.obj.index = 0
+ self.current_card().side = 0
+ self.disp_card()
+ elif key in self.config["card_last"]:
+ self.obj.index = len(self.stack) - 1
+ self.current_card().side = 0
+ self.disp_card()
+ elif key in self.config["help_disp"]:
+ self.help_obj.disp()
+ elif key in self.config["menu_disp"]:
+ self.menu_obj.disp()
+ elif key in self.config["view_one"]:
+ self.view = 1
+ elif key in self.config["view_two"]:
+ self.view = 2
+ elif key in self.config["view_three"]:
+ self.view = 3
+
+ def disp_sidebar(self):
+ """Display a sidebar with the starred terms"""
+ (mlines, mcols) = self.win.getmaxyx()
+ left = mcols - 19
+
+ self.win.addstr(
+ 0,
+ mcols - 16,
+ "STARRED CARDS",
+ curses.color_pair(3) + curses.A_BOLD,
+ )
+ self.win.vline(0, mcols - 20, 0, mlines - 3)
+
+ nstarred = self.nstarred()
+ for i, card in enumerate(nstarred):
+ term = card.get()[self.obj.side]
+ if len(term) > 18:
+ term = term[:18] + "…"
+
+ if i > mlines - 6:
+ for i in range(19):
+ self.win.addch(mlines - 3, left + i, " ")
+
+ self.win.addstr(
+ mlines - 4,
+ left,
+ f"({len(nstarred) - i - 2} more)",
+ )
+ else:
+ self.win.addstr(2 + i, left, term)
+
+ if len(self.nstarred()) == 0:
+ self.win.addstr(2, left, "None starred")