def setup_class(self): config = configparser.ConfigParser() config.read('config.ini') try: db = mysql.connector.connect( host=config['db']['host'], user=config['db']['user'], password=config['db']['pass'], database=config['db']['database'] ) db.autocommit = True except mysql.connector.errors.ProgrammingError as e: print " ERROR: Can`t connect to MySQL server! ({0})".format(str(e)) exit(0) try: mc = MongoClient(host=config['mongo']['host'], port=int(config['mongo']['port'])) mongo_collection = getattr(mc, config['mongo']['collection']) except pymongo.errors.ConnectionFailure as e: print " ERROR: Can`t connect to MongoDB server! ({0})".format(str(e)) exit(0) R = Registry() R.set('config', config) R.set('db', db) R.set('mongo', mongo_collection) R.set('wr_path', wrpath) R.set('data_path', wrpath + '/data/') R.set('ndb', Database(config['db']['host'], config['db']['user'], config['db']['pass'], config['db']['database'])) R.set('proxies', Proxies()) self.db = R.get('ndb')
def check_backups(self): """ Simple check backups """ Http = Registry().get('http') check = [] result = [] dirs = file_to_list(Registry().get('wr_path') + "/bases/pre-backups-dirs.txt") dirs.append(self.options['host'].value) files = file_to_list(Registry().get('wr_path') + "/bases/pre-backups-files.txt") dirs_exts = file_to_list(Registry().get('wr_path') + "/bases/bf-dirs.txt") files_exts = file_to_list(Registry().get('wr_path') + "/bases/bf-files.txt") for _dir in dirs: for dir_ext in dirs_exts: check.append(dir_ext.replace('|name|', _dir)) for _file in files: for file_ext in files_exts: check.append(file_ext.replace('|name|', _file)) for url in check: r = Http.get(self.root_url + url) if not self._response_404(r): result.append({'url': self.root_url + url, 'code': r.status_code}) return result
def check_backups(self): """ Simple check backups """ Http = Registry().get('http') check = [] result = [] dirs = file_to_list(Registry().get('wr_path') + "/bases/pre-backups-dirs.txt") dirs.append(self.options['host'].value) files = file_to_list(Registry().get('wr_path') + "/bases/pre-backups-files.txt") dirs_exts = file_to_list(Registry().get('wr_path') + "/bases/bf-dirs.txt") files_exts = file_to_list(Registry().get('wr_path') + "/bases/bf-files.txt") for _dir in dirs: for dir_ext in dirs_exts: check.append(dir_ext.replace('|name|', _dir)) for _file in files: for file_ext in files_exts: check.append(file_ext.replace('|name|', _file)) for url in check: r = Http.get(self.root_url + url) if not self._response_404(r): result.append({ 'url': self.root_url + url, 'code': r.status_code }) return result
def check_dafs(self, _type): """ Simple dafs search """ Http = Registry().get('http') result = [] _dict = open(Registry().get('wr_path') + '/bases/pre-dafs-{0}.txt'.format(_type), 'r').read().split("\n") for obj in _dict: r = Http.get(self.root_url + obj) if r is not None and not self._response_404(r): result.append({'url': "/" + obj, 'code': r.status_code, 'time': 0}) return result
def check_dafs(self, _type): """ Simple dafs search """ Http = Registry().get('http') result = [] _dict = open( Registry().get('wr_path') + '/bases/pre-dafs-{0}.txt'.format(_type), 'r').read().split("\n") for obj in _dict: r = Http.get(self.root_url + obj) if r is not None and not self._response_404(r): result.append({ 'url': "/" + obj, 'code': r.status_code, 'time': 0 }) return result