aboutsummaryrefslogtreecommitdiffstats
path: root/termcast_client/__init__.py
diff options
context:
space:
mode:
authorJesse Luehrs <doy@tozt.net>2014-10-16 23:28:46 -0400
committerJesse Luehrs <doy@tozt.net>2014-10-16 23:28:46 -0400
commit8f9a3c5831afa6e2710bed3dbe3f2502cefc0ef0 (patch)
treef2ff4e93abde9e7406c9c8a08710d42c0f6380b8 /termcast_client/__init__.py
parent597219dbcddb3e49090a2b5ae883a1e2935bf836 (diff)
parent31cea814977c32b02bf9166cf4490f70e9861d1c (diff)
downloadpython-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__.py6
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 ],
})