"""
LLVM-Lab Data Management
"""

from llvmlab import util
from llvmlab import machine, user

class Data(util.simple_repr_mixin):
    @staticmethod
    def fromdata(data):
        version = data['version']
        if version != 0:
            raise ValueError, "Unknown version"

        users = [user.User.fromdata(u)
                 for u in data['users']]
        machines = [machine.Machine.fromdata(u)
                    for u in data['machines']]
        return Data(users, machines)

    def todata(self):
        return { 'version' : 0,
                 'users' : [item.todata()
                            for item in self.users.values()
                            if item is not self.admin_user],
                 'machines' : [item.todata()
                               for item in self.machines.values()] }

    def __init__(self, users, machines):
        self.machines = dict((item.id, item) for item in machines)
        self.users = dict((item.id, item) for item in users)
        self.admin_user = None

    def set_admin_user(self, user):
        if user.id in self.users:
            raise ValueError, "database contains admin user %r" % user.id

        self.admin_user = user
        self.users[user.id] = user