From 31cea814977c32b02bf9166cf4490f70e9861d1c Mon Sep 17 00:00:00 2001 From: Thomas Ballinger Date: Thu, 16 Oct 2014 21:36:20 -0400 Subject: Add Python 2 compatibility --- termcast_client/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'termcast_client/__init__.py') diff --git a/termcast_client/__init__.py b/termcast_client/__init__.py index 0bab15b..70d1798 100644 --- a/termcast_client/__init__.py +++ b/termcast_client/__init__.py @@ -9,6 +9,7 @@ import ssl import sys from . import pity +from . import py2compat class Client(object): def __init__(self, host, port, username, password, tls, fingerprint): @@ -76,7 +77,10 @@ class Client(object): return self._build_winsize_metadata_string() def _build_winsize_metadata_string(self): - size = shutil.get_terminal_size() + if py2compat.py2: + size = py2compat.get_terminal_size() + else: + size = shutil.get_terminal_size() return self._build_metadata_string({ "geometry": [ size.columns, size.lines ], }) -- cgit v1.2.3