利用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 $?