def test_create_friendship(self):
db = User()
f = Friendship()
db.register_user("alex", "123", self.engine)
db.register_user("ben", "123", self.engine)
f1 = db.login_user("ben", "123", self.engine)
f2 = db.login_user("alex", "123", self.engine)
self.assertTrue(f.create_friendship(f1, 'alex', self.engine))
def test_send_friend_request(self):
db = User()
r = FriendRequest()
db.init_data_base(self.engine)
db.register_user("ben", "123", self.engine)
db.register_user("alex", "123", self.engine)
f1 = db.login_user("ben", "123", self.engine)
f2 = db.login_user("alex", "123", self.engine)
waiting = r.check_waiting_request(f1, self.engine)
self.assertEqual(len(waiting), 0)
self.assertTrue(r.send_friend_request(f2, 'ben', self.engine))
waiting = r.check_waiting_request(f1, self.engine)
self.assertEqual(len(waiting), 1)
self.assertTrue(r.denial_request(f1, 'alex', self.engine))
waiting = r.check_waiting_request(f1, self.engine)
self.assertEqual(len(waiting), 0)
self.assertTrue(r.send_friend_request(f2, 'ben', self.engine))
self.assertTrue(r.confirm_request(f1, 'alex', self.engine))
self.assertEqual(len(waiting), 0)
def setUp(self):
self.engine = create_engine('sqlite:///:memory:')
db = User()
db.init_data_base(self.engine)
def test_update_ip(self):
db = User()
db.register_user("ben", "1234", self.engine)
t = db.login_user("ben", "1234", self.engine)
self.assertTrue(db.set_user_ip(t, "127.0.0.1", self.engine))
def test_login(self):
db = User()
db.register_user("ben1", "1234", self.engine)
self.assertTrue(db.login_user("ben1", "1234", self.engine))
def test_register(self):
db = User()
self.assertTrue(db.register_user("ben", "123", self.engine))
import telebot, time
from random import choice
from telebot import types
import config
from data_base import User
bot = telebot.TeleBot(config.TOKEN)
db = User("data.db")
undone_time = lambda x: time.strftime("%H:%M:%S %d.%m.%Y", time.localtime(x))
bad_advice_list = db.copy_bad_advice()
choice_advice_list = db.copy_choice_advice()
rest_list = db.copy_rest()
helper_list = db.copy_helper()
@bot.message_handler(commands=['start'])
def menu(message):
bot.send_message(message.chat.id,
text='Вот что я умею: ',
reply_markup=keyboard())
@bot.message_handler(content_types=['text'])
def message_handler(message):
user_id = message.from_user.id
stat = undone_time(message.date)
if (not db.get_all_id(user_id)):
db.add_id_to_db(user_id)
