Как импортировать нужную функцию из другого файла .py?

Question

[Sadons]

Пишу асинхронного бота, возникла необходимость смены асинхронного aiosqlite на асинхронный postgresql(asyncpg)
Возникла проблема, мне нужно импортировать в данном случае sql внутри функции create_bot_database()

async def create_bot_database():
    sql = await asyncpg.connect(
        host=HOST,
        user=USER,
        password=PASSWORD,
        database=DB_NAME
    )

есть ли такая возможность как то вытянуть с функции данную переменную?
Или же нужно заново прописывать коннект в каждом файле, так как вся структура "проекта" разбита на несколько функций
Посредством импорта я могу вытянуть оттуда только саму функцию - create_bot_database()
А мне нужно переменную внутри функции - sql

from database.create_db import create_bot_database

async def startup(message: types.Message):  # Здороваемся
    await sql.execute(f'SELECT id FROM users WHERE id = "{message.from_user.id}"')  # Проверяем есть ли в базе
    if await sql.fetchone() is None:
        await bot.send_message(message.from_user.id,
                               "Добро пожаловать!",
                               "Перед тем, как начать работу нужно согласиться с правилами:",
                               parse_mode="html", reply_markup=kb_reg)
        await registration(message)
    else:
        await bot.send_message(message.from_user.id, "Вы уже зарегистрированы!", reply_markup=kb_main)

Это функция регистрации, мне нужно тут вызвать как раз функцию sql, но как мне её импортировать?
Возможно не понятно обьяснил, извините заранее

Comments

[Алан Гибизов]

Пользуйтесь кнопкой </> для удобства указания типа кода

Answer 1

[pASdfsa]

import "название файла"
"название файла".test()

Comments

[Sadons]

всё верно, таким образом я импортирую функцию - которая создаёт таблицу если её нет. Но мне нужно, вытянуть с функции только переменную sql

[Vindicar]

Sadons,

вытянуть с функции только переменную sql

Выдаёт непонимание основ языка. "Вытянуть" переменную из функции нельзя.
Её можно вернуть (через return), и тогда для получения значения можно будет вызывать функцию - но тогда при каждом вызове функции будет создаваться новое подключение.
Можно переменную сделать глобальной, т.е. объявить её вне функции и сразу же присвоить значение. Тогда подключение создастся при запуске/импорте файла, где переменная объявлена, и любой файл, который импортирует этот файл, сможет к этой переменной (а значит, к подключению) обратиться.
Наконец, можно переменную сделать глобальной, задать ей значение None, потом внести её внутрь функции через оператор global, и там присвоить нужное значение. Тогда подключение создастся только при запуске функции, а для получения подключения также нужно будет обратиться к переменной.

Какое из этих поведений удобнее для тебя - тебе решать.

Answer 2

[leonidvoroninthebest]

import 'название файла'
или from 'название файла' import 'нужная его часть'