Revision 5ce3ce4f snf-astakos-app/astakos/im/management/commands/resource_remove.py
| b/snf-astakos-app/astakos/im/management/commands/resource_remove.py | ||
|---|---|---|
| 36 | 36 |
|
| 37 | 37 |
from astakos.im.models import Resource |
| 38 | 38 |
|
| 39 |
|
|
| 39 | 40 |
class Command(BaseCommand): |
| 40 | 41 |
args = "<resource>" |
| 41 | 42 |
help = "Add a resource" |
| 42 |
|
|
| 43 |
|
|
| 43 | 44 |
def handle(self, *args, **options): |
| 44 | 45 |
if len(args) < 1: |
| 45 | 46 |
raise CommandError("Invalid number of arguments")
|
| 46 |
|
|
| 47 |
|
|
| 47 | 48 |
kwargs = {}
|
| 48 | 49 |
if args[0].isdigit(): |
| 49 |
kwargs['id']=args[0]
|
|
| 50 |
kwargs['id'] = args[0]
|
|
| 50 | 51 |
else: |
| 51 |
kwargs['name']=args[0]
|
|
| 52 |
|
|
| 52 |
kwargs['name'] = args[0]
|
|
| 53 |
|
|
| 53 | 54 |
try: |
| 54 | 55 |
r = Resource.objects.get(**kwargs) |
| 55 | 56 |
except Resource.DoesNotExist, e: |
| 56 | 57 |
raise CommandError("Invalid resource")
|
| 57 |
r.delete() |
|
| 58 |
r.delete() |
|
Also available in: Unified diff