diff options
author | Jesse Luehrs <doy@tozt.net> | 2014-10-16 23:28:46 -0400 |
---|---|---|
committer | Jesse Luehrs <doy@tozt.net> | 2014-10-16 23:28:46 -0400 |
commit | 8f9a3c5831afa6e2710bed3dbe3f2502cefc0ef0 (patch) | |
tree | f2ff4e93abde9e7406c9c8a08710d42c0f6380b8 /termcast_client/__init__.py | |
parent | 597219dbcddb3e49090a2b5ae883a1e2935bf836 (diff) | |
parent | 31cea814977c32b02bf9166cf4490f70e9861d1c (diff) | |
download | python-termcast-client-8f9a3c5831afa6e2710bed3dbe3f2502cefc0ef0.tar.gz python-termcast-client-8f9a3c5831afa6e2710bed3dbe3f2502cefc0ef0.zip |
Merge pull request #1 from thomasballinger/python2-compat
Add Python 2 compatibility
Diffstat (limited to 'termcast_client/__init__.py')
-rw-r--r-- | termcast_client/__init__.py | 6 |
1 files changed, 5 insertions, 1 deletions
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 ], }) |