from tortoise import Tortoise, fields
from tortoise.models import Model
from vkbottle import Bot
from vkbottle import Message

bot = Bot("25015660b4d5f716e63b1cfc5395d5a59b60652cb880054a256b517e8edc23cc20c497bea709f3f651b82")


class Chats(Model):
    chat_id = fields.IntField(pk=True)

    class Meta:
        table = "Chats"


class Users(Model):
    fk_chat_id = fields.relational.ForeignKeyField('models.Chats', related_name='fk_chat', to_field='chat_id')
    uid = fields.IntField()


async def init():
    await Tortoise.init(
        db_url='sqlite://db.sqlite3',
        modules={'models': ["__main__"]}
    )
    await Tortoise.generate_schemas()


async def create(chat_id, uid):
    await Chats.create(chat_id=chat_id)
    await Users.create(fk_chat_id=chat_id, uid=uid)


@bot.on.message_handler(text=['db create'])
async def wrapper(answer: Message):
    # try:
    await create(chat_id=answer.peer_id, uid=answer.from_id)
    return 'Database was created!'
    # except:
    # return 'Database wasn\'t created :C'


@bot.on.message_handler(text=['db show'])
async def wrapper(answer: Message):
    return await Chats.all().values('id')



bot.run_polling(on_startup=init)