From 0413dcaaa415e871614bf8f6125fa7a2c03af387 Mon Sep 17 00:00:00 2001
From: Michael Spang <mspang@csclub.uwaterloo.ca>
Date: Sun, 23 Aug 2009 15:40:20 -0400
Subject: [PATCH] Add labels to main menu

Most of CEO is restricted to office staff or worse, but anyone can run
it. This add labels to make necessary privileges clear to the user.
---
 ceo/urwid/main.py    | 56 +++++++++++++++++++++++++++++++++++---------
 ceo/urwid/widgets.py | 10 ++++++++
 2 files changed, 55 insertions(+), 11 deletions(-)

diff --git a/ceo/urwid/main.py b/ceo/urwid/main.py
index 252ddff..1fb1cd9 100644
--- a/ceo/urwid/main.py
+++ b/ceo/urwid/main.py
@@ -1,4 +1,4 @@
-import sys, random, urwid.curses_display
+import os, grp, pwd, sys, random, urwid.curses_display
 from ceo.urwid.widgets import *
 from ceo.urwid.window import *
 from ceo.urwid import newmember, renew, info, search, positions, groups, \
@@ -143,26 +143,60 @@ def change_shell(data):
         shell.EndPage
     ], (50, 20))
 
-def run():
-    menu = make_menu([
+def check_group(group):
+    try:
+        me = pwd.getpwuid(os.getuid()).pw_name
+        return me in grp.getgrnam(group).gr_mem
+    except KeyError:
+        pass
+
+def top_menu():
+    office_only = [
         ("New Member", new_member, None),
         ("New Club Rep", new_club_user, None),
         ("Renew Membership", renew_member, None),
         ("Renew Club Rep", renew_club_user, None),
         ("New Club", new_club, None),
-        ("Display Member", display_member, None),
-        ("Change Shell", change_shell, None),
-        ("Search", search_members, None),
+        ("Library", library.library, None),
+        ("Databases", databases.databases, None),
+    ]
+    syscom_only = [
         ("Manage Club or Group Members", manage_group, None),
         ("Manage Positions", manage_positions, None),
         ("Manage Office Staff", groups.group_members, office_data),
         ("Manage Systems Committee", groups.group_members, syscom_data),
-        ("Library", library.library, None),
-        ("Databases", databases.databases, None),
+    ]
+    unrestricted = [
+        ("Display Member", display_member, None),
+        ("Change Shell", change_shell, None),
+        ("Search", search_members, None),
+    ]
+    footer = [
         ("Exit", raise_abort, None),
-    ])
-    push_window( menu, program_name() )
-    event_loop( ui )
+    ]
+    menu = None
+
+    # reorder the menu for convenience
+    if not check_group('office') and not check_group('syscom'):
+        menu = labelled_menu([
+            ('Unrestricted', unrestricted),
+            ('Office Staff', office_only),
+            ('Systems Committee', syscom_only),
+            (None, footer)
+        ])
+    else:
+        menu = labelled_menu([
+            ('Office Staff', office_only),
+            ('Unrestricted', unrestricted),
+            ('Systems Committee', syscom_only),
+            (None, footer)
+        ])
+
+    return menu
+
+def run():
+    push_window(top_menu(), program_name())
+    event_loop(ui)
 
 def start():
     ui.run_wrapper( run )
diff --git a/ceo/urwid/widgets.py b/ceo/urwid/widgets.py
index cf1f31c..2806db1 100644
--- a/ceo/urwid/widgets.py
+++ b/ceo/urwid/widgets.py
@@ -11,6 +11,16 @@ def make_menu(items):
     items = [ urwid.AttrWrap( ButtonText( cb, data, txt ), 'menu', 'selected') for (txt, cb, data) in items ]
     return urwid.ListBox( items )
 
+def labelled_menu(itemses):
+    widgets = []
+    for label, items in itemses:
+        if label:
+            widgets.append(urwid.Text(label))
+        widgets += (urwid.AttrWrap(ButtonText(cb, data, txt), 'menu', 'selected') for (txt, cb, data) in items)
+        widgets.append(urwid.Divider())
+    widgets.pop()
+    return urwid.ListBox(widgets)
+
 def push_wizard(name, pages, dimensions=(50, 10)):
     state = {}
     wiz = Wizard()