slider-core/src/test/python/agent.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


import logging
import sys
import datetime
import time
from optparse import OptionParser
import os


# A representative Agent code for the embedded agent
def main():
  print "Executing src/test/python/agent.py"
  try:
    print 'Argument List: {0}'.format(str(sys.argv))
  except AttributeError:
    pass

  parser = OptionParser()
  parser.add_option("--log", dest="log_folder", help="log destination")
  parser.add_option("--config", dest="conf_folder", help="conf folder")
  parser.add_option('--sleep', dest='sleep', help='sleep time')
  parser.add_option('--exitcode', dest='exitcode', help='exit code to return')

  (options, args) = parser.parse_args()

  if options.log_folder:
    log_file_name = "echo" + str(datetime.datetime.now()) + ".log"
    log_file_path = os.path.join(options.log_folder, log_file_name)
    logging.basicConfig(filename=log_file_path, level=logging.DEBUG)
    print log_file_path
  logging.debug('Starting echo script ...')

  logging.info("Number of arguments: %s arguments.", str(len(sys.argv)))
  logging.info("Argument List: %s", str(sys.argv))
  sleeptime = 30
  if options.sleep:
    sleeptime = int(options.sleep)
  if sleeptime > 0:
    logging.info("Sleeping for %d seconds", sleeptime)
    time.sleep(sleeptime)
  exitcode = 0
  if options.exitcode:
    exitcode = int(options.exitcode)
  return exitcode


if __name__ == "__main__":
  sys.exit(main())