Как вернуться с заявлением?

Question

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

def connect(params): 
    for user, passw, host in params: 
     try: 
      import pdb;pdb.set_trace() 
      with FTPHost(host, user, passw) as h: 
       return h 
     except FTPError as e: 
      logger.debug("Can't connect to ftp error is {}".format(e)) 
    else: 
     raise Exception(
      "Can't connect to ftp server with none of the {}".format(params) 
     ) 

Далее в коде я пытаюсь что-то вроде этого

host = connect(*args) 
host.walk() 

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

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

def connect(params, some_application_function): 
    for user, passw, host in params: 
     try: 
      import pdb;pdb.set_trace() 
      with FTPHost(host, user, passw) as host: 
       some_application_function(host) 
     except FTPError as e: 
      logger.debug("Can't connect to ftp error is {}".format(e)) 
    else: 
     raise Exception(
      "Can't connect to ftp server with none of the {}".format(params) 
     ) 

Но это кажется менее читаемым. Есть ли другие варианты?

Answer 1

я бы, вероятно, преобразовать connect функции в самом менеджеру контекста, который внутренне вызывает магические методы контекста вашего FTPHost управляющие:

class FTPConnector: 
    def __init__(self, params): 
     self.params = params 
    def __enter__(self): 
     for user, passw, host in params: 
      try: 
       # import pdb;pdb.set_trace() # not sure what you want to do with that... 
       self.h_context = FTPHost(host, user, passw) 
       return self.h_context.__enter__() 
      except FTPError as e: 
       logger.debug("Can't connect to ftp error is {}".format(e)) 
     else: 
      raise Exception(
       "Can't connect to ftp server with none of the {}".format(params) 
      ) 
    def __exit__(self, exc_type, exc_value, traceback): 
     return self.h_context.__exit(exc_type, exc_value, traceback) 

Теперь вы можете запустить это в вашей основной функции:

with FTPConnector(params) as host: 
    host.walk() 
    # do something useful with the connection here 

# here the connection gets closed again 

Answer 2

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

def connect(params): 
    for user, passw, host in params: 
     try: 
      import pdb;pdb.set_trace() 
      return FTPHost(host, user, passw) 
     except FTPError as e: 
      logger.debug("Can't connect to ftp error is {}".format(e)) 
    else: 
     raise Exception(
      "Can't connect to ftp server with none of the {}".format(params) 
     ) 

def main(): 
    with connect(params) as h: 
     do_something(h)