Set the default vm type
[invirt/packages/invirt-web.git] / code / webcommon.py
1 """Exceptions for the web interface."""
2
3 from sipb_xen_database import Machine, MachineAccess
4
5 class MyException(Exception):
6     """Base class for my exceptions"""
7     pass
8
9 class InvalidInput(MyException):
10     """Exception for user-provided input is invalid but maybe in good faith.
11
12     This would include setting memory to negative (which might be a
13     typo) but not setting an invalid boot CD (which requires bypassing
14     the select box).
15     """
16     def __init__(self, err_field, err_value, expl=None):
17         MyException.__init__(self, expl)
18         self.err_field = err_field
19         self.err_value = err_value
20
21 class CodeError(MyException):
22     """Exception for internal errors or bad faith input."""
23     pass
24
25 import controls
26
27 def cachedproperty(func):
28     name = '__cache_' + func.__name__ + '_' + str(id(func))
29     def getter(self):
30         try:
31             return getattr(self, name)
32         except AttributeError:
33             value = func(self)
34             setattr(self, name, value)
35             return value
36     return property(getter)
37
38 class Global(object):
39     """Global state of the system, to avoid duplicate remctls to get state"""
40     def __init__(self, user):
41         self.user = user
42     
43     machines = cachedproperty(lambda self: 
44                              [ma.machine for ma in 
45                               MachineAccess.select_by(user=self.user)])
46     uptimes = cachedproperty(lambda self: 
47                              controls.getUptimes(self.machines))
48
49     def clear(self):
50         """Clear the state so future accesses reload it."""
51         for attr in list(self.__dict__):
52             if attr.startswith('__cache_'):
53                 delattr(self, attr)
54
55 g = Global(None)