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.
pull/5/head
Michael Spang 14 years ago
parent 5dc46021c5
commit 0413dcaaa4
  1. 56
      ceo/urwid/main.py
  2. 10
      ceo/urwid/widgets.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 )

@ -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()

Loading…
Cancel
Save