def setUp(self):
os.chdir(CURRENT_DIR)
self.test_name = os.path.basename(CURRENT_DIR)
self.test_info = load_test_info(self.test_name)
self.coll_name = self.test_info['coll_names'][0]
self.conn_src = self.test_info['conn_src']
self.conn_dest = self.test_info['conn_dest']
self.db_src = self.test_info['db_src']
self.db_dest = self.test_info['db_dest']
self.coll_src = self.db_src[self.coll_name]
self.coll_dest = self.db_dest[self.coll_name]
remove_res(self.test_info['temp_res'])
setup_dataset(
coll=self.coll_src,
dataset_file=os.path.join(CURRENT_DIR, 'data.json')
)
print_msg('Setting up progress file')
with open('current_progress.yaml', 'w') as output:
output.write("{}: 555317f7d290053143db668b\n".format(
self.coll_name
))
def tearDown(self):
print_msg("Dropping {} in source".format(
self.coll_name
))
self.coll_src.drop()
self.conn_src.close()
remove_res(self.test_info['temp_res'])
def tearDown(self):
print_msg("Dropping {} in source and destination".format(self.coll_name))
self.coll_src.drop()
self.coll_dest.drop()
self.conn_src.close()
self.conn_dest.close()
remove_res(self.test_info["temp_res"])
def tearDown(self):
print_msg("Dropping {} in source and destination".format(
self.coll_name))
self.coll_src.drop()
self.coll_dest.drop()
self.conn_src.close()
self.conn_dest.close()
remove_res(self.test_info['temp_res'])
def setUp(self):
os.chdir(CURRENT_DIR)
self.test_name = os.path.basename(CURRENT_DIR)
self.test_info = load_test_info(self.test_name)
self.coll_name = self.test_info['coll_names'][0]
self.conn_src = self.test_info['conn_src']
self.db_src = self.test_info['db_src']
self.db_dest = self.test_info['db_dest']
self.coll_src = self.db_src[self.coll_name]
remove_res(self.test_info['temp_res'])
setup_dataset(coll=self.coll_src,
dataset_file=os.path.join(CURRENT_DIR, 'data.json'))
def setUp(self):
os.chdir(CURRENT_DIR)
self.test_name = os.path.basename(CURRENT_DIR)
self.test_info = load_test_info(self.test_name)
self.coll_name = self.test_info["coll_names"][0]
self.conn_src = self.test_info["conn_src"]
self.conn_dest = self.test_info["conn_dest"]
self.db_src = self.test_info["db_src"]
self.db_dest = self.test_info["db_dest"]
self.coll_src = self.db_src[self.coll_name]
self.coll_dest = self.db_dest[self.coll_name]
remove_res(self.test_info["temp_res"])
setup_dataset(coll=self.coll_src, dataset_file=os.path.join(CURRENT_DIR, "data.json"))
def setUp(self):
os.chdir(CURRENT_DIR)
self.test_name = os.path.basename(CURRENT_DIR)
self.test_info = load_test_info(self.test_name)
self.coll_name = self.test_info['coll_names'][0]
self.conn_src = self.test_info['conn_src']
self.db_src = self.test_info['db_src']
self.db_dest = self.test_info['db_dest']
self.coll_src = self.db_src[self.coll_name]
remove_res(self.test_info['temp_res'])
setup_dataset(
coll=self.coll_src,
dataset_file=os.path.join(CURRENT_DIR, 'data.json')
)
def check_answer(message):
# Если функция возвращает None -> Человек не в игре
answer = utils.get_answer_for_user(message.chat.id)
# Если None:
if not answer:
if message.text == "Начать игру":
# Подключаемся к БД
db_worker = SQLighter(config.database_name)
# Получаем слачайную строку из БД
row = db_worker.select_single(random.randint(1, utils.get_rows_count()))
# Формируем разметку
markup = utils.generate_markup(row[2], row[3])
# отправляем аудиофайл с вариантами ответа
bot.send_voice(message.chat.id, row[1], reply_markup=markup)
# Включаем "игровой режим"
utils.set_user_game(message.chat.id, row[2])
# Отсоединяемся от БД
db_worker.close()
elif message.text == 'Cтатистика':
Res = utils.return_res(message.chat.id)
bot.send_message(message.chat.id, 'Ваш cчет: ' + str(Res))
else:
bot.send_message(message.chat.id, 'Чтобы начать игру, напишите команду /game')
else:
# Уберём клавиатуру с вариантами ответа, и вернём предыдущую.
markup = utils.menu()
# Если ответ правильный/неправильный
if message.text == answer:
bot.send_message(message.chat.id, 'Верно', reply_markup=markup)
res = True
else:
bot.send_message(message.chat.id, 'Не верно. Попробуйте еще раз!', reply_markup=markup)
res = False
utils.remove_res(message.chat.id, res)
# Удаляем юзера из хранилища (игра закончена)
utils.finish_user_game(message.chat.id)
def setUp(self):
os.chdir(CURRENT_DIR)
self.test_name = os.path.basename(CURRENT_DIR)
self.test_info = load_test_info(self.test_name)
self.coll_name = self.test_info['coll_names'][0]
self.conn_src = self.test_info['conn_src']
self.conn_dest = self.test_info['conn_dest']
self.db_src = self.test_info['db_src']
self.db_dest = self.test_info['db_dest']
self.coll_src = self.db_src[self.coll_name]
self.coll_dest = self.db_dest[self.coll_name]
remove_res(self.test_info['temp_res'])
setup_dataset(coll=self.coll_src,
dataset_file=os.path.join(CURRENT_DIR, 'data.json'))
print_msg('Setting up progress file')
with open('current_progress.yaml', 'w') as output:
output.write("{}: 555317f7d290053143db668b\n".format(
self.coll_name))
