利用python编写不同环境下都能运行的测试脚本

利用bash来获取当前电脑的环境变量，可以写一个.sh文件，里面获取当前环境，然后在调用python文件执行

 # -*- coding: utf-8 -*-

 import logging
 import urllib
 import sys
 import json
 import string
 import os
 import time
 import getopt
 import requests

 sys.path.append('../')

 from common.monitor_logging import *
 from common.util import *
 from common.nagios_status import *
 from hongbao_api.baidu_api import *
 from common.monitor_config import *
 from common.monitor_testcase import *
 from common.pinger import *

 def Usage():
     print "exception"

 DEBUG_FLAG=True
 ROOT_DIR=''

 def get_user_opt():
     global DEBUG_FLAG, ROOT_DIR
     if len(sys.argv) > 1:
         try:
             opts, args = getopt.getopt(sys.argv[1:], 'dr:h')
             if len(opts) == 0:
                 return

             for opt, arg in opts:
                 if opt in ('-h', '--help'):
                     Usage()
                     sys.exit(3)
                 elif opt == '-d':
                     DEBUG_FLAG = True
                 elif opt == '-r':
                     ROOT_DIR = arg

         except getopt.GetoptError:
             Usage()
             sys.exit(-1)

 '''

 '''
 class ScoreAuctionDealsUrl():
     def __init__(self, testcase_file):
         self.config = MonitorConfig()
         #获取需要解析的xml的testcase的name
         self.testcase = MonitorTestcase(testcase_file)
         #日志文件
         self.logger = logging.getLogger('baidu')

     #打印返回值和退出
     def print_result_and_exit(self, status, msg):
         print msg
         sys.exit(status)
     #检查主机是否联网
     def check_host_is_alive(self):
         pinger = Pinger()
         ret = pinger.check()
         return ret

     #开始检测url
     def test_all_category(self):
         #if not self.check_host_is_alive():
         #    msg = 'Network is not availabe'
         #    return STATE_UNKNOWN, msg
         testcase_name = 'baidu'
         url=self.testcase.get_url(testcase_name)
         self.logger.debug('testcase=%s' % testcase_name)
         self.logger.info('url=%s' % url)

         #请求数据
         deals=GetBaiduDeals(url)
         #请求json获取返回的状态
         status = deals.get_baidu()
         print("status:%s"%status)
         #一个个遍历：
         json_data=deals.json_data
         print(json_data)

         return STATE_WARNING, "aaaa"

         if status == 0:
             msg0 = 'return status is 0'
             return STATE_WARNING, msg0
         elif status ==1:
             msg1 = 'return status is 1'
             return STATE_WARNING,msg1
         elif status ==2:
             msg2 = 'return status is 2'
             return STATE_CRITICAL,msg2
         else:
             msg3='return status is 3'
             return STATE_UNKNOWN,msg3

 if __name__ == '__main__':

       #######################################################
     get_user_opt()
     print(DEBUG_FLAG)
     print(ROOT_DIR)
     #
     if DEBUG_FLAG:
         print 'DEBUG_FLAG is True'
         log_console_flag = True
     else:
         print 'DEBUG_FLAG is False'
         log_console_flag = False

     if ROOT_DIR:
         log_file = ROOT_DIR + '/log/monitor_log.txt'
     else:
         log_file= os.getcwd() + '/../../log/monitor_log.txt'
     log_level = logging.DEBUG
     set_logging(log_file, log_level, log_console_flag)

     logger = logging.getLogger("hongbao")
     print(logger)
     #
     ########################################################
     logger.info('start...')
     testcase_file = '../../test/hongbao.xml'
     print(testcase_file)
     score_auction_deals_url = ScoreAuctionDealsUrl(testcase_file)
     print(score_auction_deals_url)
     status, msg = score_auction_deals_url.test_all_category()
     print(status,msg)
     #
     status_str = get_status_str(status)
     output_msg = '%s, %s' % (status_str, msg)
     print output_msg
     exit(status)

#!/bin/sh

_DEBUG=false

# input env
shell_dir=$(cd "$(dirname "$0")"; pwd)
source ${shell_dir}/set_env.sh _DEBUG

ROOT_DIR="$(dirname ${shell_dir})"
_log "ROOR_DIR=${ROOT_DIR}"

#LOG_FILE=${ROOT_DIR}/log/report.log
#_log "LOG FILE=${LOG_FILE}"

cd ${ROOT_DIR}/src/hongbao

python baidu_url.py -r ${ROOT_DIR} $*
exit $?