Как разнести функционал бота по файлам?

Question

[realfreshman]

Нужно один файл - функция get_data(), а во втором файле бот. Как это сделать?

import requests
import telebot
from auth_data import token
from bs4 import BeautifulSoup
import lxml
import telebot
from telebot import types


def get_data(x):                                                # Получение данных с school№.edu-penza (Адрес, Почта, Телефон)
    url = f"http://school{x}.edu-penza.ru/about/contacts/"
    print(url)
    response = requests.get(url)
    soup = BeautifulSoup(response.text, 'lxml', )

    value = soup.find_all("td")
    item_list = []

    for item in value:
        item_list.append(item.text.strip())


    dict_items = {}

    for items in item_list:
        if items == 'Адрес':
            dict_items['Адрес'] = item_list[item_list.index("Адрес") + 1]
        if items == 'E-mail':
            dict_items['E-mail'] = item_list[item_list.index("E-mail") + 1]
        if items == 'Телефон':
            dict_items['Телефон'] = item_list[item_list.index("Телефон") + 1]

        dict_items['Сайт'] = f'http://school{x}.edu-penza.ru'

    return dict_items


bot = telebot.TeleBot('5116629610:AAGaVs83TrbnUmHiwMLnjSBH0UaW4gcN2o4')

@bot.message_handler(commands=['start', 'help'])
def start_message(message):                         # Приветственное сообщение бота
    bot.send_message(message.chat.id, "Привет")


@bot.message_handler(content_types=['text'])
def get_text_messages(message):                     # Бот отправляет данные с сайта (используется функция get_data())
    if message.text == message.text:
        result = message.text.split()
        school_ticket = get_data(result[1])

        strings = []
        for key, item in school_ticket.items():
            strings.append("{}: {}".format(key.capitalize(), item))
        result = "\n".join(strings)


        bot.send_message(message.chat.id, result)


if __name__ == '__main__':
    bot.polling()

Answer 1

[Максим]

Создайте новый файл .py, закиньте туда get_data() и через import подключите ваш файл

Comments

[realfreshman]

а можешь кодом написать? я так и сделал но не работает бот

[Максим]

realfreshman, допустим у тебя файл с ботом называется file1, а второй файл с get_data() file2, тебе нужно в file1 прописать import file2. Далее можешь обращаться к методу таким образом: file2.get_data()

[realfreshman]

Максим, я просто импортировал get_data(). Но мне кажется что то надо поменять в if name...; Или где то прописать использование функции, но я не понимаю где

[shurshur]

realfreshman, перекидываем функцию get_data в data_functions.py (это пример, назвать можно и иначе). В основном файле пишем:

from data_functions import get_data

Если функций несколько, можно через запятую:

from data_functions import get_data, get_last_error

Или даже импортировать всё подряд (функции, классы, переменные...), но тут важно не нарваться на конфликт имён:

from data_functions import *

Конфликта имён можно избежать, если не импортировать имена из модуля в текущий контекст:

import data_functions 

def get_data():
  return ["some_data"]

data1 = data_functions.get_data()
data2 = get_data()

Если длинное название data_functions не нравится, можно сделать так:

import data_functions as df

data = df.get_data()

Можно и любое имя из файла импортировать с изменённым именем:

from data_functions import get_data as df_get_data

data = df_get_data()

Можно комбинировать:

import data_functions as df
from data_functions import get_data as secret_data_load

data = secret_data_load()
last_error = df.get_last_error()

В общем, можно много чего такого сделать по вкусу.

В if __name__ ... ничего менять не надо. Это по большому счёту не очень важно, просто если файл вдруг подключить как модуль, то __name__ будет иметь другое значение и bot.polling не будет запущен. Если есть уверенность, что никогда такой import не сделаешь, то bot.polling можно вызывать без if.