script: Encapsuled Connection into with

4 years ago · 45e97f8523
parent 3b91105aea
commit 45e97f8523
1 changed files with 3 additions and 3 deletions
--- a/scripts/tinytinypy
+++ b/scripts/tinytinypy
@ -69,9 +69,9 @@ def main():
    host = urlM['host']
    # Call api
    # TODO Support port
-    server = Connection(proto=proto, host=host)
-    server.login(args.user, passwd)
-    args.func(server, args)
+    with Connection(proto=proto, host=host) as server:
+        server.login(args.user, passwd)
+        args.func(server, args)

 if __name__ == "__main__":
    main()