commited neotuli's patch for source links

1 files changed, 11 insertions, 7 deletions

diff --git a/tupkg/client/tupkg b/tupkg/client/tupkg
index a96923e..380dc53 100755
--- a/tupkg/client/tupkg
+++ b/tupkg/client/tupkg
@@ -159,17 +159,21 @@ def main(argv=None):
 			return 1
 
 	cs = ClientSocket(files, confdict['host'], confdict['port'], confdict['user'], confdict['password'])
-	cs.connect()
+	try:
+		cs.connect()
 
-	if not cs.auth():
-		print "Error authenticating you, you bastard"
-		return 1
+		if not cs.auth():
+			print "Error authenticating you, you bastard"
+			return 1
 
-	cs.sendFileMeta()
+		cs.sendFileMeta()
 
-	cs.sendFiles()
+		cs.sendFiles()
 
-	cs.close()
+		cs.close()
+	except KeyboardInterrupt:
+		print "Cancelling"
+		cs.close()
 
 	return 0