From 31cea814977c32b02bf9166cf4490f70e9861d1c Mon Sep 17 00:00:00 2001
From: Thomas Ballinger <thomasballinger@gmail.com>
Date: Thu, 16 Oct 2014 21:36:20 -0400
Subject: Add Python 2 compatibility

---
 termcast_client/py2compat.py | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 termcast_client/py2compat.py

(limited to 'termcast_client/py2compat.py')

diff --git a/termcast_client/py2compat.py b/termcast_client/py2compat.py
new file mode 100644
index 0000000..16f2b34
--- /dev/null
+++ b/termcast_client/py2compat.py
@@ -0,0 +1,40 @@
+import fcntl
+import os
+import struct
+import sys
+import termios
+
+from collections import namedtuple
+
+terminal_size = namedtuple('terminal_size', ['columns', 'lines'])
+
+py2 = sys.version_info[0] == 2
+
+def get_terminal_size(fallback=(80, 24)):
+    """Based on Python 3 os.get_terminal_size()"""
+    # columns, lines are the working values
+    try:
+        columns = int(os.environ['COLUMNS'])
+    except (KeyError, ValueError):
+        columns = 0
+
+    try:
+        lines = int(os.environ['LINES'])
+    except (KeyError, ValueError):
+        lines = 0
+
+    # only query if necessary
+    if columns <= 0 or lines <= 0:
+        try:
+            winsize = fcntl.ioctl(sys.__stdout__.fileno(),
+                                  termios.TIOCGWINSZ, '\000' * 8)
+            size = terminal_size(*struct.unpack('hhhh', winsize)[1::-1])
+
+        except (NameError, OSError):
+            size = terminal_size(*fallback)
+        if columns <= 0:
+            columns = size.columns
+        if lines <= 0:
+            lines = size.lines
+
+    return terminal_size(*size)
-- 
cgit v1.2.3-54-g00ecf