diff options
-rw-r--r-- | bitbake/lib/toaster/toastermain/management/commands/builddelete.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/bitbake/lib/toaster/toastermain/management/commands/builddelete.py b/bitbake/lib/toaster/toastermain/management/commands/builddelete.py index 343d3114c0..ff93e549df 100644 --- a/bitbake/lib/toaster/toastermain/management/commands/builddelete.py +++ b/bitbake/lib/toaster/toastermain/management/commands/builddelete.py | |||
@@ -1,4 +1,5 @@ | |||
1 | from django.core.management.base import BaseCommand, CommandError | 1 | from django.core.management.base import BaseCommand, CommandError |
2 | from django.core.exceptions import ObjectDoesNotExist | ||
2 | from orm.models import Build | 3 | from orm.models import Build |
3 | from django.db import OperationalError | 4 | from django.db import OperationalError |
4 | import os | 5 | import os |
@@ -6,12 +7,16 @@ import os | |||
6 | 7 | ||
7 | 8 | ||
8 | class Command(BaseCommand): | 9 | class Command(BaseCommand): |
9 | args = "buildId" | 10 | args = '<buildID1 buildID2 .....>' |
10 | help = "Deletes selected build(s)" | 11 | help = "Deletes selected build(s)" |
11 | 12 | ||
12 | def handle(self, buildId, *args, **options): | 13 | def handle(self, *args, **options): |
13 | for bid in buildId.split(","): | 14 | for bid in args: |
14 | b = Build.objects.get(pk = bid) | 15 | try: |
16 | b = Build.objects.get(pk = bid) | ||
17 | except ObjectDoesNotExist: | ||
18 | print 'build %s does not exist, skipping...' %(bid) | ||
19 | continue | ||
15 | # theoretically, just b.delete() would suffice | 20 | # theoretically, just b.delete() would suffice |
16 | # however SQLite runs into problems when you try to | 21 | # however SQLite runs into problems when you try to |
17 | # delete too many rows at once, so we delete some direct | 22 | # delete too many rows at once, so we delete some direct |