with文から復帰するには？

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

私はおそらく内部的にあなたのFTPHOSTコンテキストマネージャの魔法のメソッド呼び出しコンテキストマネージャ自体にあなたのconnect機能を変換します：

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

# here the connection gets closed again 

：

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) 

を今すぐあなたのメイン機能でこれを実行することができ

Answer 2

おそらくwith文をconnect関数の外に移動しますか？

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)