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