def __init__(self, app, test_type, count, users, vverbose): try: parser = scp() dir_path = os.path.dirname(__file__) config_path = os.path.join(dir_path, r"../mainConfig.ini") parser.read(config_path) self.http = httplib2.Http() self.host = parser.get('hostinfo', 'host') self.email = parser.get('userinfo', 'email') self.password = parser.get('userinfo', 'password') self.customerid = parser.get('userinfo', 'customerid') self.openid = parser.get('userinfo', 'openid') self.appid = parser.get('appinfo', 'appid') self.app = app self.event_id = self.__get_eventid() if self.app not in [ 'nine', 'xmas', 'chun', 'rotate', 'lottery', 'scratch', 'box', 'face', 'shake']: raise ParamError(self.app) self.test_type = test_type if self.test_type not in ['page', 'draw']: raise ParamError(self.test_type) # self.count = int(count) # self.users = int(users) self.queue = self.__generate_queue(int(count), int(users)) self.vverbose = vverbose except ConfigParser.Error, e: lg.error("StressTest init failed.") lg.error("Exception: %s" % e) raise e
from ConfigParser import SafeConfigParser as scp _config = scp() from autoworkup import *