From e83c6629858f49f674cdf65fadad4b02a5a60c1c Mon Sep 17 00:00:00 2001
From: Nate Levesque <nate@thenaterhood.com>
Date: Wed, 21 Sep 2016 21:38:04 -0400
Subject: [PATCH] Fix classes to work with Python 2.7

---
 src/basically_ti_basic/__main__.py          |  6 ++++--
 src/basically_ti_basic/compiler/__init__.py | 21 +++++++++++++++++++--
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/src/basically_ti_basic/__main__.py b/src/basically_ti_basic/__main__.py
index 8baa8eb..dca658e 100644
--- a/src/basically_ti_basic/__main__.py
+++ b/src/basically_ti_basic/__main__.py
@@ -9,7 +9,8 @@ def compile_file(inputfile, outputfile):
         for line in f:
             file_lines.append(line)
 
-    compiled_file = PrgmCompiler.compile(file_lines)
+    compiler = PrgmCompiler()
+    compiled_file = compiler.compile(file_lines)
     if outputfile == "stdout":
         print("".join(compiled_file.prgmdata))
     else:
@@ -18,7 +19,8 @@ def compile_file(inputfile, outputfile):
 def decompile_file(inputfile, outputfile):
     tifile = TIPrgmFile(inputfile)
 
-    decompiled = PrgmCompiler.decompile(tifile)
+    compiler = PrgmCompiler()
+    decompiled = compiler.decompile(tifile)
     if outputfile == 'stdout':
         print("\n".join(decompiled))
     else:
diff --git a/src/basically_ti_basic/compiler/__init__.py b/src/basically_ti_basic/compiler/__init__.py
index c5ecccd..e6acc32 100644
--- a/src/basically_ti_basic/compiler/__init__.py
+++ b/src/basically_ti_basic/compiler/__init__.py
@@ -8,7 +8,7 @@ class PrgmCompiler(object):
     program files
     """
 
-    def compile(raw_text):
+    def compile(self, raw_text=None):
         """
         Compiles to 8Xp format. This logic works, but the TIFile class is
         incomplete so the file may not work properly on the TI calculator.
@@ -18,6 +18,14 @@ def compile(raw_text):
         Returns:
             TIFile
         """
+        # To be compatible with Python 2.7 without changing
+        # the public class API, we do some gross magic. This
+        # has a side effect that we can use this method either
+        # as a static method or not.
+        # FIXME
+        if not isinstance(self, PrgmCompiler):
+            raw_text = self
+
         tifile = TIPrgmFile()
         tifile.prgmdata = []
         tokens = get_inverse_tokens()
@@ -45,7 +53,7 @@ def compile(raw_text):
         return tifile
 
 
-    def decompile(tifile):
+    def decompile(self, tifile=None):
         """
         Decompiles to plaintext.
 
@@ -54,6 +62,15 @@ def decompile(tifile):
         Returns:
             Array[string]
         """
+
+        # To be compatible with Python 2.7 without changing
+        # the public class API, we do some gross magic. This
+        # has a side effect that we can use this method either
+        # as a static method or not.
+        # FIXME
+        if not isinstance(self, PrgmCompiler):
+            tifile = self
+
         prgm_data = tifile.prgmdata
         plaintext = []
         tokens = get_tokens()