1 files changed, 48 insertions, 0 deletions
diff --git a/todo/gui/tasks/table/menu.py b/todo/gui/tasks/table/menu.py
new file mode 100644
index 0000000..bc039b2
--- /dev/null
+++ b/todo/gui/tasks/table/menu.py
@@ -0,0 +1,48 @@
+from PyQt5 import QtWidgets, QtCore
+from typing import List
+
+import todo.gui.tasks.dialog
+from todo.model.status import Status
+from todo.model.task import Task, ValidTaskForm
+from todo.model.tag import Tag
+
+def open(table: QtWidgets.QTableWidget, status: Status, update_task_signal, position):
+    rows = set([index.row() for index in table.selectedIndexes()])
+
+    menu = QtWidgets.QMenu(table)
+
+    if len(rows) == 1:
+        modify_action = menu.addAction(todo.gui.icon.dialog_open(menu.style()), "modify")
+    else:
+        modify_action = QtWidgets.QAction(menu)
+
+    delete_action = menu.addAction(todo.gui.icon.trash(menu.style()), "delete")
+
+    if status != Status.READY:
+        move_to_ready = menu.addAction(todo.gui.icon.task_ready(menu.style()), "move to ready")
+    else:
+        move_to_ready = QtWidgets.QAction(menu)
+
+    if status != Status.WAITING:
+        move_to_waiting = menu.addAction(todo.gui.icon.task_waiting(menu.style()), "move to waiting")
+    else:
+        move_to_waiting = QtWidgets.QAction(menu)
+
+    if status != Status.MAYBE:
+        move_to_maybe = menu.addAction(todo.gui.icon.task_maybe(menu.style()), "move to maybe")
+    else:
+        move_to_maybe = QtWidgets.QAction(menu)
+
+    action = menu.exec_(table.mapToGlobal(position + QtCore.QPoint(15, 20)))
+    if action == modify_action and len(rows) == 1:
+        row = list(rows)[0]
+        (task, tags) = table.get_at(row)
+        todo.gui.tasks.dialog.update(table, update_task_signal, row, task, tags).exec_()
+    elif action == delete_action:
+        todo.gui.tasks.dialog.confirm_delete(table, rows, lambda: table.delete_rows(rows))
+    elif action == move_to_ready:
+        todo.gui.tasks.dialog.confirm_move(table, rows, Status.READY, lambda: table.update_status(rows, Status.READY))
+    elif action == move_to_waiting:
+        todo.gui.tasks.dialog.confirm_move(table, rows, Status.WAITING, lambda: table.update_status(rows, Status.WAITING))
+    elif action == move_to_maybe:
+        todo.gui.tasks.dialog.confirm_move(table, rows, Status.MAYBE, lambda: table.update_status(rows, Status.MAYBE))