source: tools/policy/principal_tree.py @ e69367b

#!/usr/local/bin/python

import sys
import os
import os.path
import re

import gtk

import ABAC
import Creddy

class principal_tree(gtk.TreeView):
    def __init__(self, policy, title):
        gtk.TreeView.__init__(self, gtk.TreeStore(str, str, str))
        self.principals = set()
        self.creds = { }
        self.policy = policy

        self.append_column(gtk.TreeViewColumn("Principal", 
            gtk.CellRendererText(), text=0))
        self.append_column(gtk.TreeViewColumn(title, gtk.CellRendererText(), 
            text=1, background=2))

        self.recalc();

    def recalc(self):
        ts = self.get_model()
        ts.clear()
        self.principals = set()
        self.creds = { }
        row = { }
        for c in self.policy.credentials():
            p = self.selector(c)
            if p :
                if p not in self.creds:
                    self.creds[p] = [] 
                self.creds[p].append(c)
        
        for k, p in self.policy.principal_items():
            ri = ts.append(None, [p, '', 'white'])
            row[k] = ri

        for p, cred in self.creds.items():
            if p in row: 
                cstr = [ self.printer(c) for c in cred ]
                cstr.sort()
                for c in cstr:
                    ts.append(row[p], [ "", c, '#a5a5ff'])


class principal_issued_tree(principal_tree):
    def __init__(self, policy):
        principal_tree.__init__(self, policy, "Attributes Issued")


    def selector(self, c):
        if c.tail().is_principal():
            return c.head().principal()
        else:
            return None

    def printer(self, c):
        return "%s issued to %s" % (
                self.policy.replace_keyids(c.head().string()), 
                self.policy.replace_keyids(c.tail().string()))


class principal_rules_tree(principal_tree):
    def __init__(self, policy):
        principal_tree.__init__(self, policy, "Rules Issued")


    def selector(self, c):
        if not c.tail().is_principal():
            return c.head().principal()
        else:
            return None

    def printer(self, c):
        return self.policy.translate_cred(c)

class principal_assigned_tree(principal_tree):
    def __init__(self, policy):
        principal_tree.__init__(self, policy, "Attributes Held")

    def selector(self, c):
        if c.tail().is_principal():
            return c.tail().string()
        else:
            return None

    def printer(self, c):
        return "%s" % self.policy.replace_keyids(c.head().string())

class principal_attribute_tree_base(gtk.TreeView):
    def __init__(self, policy):
        gtk.TreeView.__init__(self, gtk.TreeStore(str, str, str, str))
        self.principals = set()
        self.creds = { }
        self.policy = policy

        self.append_column(gtk.TreeViewColumn("Principal", 
            gtk.CellRendererText(), text=0, background=2))
        self.append_column(gtk.TreeViewColumn("Attributes", 
            gtk.CellRendererText(), text=1, background=3))

        self.recalc()

    def recalc(self):
        ts = self.get_model()
        ts.clear()
        row = { }
        attrs = set()
        for c in self.policy.credentials():
            if self.selector(c):
                attrs.add(c.head().string())
        
        for k, p in self.policy.principal_items():
            ri = ts.append(None, [p, "", 'white', 'white'])
            row[k] = ri

        for a in attrs:
            for p in self.policy.principal_keyids():
                ok, proof = self.policy.query(a, p)
                if ok:
                    pi = ts.append(row[p], 
                            ["", self.policy.replace_keyids(a), 'white', 
                                '#a5a5ff'])
                    for c in proof:
                        ts.append(pi, 
                                ["", self.policy.translate_cred(c), 'white', 
                                    '#ffa5a5'])

class principal_attribute_tree(principal_attribute_tree_base):
    def selector(self, c):
        return True

class principal_action_tree(principal_attribute_tree_base):
    def selector(self, c):
        return c.head().string() in self.policy.actions

class attribute_principal_tree_base(gtk.TreeView):
    def __init__(self, policy):
        gtk.TreeView.__init__(self, gtk.TreeStore(str, str, str, str))
        self.creds = { }
        self.policy = policy

        self.append_column(gtk.TreeViewColumn("Attributes", 
            gtk.CellRendererText(), text=0, background=2))
        self.append_column(gtk.TreeViewColumn("Principal", 
            gtk.CellRendererText(), text=1, background=3))

        self.recalc()

    def recalc(self):
        ts = self.get_model()
        ts.clear()
        row = { }
        principals = set()
        attrs = set()
        for c in self.policy.credentials():
            if self.selector(c):
                attrs.add(c.head().string())
        
        for a in attrs:
            ri = ts.append(None, [self.policy.replace_keyids(a), 
                "", 'white', 'white'])
            row[a] = ri

        for a in attrs:
            for p in self.policy.principal_keyids():
                ok, proof = self.policy.query(a, p)
                if ok:
                    pi = ts.append(row[a], 
                            ["", self.policy.replace_keyids(p), 'white', 
                                '#a5a5ff'])
                    for c in proof:
                        ts.append(pi, 
                                ["", self.policy.translate_cred(c), 'white', 
                                    '#ffa5a5'])

class attribute_principal_tree(attribute_principal_tree_base):
    def selector(self, c):
        return True

class action_principal_tree(attribute_principal_tree_base):
    def selector(self, c):
        return c.head().string() in self.policy.actions