From e4ecc40711e80ddfa14ea1ffb2704a46bb8acdd4 Mon Sep 17 00:00:00 2001
From: Calvin Morrison <mutantturkey@gmail.com>
Date: Mon, 22 Jul 2013 19:29:38 -0400
Subject: throw an exception for argv[1] and give it a shebang as well as
 generalize the sed call

---
 pc.py    | 82 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 rtsed.py | 76 -----------------------------------------------------------
 2 files changed, 82 insertions(+), 76 deletions(-)
 create mode 100755 pc.py
 delete mode 100644 rtsed.py

diff --git a/pc.py b/pc.py
new file mode 100755
index 0000000..35da4fb
--- /dev/null
+++ b/pc.py
@@ -0,0 +1,82 @@
+#!/usr/bin/python
+import curses
+import os
+import sys
+import StringIO
+
+from subprocess import *
+import subprocess
+
+# our screen
+screen = curses.initscr()
+curses.noecho()
+curses.cbreak()
+curses.curs_set(0)
+curses.start_color()
+
+curses.init_pair(1, curses.COLOR_WHITE, curses.COLOR_CYAN)
+curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_RED) 
+
+# our stdin input
+input_buffer = ""
+
+# rows and columns of our screen
+cols = int(subprocess.check_output(["tput", "cols"]))
+rows = int(subprocess.check_output(["tput", "lines"]))
+
+def call_process(process, string):
+  process = Popen(["sed", "-e", string], stdout=PIPE, stdin=PIPE, stderr=PIPE) 
+  ret = process.communicate(input_buffer)
+  if(process.returncode is not 0):
+    screen.addstr(ret[1][:-1].ljust(cols), curses.color_pair(2));
+  else:
+    screen.addstr(ret[0]);
+  return
+
+
+def main(argv):
+
+  input_string = ""
+  global input_buffer
+
+  try:
+    input_fh = open(argv[1], "r")
+    for line in input_fh:
+      input_buffer = input_buffer + line
+  except:
+    curses.endwin()
+    sys.exit("please specify a file")
+  
+  # init our screen
+  screen.addstr("".ljust(cols), curses.color_pair(1));
+  screen.addstr(input_buffer)
+  screen.addstr("EOF")
+
+  try: 
+    while 1:
+      key = screen.getch()
+      screen.clear()
+      if(key > 256):
+        continue;
+      if(key == 127):
+        input_string = input_string[:-1]
+        screen.addstr(input_string.ljust(cols), curses.color_pair(1));
+        screen.addstr(input_buffer)
+        screen.addstr("EOF")
+        continue;
+      elif(key == 13 or key == 10):
+        screen.addstr(input_string.ljust(cols), curses.color_pair(1));
+        call_process("sed", input_string)
+        screen.addstr("EOF")
+        continue;
+      else:
+        input_string+=chr(key)
+        screen.addstr(input_string.ljust(cols), curses.color_pair(1));
+        screen.addstr(input_buffer)
+        screen.addstr("EOF")
+  finally:
+    curses.endwin()
+
+if __name__ == "__main__":
+    sys.exit(main(sys.argv))
+
diff --git a/rtsed.py b/rtsed.py
deleted file mode 100644
index af17b2b..0000000
--- a/rtsed.py
+++ /dev/null
@@ -1,76 +0,0 @@
-import curses
-import os
-import sys
-import StringIO
-
-from subprocess import *
-import subprocess
-
-# our screen
-screen = curses.initscr()
-curses.noecho()
-curses.cbreak()
-curses.curs_set(0)
-curses.start_color()
-
-curses.init_pair(1, curses.COLOR_WHITE, curses.COLOR_CYAN)
-curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_RED) 
-
-# our stdin input
-input_buffer = ""
-
-# rows and columns of our screen
-cols = int(subprocess.check_output(["tput", "cols"]))
-rows = int(subprocess.check_output(["tput", "lines"]))
-
-def call_sed(string):
-  process = Popen(["sed", "-e", string], stdout=PIPE, stdin=PIPE, stderr=PIPE) 
-  ret = process.communicate(input_buffer)
-  if(process.returncode is not 0):
-    screen.addstr(ret[1][:-1].ljust(cols), curses.color_pair(2));
-  else:
-    screen.addstr(ret[0]);
-  return
-
-
-def main():
-
-  input_string = ""
-  global input_buffer
-
-  for line in open(sys.argv[1], "r"):
-    input_buffer = input_buffer + line
-  
-  # init our screen
-  screen.addstr("".ljust(cols), curses.color_pair(1));
-  screen.addstr(input_buffer)
-  screen.addstr("EOF")
-
-  try: 
-    while 1:
-      key = screen.getch()
-      screen.clear()
-      if(key > 256):
-        continue;
-      if(key == 127):
-        input_string = input_string[:-1]
-        screen.addstr(input_string.ljust(cols), curses.color_pair(1));
-        screen.addstr(input_buffer)
-        screen.addstr("EOF")
-        continue;
-      elif(key == 13 or key == 10):
-        screen.addstr(input_string.ljust(cols), curses.color_pair(1));
-        call_sed(input_string)
-        screen.addstr("EOF")
-        continue;
-      else:
-        input_string+=chr(key)
-        screen.addstr(input_string.ljust(cols), curses.color_pair(1));
-        screen.addstr(input_buffer)
-        screen.addstr("EOF")
-  finally:
-    curses.endwin()
-
-if __name__ == "__main__":
-    sys.exit(main())
-
-- 
cgit v1.2.1