showard | fb2a7fa | 2008-07-17 17:04:12 +0000 | [diff] [blame] | 1 | from datetime import datetime |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 2 | from django.db import models as dbmodels, connection |
showard | df06256 | 2008-07-03 19:56:37 +0000 | [diff] [blame] | 3 | from frontend.afe import model_logic |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 4 | from frontend import settings, thread_local |
showard | b1e5187 | 2008-10-07 11:08:18 +0000 | [diff] [blame^] | 5 | from autotest_lib.client.common_lib import enum, host_protections, global_config |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 6 | |
| 7 | |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 8 | class AclAccessViolation(Exception): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 9 | """\ |
| 10 | Raised when an operation is attempted with proper permissions as |
| 11 | dictated by ACLs. |
| 12 | """ |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 13 | |
| 14 | |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 15 | class Label(model_logic.ModelWithInvalid, dbmodels.Model): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 16 | """\ |
| 17 | Required: |
| 18 | name: label name |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 19 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 20 | Optional: |
| 21 | kernel_config: url/path to kernel config to use for jobs run on this |
| 22 | label |
| 23 | platform: if True, this is a platform label (defaults to False) |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 24 | only_if_needed: if True, a machine with this label can only be used if that |
| 25 | that label is requested by the job/test. |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 26 | """ |
| 27 | name = dbmodels.CharField(maxlength=255, unique=True) |
| 28 | kernel_config = dbmodels.CharField(maxlength=255, blank=True) |
| 29 | platform = dbmodels.BooleanField(default=False) |
| 30 | invalid = dbmodels.BooleanField(default=False, |
| 31 | editable=settings.FULL_ADMIN) |
showard | b1e5187 | 2008-10-07 11:08:18 +0000 | [diff] [blame^] | 32 | only_if_needed = dbmodels.BooleanField(default=False) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 33 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 34 | name_field = 'name' |
| 35 | objects = model_logic.ExtendedManager() |
| 36 | valid_objects = model_logic.ValidObjectsManager() |
mbligh | 5244cbb | 2008-04-24 20:39:52 +0000 | [diff] [blame] | 37 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 38 | def clean_object(self): |
| 39 | self.host_set.clear() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 40 | |
| 41 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 42 | def enqueue_job(self, job): |
| 43 | 'Enqueue a job on any host of this label.' |
| 44 | queue_entry = HostQueueEntry(meta_host=self, job=job, |
| 45 | status=Job.Status.QUEUED, |
| 46 | priority=job.priority) |
| 47 | queue_entry.save() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 48 | |
| 49 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 50 | class Meta: |
| 51 | db_table = 'labels' |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 52 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 53 | class Admin: |
| 54 | list_display = ('name', 'kernel_config') |
| 55 | # see Host.Admin |
| 56 | manager = model_logic.ValidObjectsManager() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 57 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 58 | def __str__(self): |
| 59 | return self.name |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 60 | |
| 61 | |
showard | fb2a7fa | 2008-07-17 17:04:12 +0000 | [diff] [blame] | 62 | class User(dbmodels.Model, model_logic.ModelExtensions): |
| 63 | """\ |
| 64 | Required: |
| 65 | login :user login name |
| 66 | |
| 67 | Optional: |
| 68 | access_level: 0=User (default), 1=Admin, 100=Root |
| 69 | """ |
| 70 | ACCESS_ROOT = 100 |
| 71 | ACCESS_ADMIN = 1 |
| 72 | ACCESS_USER = 0 |
| 73 | |
| 74 | login = dbmodels.CharField(maxlength=255, unique=True) |
| 75 | access_level = dbmodels.IntegerField(default=ACCESS_USER, blank=True) |
| 76 | |
| 77 | name_field = 'login' |
| 78 | objects = model_logic.ExtendedManager() |
| 79 | |
| 80 | |
| 81 | def save(self): |
| 82 | # is this a new object being saved for the first time? |
| 83 | first_time = (self.id is None) |
| 84 | user = thread_local.get_user() |
| 85 | if user and not user.is_superuser(): |
| 86 | raise AclAccessViolation("You cannot modify users!") |
| 87 | super(User, self).save() |
| 88 | if first_time: |
| 89 | everyone = AclGroup.objects.get(name='Everyone') |
| 90 | everyone.users.add(self) |
| 91 | |
| 92 | |
| 93 | def is_superuser(self): |
| 94 | return self.access_level >= self.ACCESS_ROOT |
| 95 | |
| 96 | |
| 97 | class Meta: |
| 98 | db_table = 'users' |
| 99 | |
| 100 | class Admin: |
| 101 | list_display = ('login', 'access_level') |
| 102 | search_fields = ('login',) |
| 103 | |
| 104 | def __str__(self): |
| 105 | return self.login |
| 106 | |
| 107 | |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 108 | class Host(model_logic.ModelWithInvalid, dbmodels.Model): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 109 | """\ |
| 110 | Required: |
| 111 | hostname |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 112 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 113 | optional: |
| 114 | locked: host is locked and will not be queued |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 115 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 116 | Internal: |
| 117 | synch_id: currently unused |
| 118 | status: string describing status of host |
| 119 | """ |
| 120 | Status = enum.Enum('Verifying', 'Running', 'Ready', 'Repairing', |
| 121 | 'Repair Failed', 'Dead', 'Rebooting', |
| 122 | string_values=True) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 123 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 124 | hostname = dbmodels.CharField(maxlength=255, unique=True) |
| 125 | labels = dbmodels.ManyToManyField(Label, blank=True, |
| 126 | filter_interface=dbmodels.HORIZONTAL) |
| 127 | locked = dbmodels.BooleanField(default=False) |
| 128 | synch_id = dbmodels.IntegerField(blank=True, null=True, |
| 129 | editable=settings.FULL_ADMIN) |
| 130 | status = dbmodels.CharField(maxlength=255, default=Status.READY, |
| 131 | choices=Status.choices(), |
| 132 | editable=settings.FULL_ADMIN) |
| 133 | invalid = dbmodels.BooleanField(default=False, |
| 134 | editable=settings.FULL_ADMIN) |
showard | d5afc2f | 2008-08-12 17:19:44 +0000 | [diff] [blame] | 135 | protection = dbmodels.SmallIntegerField(null=False, blank=True, |
showard | df06256 | 2008-07-03 19:56:37 +0000 | [diff] [blame] | 136 | choices=host_protections.choices, |
| 137 | default=host_protections.default) |
showard | fb2a7fa | 2008-07-17 17:04:12 +0000 | [diff] [blame] | 138 | locked_by = dbmodels.ForeignKey(User, null=True, blank=True, editable=False) |
| 139 | lock_time = dbmodels.DateTimeField(null=True, blank=True, editable=False) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 140 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 141 | name_field = 'hostname' |
| 142 | objects = model_logic.ExtendedManager() |
| 143 | valid_objects = model_logic.ValidObjectsManager() |
mbligh | 5244cbb | 2008-04-24 20:39:52 +0000 | [diff] [blame] | 144 | |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 145 | @staticmethod |
| 146 | def create_one_time_host(hostname): |
| 147 | query = Host.objects.filter(hostname=hostname) |
| 148 | if query.count() == 0: |
| 149 | host = Host(hostname=hostname, invalid=True) |
showard | a8411af | 2008-08-07 22:35:58 +0000 | [diff] [blame] | 150 | host.do_validate() |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 151 | else: |
| 152 | host = query[0] |
| 153 | if not host.invalid: |
| 154 | raise model_logic.ValidationError({ |
| 155 | 'hostname' : '%s already exists!' % hostname |
| 156 | }) |
| 157 | host.clean_object() |
showard | e65d2af | 2008-07-30 23:34:21 +0000 | [diff] [blame] | 158 | AclGroup.objects.get(name='Everyone').hosts.add(host) |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 159 | host.status = Host.Status.READY |
showard | 1ab512b | 2008-07-30 23:39:04 +0000 | [diff] [blame] | 160 | host.protection = host_protections.Protection.DO_NOT_REPAIR |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 161 | host.save() |
| 162 | return host |
mbligh | 5244cbb | 2008-04-24 20:39:52 +0000 | [diff] [blame] | 163 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 164 | def clean_object(self): |
| 165 | self.aclgroup_set.clear() |
| 166 | self.labels.clear() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 167 | |
| 168 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 169 | def save(self): |
| 170 | # extra spaces in the hostname can be a sneaky source of errors |
| 171 | self.hostname = self.hostname.strip() |
| 172 | # is this a new object being saved for the first time? |
| 173 | first_time = (self.id is None) |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 174 | if not first_time: |
| 175 | AclGroup.check_for_acl_violation_hosts([self]) |
showard | fb2a7fa | 2008-07-17 17:04:12 +0000 | [diff] [blame] | 176 | if self.locked and not self.locked_by: |
| 177 | self.locked_by = thread_local.get_user() |
| 178 | self.lock_time = datetime.now() |
| 179 | elif not self.locked and self.locked_by: |
| 180 | self.locked_by = None |
| 181 | self.lock_time = None |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 182 | super(Host, self).save() |
| 183 | if first_time: |
| 184 | everyone = AclGroup.objects.get(name='Everyone') |
| 185 | everyone.hosts.add(self) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 186 | |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 187 | def delete(self): |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 188 | AclGroup.check_for_acl_violation_hosts([self]) |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 189 | for queue_entry in self.hostqueueentry_set.all(): |
| 190 | queue_entry.deleted = True |
showard | 9a1f2e1 | 2008-10-02 11:14:29 +0000 | [diff] [blame] | 191 | queue_entry.abort(thread_local.get_user()) |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 192 | super(Host, self).delete() |
| 193 | |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 194 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 195 | def enqueue_job(self, job): |
| 196 | ' Enqueue a job on this host.' |
| 197 | queue_entry = HostQueueEntry(host=self, job=job, |
| 198 | status=Job.Status.QUEUED, |
| 199 | priority=job.priority) |
| 200 | # allow recovery of dead hosts from the frontend |
| 201 | if not self.active_queue_entry() and self.is_dead(): |
| 202 | self.status = Host.Status.READY |
| 203 | self.save() |
| 204 | queue_entry.save() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 205 | |
showard | 08f981b | 2008-06-24 21:59:03 +0000 | [diff] [blame] | 206 | block = IneligibleHostQueue(job=job, host=self) |
| 207 | block.save() |
| 208 | |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 209 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 210 | def platform(self): |
| 211 | # TODO(showard): slighly hacky? |
| 212 | platforms = self.labels.filter(platform=True) |
| 213 | if len(platforms) == 0: |
| 214 | return None |
| 215 | return platforms[0] |
| 216 | platform.short_description = 'Platform' |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 217 | |
| 218 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 219 | def is_dead(self): |
| 220 | return self.status == Host.Status.REPAIR_FAILED |
mbligh | 3cab4a7 | 2008-03-05 23:19:09 +0000 | [diff] [blame] | 221 | |
| 222 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 223 | def active_queue_entry(self): |
| 224 | active = list(self.hostqueueentry_set.filter(active=True)) |
| 225 | if not active: |
| 226 | return None |
| 227 | assert len(active) == 1, ('More than one active entry for ' |
| 228 | 'host ' + self.hostname) |
| 229 | return active[0] |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 230 | |
| 231 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 232 | class Meta: |
| 233 | db_table = 'hosts' |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 234 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 235 | class Admin: |
| 236 | # TODO(showard) - showing platform requires a SQL query for |
| 237 | # each row (since labels are many-to-many) - should we remove |
| 238 | # it? |
| 239 | list_display = ('hostname', 'platform', 'locked', 'status') |
showard | df06256 | 2008-07-03 19:56:37 +0000 | [diff] [blame] | 240 | list_filter = ('labels', 'locked', 'protection') |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 241 | search_fields = ('hostname', 'status') |
| 242 | # undocumented Django feature - if you set manager here, the |
| 243 | # admin code will use it, otherwise it'll use a default Manager |
| 244 | manager = model_logic.ValidObjectsManager() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 245 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 246 | def __str__(self): |
| 247 | return self.hostname |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 248 | |
| 249 | |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 250 | class Test(dbmodels.Model, model_logic.ModelExtensions): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 251 | """\ |
| 252 | Required: |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 253 | author: author name |
| 254 | description: description of the test |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 255 | name: test name |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 256 | time: short, medium, long |
| 257 | test_class: This describes the class for your the test belongs in. |
| 258 | test_category: This describes the category for your tests |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 259 | test_type: Client or Server |
| 260 | path: path to pass to run_test() |
| 261 | synch_type: whether the test should run synchronously or asynchronously |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 262 | sync_count: is a number >=1 (1 being the default). If it's 1, then it's an |
| 263 | async job. If it's >1 it's sync job for that number of machines |
| 264 | i.e. if sync_count = 2 it is a sync job that requires two |
| 265 | machines. |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 266 | Optional: |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 267 | dependencies: What the test requires to run. Comma deliminated list |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 268 | dependency_labels: many-to-many relationship with labels corresponding to |
| 269 | test dependencies. |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 270 | experimental: If this is set to True production servers will ignore the test |
| 271 | run_verify: Whether or not the scheduler should run the verify stage |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 272 | """ |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 273 | TestTime = enum.Enum('SHORT', 'MEDIUM', 'LONG', start_value=1) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 274 | SynchType = enum.Enum('Asynchronous', 'Synchronous', start_value=1) |
| 275 | # TODO(showard) - this should be merged with Job.ControlType (but right |
| 276 | # now they use opposite values) |
| 277 | Types = enum.Enum('Client', 'Server', start_value=1) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 278 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 279 | name = dbmodels.CharField(maxlength=255, unique=True) |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 280 | author = dbmodels.CharField(maxlength=255) |
| 281 | test_class = dbmodels.CharField(maxlength=255) |
| 282 | test_category = dbmodels.CharField(maxlength=255) |
| 283 | dependencies = dbmodels.CharField(maxlength=255, blank=True) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 284 | description = dbmodels.TextField(blank=True) |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 285 | experimental = dbmodels.BooleanField(default=True) |
| 286 | run_verify = dbmodels.BooleanField(default=True) |
| 287 | test_time = dbmodels.SmallIntegerField(choices=TestTime.choices(), |
| 288 | default=TestTime.MEDIUM) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 289 | test_type = dbmodels.SmallIntegerField(choices=Types.choices()) |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 290 | sync_count = dbmodels.IntegerField(default=1) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 291 | synch_type = dbmodels.SmallIntegerField(choices=SynchType.choices(), |
| 292 | default=SynchType.ASYNCHRONOUS) |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 293 | path = dbmodels.CharField(maxlength=255, unique=True) |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 294 | dependency_labels = dbmodels.ManyToManyField( |
| 295 | Label, blank=True, filter_interface=dbmodels.HORIZONTAL) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 296 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 297 | name_field = 'name' |
| 298 | objects = model_logic.ExtendedManager() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 299 | |
| 300 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 301 | class Meta: |
| 302 | db_table = 'autotests' |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 303 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 304 | class Admin: |
| 305 | fields = ( |
| 306 | (None, {'fields' : |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 307 | ('name', 'author', 'test_category', 'test_class', |
| 308 | 'test_time', 'synch_type', 'test_type', 'sync_count', |
| 309 | 'path', 'dependencies', 'experimental', 'run_verify', |
| 310 | 'description')}), |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 311 | ) |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 312 | list_display = ('name', 'test_type', 'description', 'synch_type') |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 313 | search_fields = ('name',) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 314 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 315 | def __str__(self): |
| 316 | return self.name |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 317 | |
| 318 | |
showard | 2b9a88b | 2008-06-13 20:55:03 +0000 | [diff] [blame] | 319 | class Profiler(dbmodels.Model, model_logic.ModelExtensions): |
| 320 | """\ |
| 321 | Required: |
| 322 | name: profiler name |
| 323 | test_type: Client or Server |
| 324 | |
| 325 | Optional: |
| 326 | description: arbirary text description |
| 327 | """ |
| 328 | name = dbmodels.CharField(maxlength=255, unique=True) |
| 329 | description = dbmodels.TextField(blank=True) |
| 330 | |
| 331 | name_field = 'name' |
| 332 | objects = model_logic.ExtendedManager() |
| 333 | |
| 334 | |
| 335 | class Meta: |
| 336 | db_table = 'profilers' |
| 337 | |
| 338 | class Admin: |
| 339 | list_display = ('name', 'description') |
| 340 | search_fields = ('name',) |
| 341 | |
| 342 | def __str__(self): |
| 343 | return self.name |
| 344 | |
| 345 | |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 346 | class AclGroup(dbmodels.Model, model_logic.ModelExtensions): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 347 | """\ |
| 348 | Required: |
| 349 | name: name of ACL group |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 350 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 351 | Optional: |
| 352 | description: arbitrary description of group |
| 353 | """ |
| 354 | name = dbmodels.CharField(maxlength=255, unique=True) |
| 355 | description = dbmodels.CharField(maxlength=255, blank=True) |
showard | 04f2cd8 | 2008-07-25 20:53:31 +0000 | [diff] [blame] | 356 | users = dbmodels.ManyToManyField(User, blank=True, |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 357 | filter_interface=dbmodels.HORIZONTAL) |
| 358 | hosts = dbmodels.ManyToManyField(Host, |
| 359 | filter_interface=dbmodels.HORIZONTAL) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 360 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 361 | name_field = 'name' |
| 362 | objects = model_logic.ExtendedManager() |
showard | eb3be4d | 2008-04-21 20:59:26 +0000 | [diff] [blame] | 363 | |
showard | 08f981b | 2008-06-24 21:59:03 +0000 | [diff] [blame] | 364 | @staticmethod |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 365 | def check_for_acl_violation_hosts(hosts): |
| 366 | user = thread_local.get_user() |
| 367 | if user.is_superuser(): |
| 368 | return None |
| 369 | accessible_host_ids = set( |
| 370 | host.id for host in Host.objects.filter(acl_group__users=user)) |
| 371 | for host in hosts: |
| 372 | # Check if the user has access to this host, |
| 373 | # but only if it is not a metahost |
| 374 | if (isinstance(host, Host) |
| 375 | and int(host.id) not in accessible_host_ids): |
| 376 | raise AclAccessViolation("You do not have access to %s" |
| 377 | % str(host)) |
| 378 | |
| 379 | def check_for_acl_violation_acl_group(self): |
| 380 | user = thread_local.get_user() |
| 381 | if user.is_superuser(): |
| 382 | return None |
| 383 | if not user in self.users.all(): |
| 384 | raise AclAccessViolation("You do not have access to %s" |
| 385 | % self.name) |
| 386 | |
| 387 | @staticmethod |
showard | 08f981b | 2008-06-24 21:59:03 +0000 | [diff] [blame] | 388 | def on_host_membership_change(): |
| 389 | everyone = AclGroup.objects.get(name='Everyone') |
| 390 | |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 391 | # find hosts that aren't in any ACL group and add them to Everyone |
showard | 08f981b | 2008-06-24 21:59:03 +0000 | [diff] [blame] | 392 | # TODO(showard): this is a bit of a hack, since the fact that this query |
| 393 | # works is kind of a coincidence of Django internals. This trick |
| 394 | # doesn't work in general (on all foreign key relationships). I'll |
| 395 | # replace it with a better technique when the need arises. |
| 396 | orphaned_hosts = Host.valid_objects.filter(acl_group__id__isnull=True) |
| 397 | everyone.hosts.add(*orphaned_hosts.distinct()) |
| 398 | |
| 399 | # find hosts in both Everyone and another ACL group, and remove them |
| 400 | # from Everyone |
| 401 | hosts_in_everyone = Host.valid_objects.filter_custom_join( |
| 402 | '_everyone', acl_group__name='Everyone') |
| 403 | acled_hosts = hosts_in_everyone.exclude(acl_group__name='Everyone') |
| 404 | everyone.hosts.remove(*acled_hosts.distinct()) |
| 405 | |
| 406 | |
| 407 | def delete(self): |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 408 | if (self.name == 'Everyone'): |
| 409 | raise AclAccessViolation("You cannot delete 'Everyone'!") |
| 410 | self.check_for_acl_violation_acl_group() |
showard | 08f981b | 2008-06-24 21:59:03 +0000 | [diff] [blame] | 411 | super(AclGroup, self).delete() |
| 412 | self.on_host_membership_change() |
| 413 | |
| 414 | |
showard | 04f2cd8 | 2008-07-25 20:53:31 +0000 | [diff] [blame] | 415 | def add_current_user_if_empty(self): |
| 416 | if not self.users.count(): |
| 417 | self.users.add(thread_local.get_user()) |
| 418 | |
| 419 | |
showard | 08f981b | 2008-06-24 21:59:03 +0000 | [diff] [blame] | 420 | # if you have a model attribute called "Manipulator", Django will |
| 421 | # automatically insert it into the beginning of the superclass list |
| 422 | # for the model's manipulators |
| 423 | class Manipulator(object): |
| 424 | """ |
| 425 | Custom manipulator to get notification when ACLs are changed through |
| 426 | the admin interface. |
| 427 | """ |
| 428 | def save(self, new_data): |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 429 | user = thread_local.get_user() |
showard | 31c570e | 2008-07-23 19:29:17 +0000 | [diff] [blame] | 430 | if hasattr(self, 'original_object'): |
| 431 | if (not user.is_superuser() |
| 432 | and self.original_object.name == 'Everyone'): |
| 433 | raise AclAccessViolation("You cannot modify 'Everyone'!") |
| 434 | self.original_object.check_for_acl_violation_acl_group() |
showard | 08f981b | 2008-06-24 21:59:03 +0000 | [diff] [blame] | 435 | obj = super(AclGroup.Manipulator, self).save(new_data) |
showard | 04f2cd8 | 2008-07-25 20:53:31 +0000 | [diff] [blame] | 436 | if not hasattr(self, 'original_object'): |
| 437 | obj.users.add(thread_local.get_user()) |
| 438 | obj.add_current_user_if_empty() |
showard | 08f981b | 2008-06-24 21:59:03 +0000 | [diff] [blame] | 439 | obj.on_host_membership_change() |
| 440 | return obj |
showard | eb3be4d | 2008-04-21 20:59:26 +0000 | [diff] [blame] | 441 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 442 | class Meta: |
| 443 | db_table = 'acl_groups' |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 444 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 445 | class Admin: |
| 446 | list_display = ('name', 'description') |
| 447 | search_fields = ('name',) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 448 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 449 | def __str__(self): |
| 450 | return self.name |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 451 | |
| 452 | # hack to make the column name in the many-to-many DB tables match the one |
| 453 | # generated by ruby |
| 454 | AclGroup._meta.object_name = 'acl_group' |
| 455 | |
| 456 | |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 457 | class JobManager(model_logic.ExtendedManager): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 458 | 'Custom manager to provide efficient status counts querying.' |
| 459 | def get_status_counts(self, job_ids): |
| 460 | """\ |
| 461 | Returns a dictionary mapping the given job IDs to their status |
| 462 | count dictionaries. |
| 463 | """ |
| 464 | if not job_ids: |
| 465 | return {} |
| 466 | id_list = '(%s)' % ','.join(str(job_id) for job_id in job_ids) |
| 467 | cursor = connection.cursor() |
| 468 | cursor.execute(""" |
| 469 | SELECT job_id, status, COUNT(*) |
| 470 | FROM host_queue_entries |
| 471 | WHERE job_id IN %s |
| 472 | GROUP BY job_id, status |
| 473 | """ % id_list) |
| 474 | all_job_counts = {} |
| 475 | for job_id in job_ids: |
| 476 | all_job_counts[job_id] = {} |
| 477 | for job_id, status, count in cursor.fetchall(): |
| 478 | all_job_counts[job_id][status] = count |
| 479 | return all_job_counts |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 480 | |
| 481 | |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 482 | def populate_dependencies(self, jobs): |
showard | 9a1f2e1 | 2008-10-02 11:14:29 +0000 | [diff] [blame] | 483 | if not jobs: |
| 484 | return |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 485 | job_ids = ','.join(str(job['id']) for job in jobs) |
| 486 | cursor = connection.cursor() |
| 487 | cursor.execute(""" |
showard | 56e9377 | 2008-10-06 10:06:22 +0000 | [diff] [blame] | 488 | SELECT jobs.id, labels.name |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 489 | FROM jobs |
| 490 | INNER JOIN jobs_dependency_labels |
| 491 | ON jobs.id = jobs_dependency_labels.job_id |
| 492 | INNER JOIN labels ON jobs_dependency_labels.label_id = labels.id |
| 493 | WHERE jobs.id IN (%s) |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 494 | """ % job_ids) |
showard | 56e9377 | 2008-10-06 10:06:22 +0000 | [diff] [blame] | 495 | job_dependencies = {} |
| 496 | for job_id, dependency in cursor.fetchall(): |
| 497 | job_dependencies.setdefault(job_id, []).append(dependency) |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 498 | for job in jobs: |
showard | 56e9377 | 2008-10-06 10:06:22 +0000 | [diff] [blame] | 499 | dependencies = ','.join(job_dependencies.get(job['id'], [])) |
| 500 | job['dependencies'] = dependencies |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 501 | |
| 502 | |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 503 | class Job(dbmodels.Model, model_logic.ModelExtensions): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 504 | """\ |
| 505 | owner: username of job owner |
| 506 | name: job name (does not have to be unique) |
| 507 | priority: Low, Medium, High, Urgent (or 0-3) |
| 508 | control_file: contents of control file |
| 509 | control_type: Client or Server |
| 510 | created_on: date of job creation |
| 511 | submitted_on: date of job submission |
| 512 | synch_type: Asynchronous or Synchronous (i.e. job must run on all hosts |
| 513 | simultaneously; used for server-side control files) |
| 514 | synch_count: ??? |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 515 | run_verify: Whether or not to run the verify phase |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 516 | synchronizing: for scheduler use |
showard | 3bb499f | 2008-07-03 19:42:20 +0000 | [diff] [blame] | 517 | timeout: hours until job times out |
showard | 542e840 | 2008-09-19 20:16:18 +0000 | [diff] [blame] | 518 | email_list: list of people to email on completion delimited by any of: |
| 519 | white space, ',', ':', ';' |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 520 | dependency_labels: many-to-many relationship with labels corresponding to |
| 521 | job dependencies |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 522 | """ |
showard | b1e5187 | 2008-10-07 11:08:18 +0000 | [diff] [blame^] | 523 | DEFAULT_TIMEOUT = global_config.global_config.get_config_value( |
| 524 | 'AUTOTEST_WEB', 'job_timeout_default', default=240) |
| 525 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 526 | Priority = enum.Enum('Low', 'Medium', 'High', 'Urgent') |
| 527 | ControlType = enum.Enum('Server', 'Client', start_value=1) |
| 528 | Status = enum.Enum('Created', 'Queued', 'Pending', 'Running', |
| 529 | 'Completed', 'Abort', 'Aborting', 'Aborted', |
showard | d823b36 | 2008-07-24 16:35:46 +0000 | [diff] [blame] | 530 | 'Failed', 'Starting', string_values=True) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 531 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 532 | owner = dbmodels.CharField(maxlength=255) |
| 533 | name = dbmodels.CharField(maxlength=255) |
| 534 | priority = dbmodels.SmallIntegerField(choices=Priority.choices(), |
| 535 | blank=True, # to allow 0 |
| 536 | default=Priority.MEDIUM) |
| 537 | control_file = dbmodels.TextField() |
| 538 | control_type = dbmodels.SmallIntegerField(choices=ControlType.choices(), |
showard | b1e5187 | 2008-10-07 11:08:18 +0000 | [diff] [blame^] | 539 | blank=True, # to allow 0 |
| 540 | default=ControlType.CLIENT) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 541 | created_on = dbmodels.DateTimeField(auto_now_add=True) |
| 542 | synch_type = dbmodels.SmallIntegerField( |
| 543 | blank=True, null=True, choices=Test.SynchType.choices()) |
| 544 | synch_count = dbmodels.IntegerField(blank=True, null=True) |
| 545 | synchronizing = dbmodels.BooleanField(default=False) |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 546 | run_verify = dbmodels.BooleanField(default=True) |
showard | b1e5187 | 2008-10-07 11:08:18 +0000 | [diff] [blame^] | 547 | timeout = dbmodels.IntegerField(default=DEFAULT_TIMEOUT) |
showard | 542e840 | 2008-09-19 20:16:18 +0000 | [diff] [blame] | 548 | email_list = dbmodels.CharField(maxlength=250, blank=True) |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 549 | dependency_labels = dbmodels.ManyToManyField( |
| 550 | Label, blank=True, filter_interface=dbmodels.HORIZONTAL) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 551 | |
| 552 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 553 | # custom manager |
| 554 | objects = JobManager() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 555 | |
| 556 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 557 | def is_server_job(self): |
| 558 | return self.control_type == self.ControlType.SERVER |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 559 | |
| 560 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 561 | @classmethod |
| 562 | def create(cls, owner, name, priority, control_file, control_type, |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 563 | hosts, synch_type, timeout, run_verify, email_list, |
| 564 | dependencies): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 565 | """\ |
| 566 | Creates a job by taking some information (the listed args) |
| 567 | and filling in the rest of the necessary information. |
| 568 | """ |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 569 | AclGroup.check_for_acl_violation_hosts(hosts) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 570 | job = cls.add_object( |
| 571 | owner=owner, name=name, priority=priority, |
| 572 | control_file=control_file, control_type=control_type, |
showard | 909c7a6 | 2008-07-15 21:52:38 +0000 | [diff] [blame] | 573 | synch_type=synch_type, timeout=timeout, |
showard | 542e840 | 2008-09-19 20:16:18 +0000 | [diff] [blame] | 574 | run_verify=run_verify, email_list=email_list) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 575 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 576 | if job.synch_type == Test.SynchType.SYNCHRONOUS: |
| 577 | job.synch_count = len(hosts) |
| 578 | else: |
| 579 | if len(hosts) == 0: |
| 580 | errors = {'hosts': |
| 581 | 'asynchronous jobs require at least' |
| 582 | + ' one host to run on'} |
| 583 | raise model_logic.ValidationError(errors) |
| 584 | job.save() |
showard | 989f25d | 2008-10-01 11:38:11 +0000 | [diff] [blame] | 585 | job.dependency_labels = dependencies |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 586 | return job |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 587 | |
| 588 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 589 | def queue(self, hosts): |
| 590 | 'Enqueue a job on the given hosts.' |
| 591 | for host in hosts: |
| 592 | host.enqueue_job(self) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 593 | |
| 594 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 595 | def abort(self): |
showard | 3dd47c2 | 2008-07-10 00:41:36 +0000 | [diff] [blame] | 596 | user = thread_local.get_user() |
| 597 | if not user.is_superuser() and user.login != self.owner: |
| 598 | raise AclAccessViolation("You cannot abort other people's jobs!") |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 599 | for queue_entry in self.hostqueueentry_set.all(): |
showard | 4c11904 | 2008-09-29 19:16:18 +0000 | [diff] [blame] | 600 | queue_entry.abort(user) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 601 | |
| 602 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 603 | def user(self): |
| 604 | try: |
| 605 | return User.objects.get(login=self.owner) |
| 606 | except self.DoesNotExist: |
| 607 | return None |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 608 | |
| 609 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 610 | class Meta: |
| 611 | db_table = 'jobs' |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 612 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 613 | if settings.FULL_ADMIN: |
| 614 | class Admin: |
| 615 | list_display = ('id', 'owner', 'name', 'control_type') |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 616 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 617 | def __str__(self): |
| 618 | return '%s (%s-%s)' % (self.name, self.id, self.owner) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 619 | |
| 620 | |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 621 | class IneligibleHostQueue(dbmodels.Model, model_logic.ModelExtensions): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 622 | job = dbmodels.ForeignKey(Job) |
| 623 | host = dbmodels.ForeignKey(Host) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 624 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 625 | objects = model_logic.ExtendedManager() |
showard | eb3be4d | 2008-04-21 20:59:26 +0000 | [diff] [blame] | 626 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 627 | class Meta: |
| 628 | db_table = 'ineligible_host_queues' |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 629 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 630 | if settings.FULL_ADMIN: |
| 631 | class Admin: |
| 632 | list_display = ('id', 'job', 'host') |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 633 | |
| 634 | |
showard | 7c78528 | 2008-05-29 19:45:12 +0000 | [diff] [blame] | 635 | class HostQueueEntry(dbmodels.Model, model_logic.ModelExtensions): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 636 | job = dbmodels.ForeignKey(Job) |
| 637 | host = dbmodels.ForeignKey(Host, blank=True, null=True) |
| 638 | priority = dbmodels.SmallIntegerField() |
| 639 | status = dbmodels.CharField(maxlength=255) |
| 640 | meta_host = dbmodels.ForeignKey(Label, blank=True, null=True, |
| 641 | db_column='meta_host') |
| 642 | active = dbmodels.BooleanField(default=False) |
| 643 | complete = dbmodels.BooleanField(default=False) |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 644 | deleted = dbmodels.BooleanField(default=False) |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 645 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 646 | objects = model_logic.ExtendedManager() |
showard | eb3be4d | 2008-04-21 20:59:26 +0000 | [diff] [blame] | 647 | |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 648 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 649 | def is_meta_host_entry(self): |
| 650 | 'True if this is a entry has a meta_host instead of a host.' |
| 651 | return self.host is None and self.meta_host is not None |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 652 | |
showard | 4c11904 | 2008-09-29 19:16:18 +0000 | [diff] [blame] | 653 | def log_abort(self, user): |
| 654 | abort_log = AbortedHostQueueEntry(queue_entry=self, aborted_by=user) |
| 655 | abort_log.save() |
| 656 | |
| 657 | def abort(self, user): |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 658 | if self.active: |
| 659 | self.status = Job.Status.ABORT |
showard | 4c11904 | 2008-09-29 19:16:18 +0000 | [diff] [blame] | 660 | self.log_abort(user) |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 661 | elif not self.complete: |
| 662 | self.status = Job.Status.ABORTED |
| 663 | self.active = False |
| 664 | self.complete = True |
showard | 4c11904 | 2008-09-29 19:16:18 +0000 | [diff] [blame] | 665 | self.log_abort(user) |
showard | b8471e3 | 2008-07-03 19:51:08 +0000 | [diff] [blame] | 666 | self.save() |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 667 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 668 | class Meta: |
| 669 | db_table = 'host_queue_entries' |
mbligh | e8819cd | 2008-02-15 16:48:40 +0000 | [diff] [blame] | 670 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 671 | if settings.FULL_ADMIN: |
| 672 | class Admin: |
| 673 | list_display = ('id', 'job', 'host', 'status', |
| 674 | 'meta_host') |
showard | 4c11904 | 2008-09-29 19:16:18 +0000 | [diff] [blame] | 675 | |
| 676 | |
| 677 | class AbortedHostQueueEntry(dbmodels.Model, model_logic.ModelExtensions): |
| 678 | queue_entry = dbmodels.OneToOneField(HostQueueEntry, primary_key=True) |
| 679 | aborted_by = dbmodels.ForeignKey(User) |
| 680 | aborted_on = dbmodels.DateTimeField(auto_now_add=True) |
| 681 | |
| 682 | objects = model_logic.ExtendedManager() |
| 683 | |
| 684 | class Meta: |
| 685 | db_table = 'aborted_host_queue_entries' |