| 4 | |
| 5 | |
| 6 | class HelloClient: |
| 7 | def __init__(self, endpoint): |
| 8 | self.client = Client(endpoint) |
| 9 | |
| 10 | def __enter__(self): |
| 11 | self.client.connect() |
| 12 | return self.client |
| 13 | |
| 14 | def __exit__(self, exc_type, exc_val, exc_tb): |
| 15 | self.client.disconnect() |
| 16 | |
| 17 | |
| 18 | if __name__ == '__main__': |