argo_rm_subjobs.py 1.89 KB
Newer Older
jtchilders's avatar
jtchilders committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
from django.core.management.base import BaseCommand, CommandError
from django.conf import settings
from argo.UserJobReceiver import UserJobReceiver
from argo.models import ArgoSubJob
from common import Serializer
import os,sys,time,multiprocessing,shutil
import logging
logging.basicConfig(
                    level=logging.INFO, 
                    format='%(asctime)s %(levelname)s:%(name)s:%(message)s',
                   )
logger = logging.getLogger(__name__)



def myHandleError(self, record):
    """
    Handle errors which occur during an emit() call.

    This method should be called from handlers when an exception is
    encountered during an emit() call. If raiseExceptions is false,
    exceptions get silently ignored. This is what is mostly wanted
    for a logging system - most users will not care about errors in
    the logging system, they are more interested in application errors.
    You could, however, replace this with a custom handler if you wish.
    The record which was being processed is passed in to this method.
    """
    if raiseExceptions:
        ei = sys.exc_info()
        try:
            traceback.print_exception(ei[0], ei[1], ei[2], None, sys.stderr)
        except IOError:
            pass    # see issue 5971
        finally:
            del ei
        raise


class Command(BaseCommand):
   help = 'Remove a job stored in the DB'
   logger.debug('Remove a job in the ARGO Service DB.')

   def add_arguments(self,parser):
      parser.add_argument('--pk',dest='pk',nargs='+', type=int,help='remove the subjobs with the specified pks.',required=True)

   pass
   def handle(self, *args, **options):


      subjobs = ArgoSubJob.objects.filter(pk__in=options['pk'])
      
      for subjob in subjobs:
         logger.info('removing subjob: pk=' + str(subjob.pk) + ' subjob_id=' + str(subjob.subjob_id) + ' argo_job_id=' + str(subjob.argo_job_id) )
         subjob.delete()