1 files changed, 57 insertions, 0 deletions

diff --git a/rspec.py b/rspec.py
new file mode 100644
index 0000000..72d0e45
--- /dev/null
+++ b/rspec.py
@@ -0,0 +1,57 @@
+import re
+import sys
+import inspect
+from colors import red, green
+
+
+DEBUG = False
+current_function = ""
+
+
+class Opinion(object):
+
+    def __init__(self, thing):
+        self.thing = thing
+        self.args = []
+        self.kwargs = {}
+
+    def should(self, matcher):
+        if not inspect.getargspec(matcher).args:
+            matcher = matcher()
+
+        try:
+            matcher(self.thing, *self.args, **self.kwargs)
+        except AssertionError as e:
+            print(red("".join([current_function, " failed: ", str(e)])))
+        else:
+            print(green(current_function))
+
+        return self
+
+    def __call__(self, *args, **kwargs):
+        self.args = args
+        self.kwargs = kwargs
+        return self
+
+
+def tracer(frame, event, arg):
+    if DEBUG and event == 'call':
+        print(event, frame, arg, frame.f_code.co_name, frame.f_globals.keys())
+
+    for name, function in frame.f_globals.items():
+        if callable(function) and function.__module__ == "__main__":
+            frame.f_globals[name] = Opinion(function)
+
+    return tracer
+
+
+def rspec_run(function):
+    global current_function
+    ptrace = sys.gettrace()
+
+    try:
+        sys.settrace(tracer)
+        current_function = re.sub("_", " ", function.__name__)
+        function()
+    finally:
+        sys.settrace(ptrace)