Можно ли вызвать функцию так, чтобы перед её выполнением выполнилась ещё одна функция класса?

Question

[Роман Бурч]

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

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

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

Answer 1

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

В самом простом случае вызвать метод открытия из метода чтения.

Наверное лучше сделать метод-декоратор, который до выполнения декорируемого метода открывает файл, а после - закрывает.

Comments

[Роман Бурч]

Находясь внутри класса, при обычном чтении он просит сослаться на ссылку self

[Роман Бурч]

Я тоже думаю о декораторах, но делая декоратор я получаю ошибку 'function' object has no attribute 'file_name'. Файлнейм находитися в __инит__ под self.file_name

[Роман Бурч]

Client_connect.__call__(self)

Вот так заработало. Спасибо за подсказку

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

Роман Бурч, можно, но сомнительно.
Покажите пример класса.

[Роман Бурч]

Алан Гибизов,

class Client_connect:
    def __init__(self):
        self.dbname = os.getenv('dbname')
        self.user = os.getenv('user')
        self.password = os.getenv('password')
        self.host = os.getenv('host')

    def __call__(self):
        try:
            self.conn = psycopg2.connect(dbname=self.dbname,
                                         user=self.user,
                                         password=self.password,
                                         host=self.host)
            print("[INFO] подключение к PostgreSQL")
        except Exception as e:
            print(f"[ERROR - def __call__] Ошибка получения доступа к базе данных: {e}")

    def user_exists(self, user_id):
        Client_connect.__call__(self)
        """Проверяем на наличие юзера в базе данных"""
        try:
            with self.conn.cursor() as cursor:

                cursor.execute("SELECT id FROM users WHERE tele_user_id = %s", (user_id, ))
                print("[INFO] Данные успешно получены")
                return bool(len(cursor.fetchall()))

        except Exception as e:
            print(f"[ERROR] Ошибка работы базы данных: {e}")
        finally:
            if self.conn:
                self.conn.close()
                print("[INFO] PostgreSQL соединение закрыто")

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

Роман Бурч, в общем, я бы всё делал по-другому с точки зрения ООП.

Класс должен отображать объекты. А Client_connection это как-бы не совсем объект. Ну, точнее connection можно считать объектом, но тогда у него должны быть методы создать и удалить. Возможно, еще проверить корректность. Ну и в таком духе.

А user_exist это как-то не совсем действие, которое может происходить с объектом типа connection. Этот метод в классе connection чужой. Это должен быть метод класса User. Или что-то в таком духе.

Из-за всей этой чехарды self в разных местах оказывается не тем, что по-идее должно быть по смыслу.
И код становится путаным, и классы мешают, а не помогают.

Кроме того, все Exception надо бы конкретизировать.

[Роман Бурч]

Алан Гибизов, огромное вам спасибо за ваш комментарий, это даст мне больше знаний и понимания! От чистого сердца, благодарю)

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

Роман Бурч, знаете… я долго вкатывался в ООП и еще мне вкатываться и вкатываться. Недавно я обнаружил хороший, как мне кажется, пример реализации ООП для Selenium. Он мне очень помог переосмыслить некоторые вещи. Я прям свой код полностью под это рефакторил, больше полтыщи строк в мусорку…

[Роман Бурч]

Алан Гибизов, Спасибо большое, я очень рад любой информации, которая сможет сделать мой код и путь в этом направлении лучше)

Answer 2

[Aragorn]

В чем проблема сделать так?

def read_file(self):
      self.open_file()

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

class WorkFile:
    def __init__(self):
        self.open_file()
    def open_file(self):
         # ваш код

Comments

[Роман Бурч]

Да, я так и сделал из-за этого файл открывается при запуске программы (мне так не надо). Но после каждого чтения я закрываю файл. Когда я вызываю другую функцию, файл остаётся закрытым и я получаю ошибку, так как я не открыл файл

[Aragorn]

Роман Бурч, ну так и не закрывайте его после каждого чтения. Закройте его при завершении программы

class WorkFile:
    def __init__(self):
        self.open_file()
    def open_file(self):
         # ваш код
    def __del__(self):
        self.close_file() # закроет файл при завершении программы

[Роман Бурч]

Я раньше так и делал, просто очень часто слышал что это не надёжно и хорошим тоном считается всегда закрывать файл.
Я вызвал открытие таким образом и всё заработало
Client_connect.__call__(self)

[shurshur]

Роман Бурч, можно просто помнить, открыт файл или закрыт. Например, заменяя объект файла на None при закрытии:

def __init__(self, filename):
    self.filename = filename
    self.file_handle = None

def open_file(self):
    if self.file_handle is None:
        self.file_handle = open(self.filename, "r")

def close_file(self):
    if self.file_handle is not None:
        self.file_handle.close()
        self.file_handle = None

Тогда можно свободно вызывать open_file и close_file где угодно, не беспокоясь, что будут совершаться неверные действия.

[Роман Бурч]

shurshur, ваш способ очень интересный. Так и правда можно не бояться неправильных действий