Где мне указать db.connect(), чтобы он распространился на весь проект?

Question

[Dark Hacker]

Здравствуйте. Я использую библиотеку peewee в качеств ОРМ для бота. Когда там действие доходит до взаимодействия с таблицами, то есть, .get или .create, то это может занимать до 15-20 секунд. Я предполагаю, что это связано с открытием соединения с БД. Если нет, то поправьте меня, пожалуйста. Модели находятся в отдельном файле, они импортируются в другие нужные места и там происходят get/create методы. Я попробовал открыть соединение единожды в самих моделях единожды, потому что я не знаю, как оно будет себя вести при импортах и мало ли будет несколько соединений конфликтовать и лочить БД. Не кидайтесь тапками за такие непонятки)

Код типо инициализации БД

def initialize_db():
    global db_initialized
    if not db_initialized:  # Проверяем, инициализирована ли база данных
        db.connect()
        db.create_tables([Referrer, Channels, BotTexts], safe=True)
        db_initialized = True  # Устанавливаем флаг в True

Пример применения моделей

def start_message(message):
    chat_id = message.chat.id
    args = extract_arguments(message.text)

    if Referrer.does_referrer_exist(chat_id):
        bot.send_message(chat_id, BotTexts.get(text_id=3).text)
        return

    if args:
        if Referrer.is_already_reffered(chat_id):
            bot.send_message(chat_id, BotTexts.get(text_id=4).text)
            bot.set_state(chat_id, UserState.AWAITING_NICKNAME, chat_id)
            return
        else:
            user = Referrer.get(user_id=int(args))
            user.add_referred_user(chat_id)
            bot.send_message(args, BotTexts.get(text_id=5).text)

Если дело действительно в длительном соединении с БД, то в каком файле мне нужно это сделать, чтобы соединение было для всех взаимодействий? Если же не в этом дело, то в чем? Если вдруг надо дополнить инфу, код - пишите, дам всё, что надо. Кроме токена :)

Comments

[Vindicar]

Код вставляйте не скриншотами, а текстом. Того требуют правила сайта. Кнопка </> в помощь.

[Dark Hacker]

Vindicar, Это я знаю, я не был уверен, что конкретно мне вставлять, поэтому показал побольше ¯\_(ツ)_/¯ . Щас вставлю со скринов код текстом

[Сергей Горностаев]

это может занимать до 15-20 секунд. Я предполагаю, что это связано с открытием соединения с БД

Установка соединения с СУБД занимает меньше секунды, если СУБД в этот момент не подыхает.

я не знаю, как оно будет себя вести при импортах и мало ли будет несколько соединений конфликтовать и лочить БД

Зависит от того, какая СУБД и конкурентное ли окружение. Обычно делать соединение глобальным - плохая идея.

У библиотек доступа к БД часто есть пулы соединений, а библиотек написания ботов иногда бывают собственные средства инициализации состояния переда стартом.

[Dark Hacker]

Сергей Горностаев, Бот через время после запуска начинает себя прекрасно чувствовать, очень быстро работать, отвечать и в целом бодр. Возможно, ему просто необходимо время на инициализацию всего, что нужно. Мимолётный вопрос: peewee потянет 1к юзеров одновременно? Я, естественно, использую MySQL, а не sqlite, слышал, что мускуль гораздо более выносливый, по сравнению с младшим братом. Я первый раз делаю бота на такое предполагаемое кол0во пользователей. А так заказчик прогнозирует до 100к (не одновременно) пользователей бота в пик рекламы. В среднем 10к. Пользователям нет необходимости постоянно общаться с ботом, им вкинуть ник, если хотят, то забрать реферальную ссылку и в целом всё, больше им нет необходимости с ним взаимодействовать.

[Everything_is_bad]

Dark Hacker,

Пользователям нет необходимости постоянно общаться с ботом, им вкинуть ник, если хотят, то забрать реферальную ссылку и в целом всё, больше им нет необходимости с ним взаимодействовать.

каша в голове, все эти действия это общение с ботом.

. Когда там действие доходит до взаимодействия с таблицами, то есть, .get или .create, то это может занимать до 15-20 секунд

а теперь учись профилировать, чтобы понять что именно эти секунды занимает, хотят тут два варианта или сильная криворукость, или хостинг говно.

[Dark Hacker]

Everything_is_bad, Я локально его держу пока пишу его. Возможно, ему нужно время для инициализации, потому что опять же, он через время начинает работать шустро и без запинок.

[Everything_is_bad]

Dark Hacker, еще раз, учись профилировать, а вообще, может у тебя как у мамкиного хакера, коннекты касперский лочит.

Answer 1

[Евгений Хлебников]

залез в доку
не работал с данной либой, но скорее всего вам надо воспользоваться https://docs.peewee-orm.com/en/latest/peewee/api.h...

Create a context-manager that will hold open a connection for the duration of the wrapped block.

ну и как (нафантазирую) пример кода (напоминаю, что с данной либой я не работал и могу ошибаться, это просто согласно документации)

rom peewee import *
db = SqliteDatabase('people.db')

def on_app_startup():
    with db.connection_context():
         if  db.create_tables([Referrer, Channels, BotTexts], safe=True):
                db_initialized = True
         #выполнение функций которые должны работать в контексте базы, помня, что контекст закроется как только в коде вы выйдете за "with db.connection_context()"

Answer 2

[rickshaw]

Судя по примерам в первой странице доки надо делать объект db глобальным, и вызывать connect() при старте приложения

rom peewee import *

db = SqliteDatabase('people.db')

def main():
    db.connect()
    # запуск приложения
    # закрыть соединение (вероятно db.close())

возможно стоит закрывать соединение не в конце main, а по другому, но я не помню, как в питоне это стоит делать