pyceo-broken/ceo/urwid/main.py

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, \
    shell, library, databases

def program_name():
    cwords = [ "CSC" ] * 20 + [ "Club" ] * 10 + [ "Campus" ] * 5 + \
        [ "Communist", "Canadian", "Celestial", "Cryptographic", "Calum's",
          "Canonical", "Capitalist", "Catastrophic", "Ceremonial", "Chaotic", "Civic",
          "City", "County", "Caffeinated" ]
    ewords = [ "Embellished", "Ergonomic", "Electric", "Eccentric", "European", "Economic",
        "Evil", "Egotistical", "Elliptic", "Emasculating", "Embalming",
        "Embryonic", "Emigrant", "Emissary's", "Emoting", "Employment", "Emulated",
        "Enabling", "Enamoring", "Encapsulated", "Enchanted", "Encoded", "Encrypted",
        "Encumbered", "Endemic", "Enhanced", "Enigmatic", "Enlightened", "Enormous",
        "Enrollment", "Enshrouded", "Ephemeral", "Epidemic", "Episodic", "Epsilon",
        "Equitable", "Equestrian", "Equilateral", "Erroneous", "Erratic",
        "Espresso", "Essential", "Estate", "Esteemed", "Eternal", "Ethical", "Eucalyptus",
        "Euphemistic", "Evangelist", "Evasive", "Everyday", "Evidence", "Eviction", "Evildoer's",
        "Evolution", "Exacerbation", "Exalted", "Examiner's", "Excise", "Exciting", "Exclusion",
        "Exec", "Executioner's", "Exile", "Existential", "Expedient", "Expert", "Expletive",
        "Exploiter's", "Explosive", "Exponential", "Exposing", "Extortion", "Extraction",
        "Extraneous", "Extravaganza", "Extreme", "Extraterrestrial", "Extremist", "Eerie" ]
    owords = [ "Office" ] * 50 + [ "Outhouse", "Outpost" ]

    cword = random.choice(cwords)
    eword = random.choice(ewords)
    oword = random.choice(owords)

    return "%s %s %s" % (cword, eword, oword)

def new_member(*args, **kwargs):
    push_wizard("New Member", [
        newmember.IntroPage,
        newmember.InfoPage,
        newmember.NumberOfTermsPage,
        newmember.SignPage,
        newmember.PassPage,
        newmember.EndPage,
    ], (60, 15))

def new_club(*args, **kwargs):
    push_wizard("New Club Account", [
        newmember.ClubIntroPage,
        newmember.ClubInfoPage,
        (newmember.EndPage, "club"),
    ], (60, 15))

def new_club_user(*args, **kwargs):
    push_wizard("New Club Rep Account", [
        newmember.ClubUserIntroPage,
        newmember.ClubNoPayPage,
        newmember.InfoPage,
        newmember.NumberOfTermsPage,
        newmember.SignPage,
        newmember.PassPage,
        (newmember.EndPage, "clubuser"),
    ], (60, 15))

def manage_group(*args, **kwargs):
    push_wizard("Manage Club or Group Members", [
        groups.IntroPage,
        groups.InfoPage,
    ], (60, 15))

def renew_member(*args, **kwargs):
    push_wizard("Renew Membership", [
        renew.IntroPage,
        renew.UserPage,
        renew.EmailPage,
        renew.EmailDonePage,
        renew.TermPage,
        renew.PayPage,
        renew.EndPage,
    ], (60, 15))

def renew_club_user(*args, **kwargs):
    push_wizard("Renew Club Rep Account", [
        renew.ClubUserIntroPage,
        newmember.ClubNoPayPage,
        renew.UserPage,
        renew.EmailPage,
        renew.EmailDonePage,
        (renew.TermPage, "clubuser"),
        (renew.EndPage, "clubuser"),
    ], (60, 15))

def display_member(data):
    push_wizard("Display Member", [
        renew.UserPage,
        info.InfoPage,
    ], (60, 15))

def search_members(data):
    menu = make_menu([
        ("Members by term", search_term, None),
        ("Members by name", search_name, None),
        ("Members by group", search_group, None),
        ("Back", raise_back, None),
    ])
    push_window(menu, "Search Members")

def search_name(data):
    push_wizard("By Name", [ search.NamePage ])

def search_term(data):
    push_wizard("By Term", [ search.TermPage ])

def search_group(data):
    push_wizard("By Group", [ search.GroupPage ])

def manage_positions(data):
    push_wizard("Manage Positions", [
        positions.IntroPage,
        positions.InfoPage,
        positions.EndPage,
    ], (50, 15))

def change_shell(data):
    push_wizard("Change Shell", [
        shell.IntroPage,
        shell.YouPage,
        shell.ShellPage,
        shell.EndPage
    ], (50, 20))

def create_mysql_db(data):
    push_wizard("Create MySQL database", [
        databases.IntroPage,
        databases.UserPage,
        databases.EndPage,
    ], (60, 15))

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),
        ("Library", library.library, None),
    ]
    syscom_only = [
        ("Manage Club or Group Members", manage_group, None),
        ("Manage Positions", manage_positions, None),
    ]
    unrestricted = [
        ("Display Member", display_member, None),
        ("Search Members", search_members, None),
        ("Change Shell", change_shell, None),
        ("Create MySQL database", create_mysql_db, None),
    ]
    footer = [
        ("Exit", raise_abort, None),
    ]
    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 )

if __name__ == '__main__':
    start()