Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from tortoise import Tortoise, fields
- from tortoise.models import Model
- from vkbottle import Bot
- from vkbottle import Message
- bot = Bot("25015660b4d5f716e63b1cfc5395d5a59b60652cb880054a256b517e8edc23cc20c497bea709f3f651b82")
- class Chats(Model):
- id = fields.IntField(pk=True)
- chat_id = fields.IntField()
- class Meta:
- table = "Chats"
- class Users(Model):
- id = fields.IntField(pk=True)
- fk_chat_id = fields.relational.ForeignKeyField('models.Chats',
- related_name='fk_chat_id', on_delete='CASCADE', to_field='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=1, 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('chat_id')
- bot.run_polling(on_startup=init)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement