Messed around with getting my encoding to work

text_codecs/ascii7.py

@@ -1,5 +1,5 @@
-from text_codecs.register import register_codec
 from easybits import Bits
+import codecs
 
 def encode(text):
     """An encoder which only handles ASCII: non-ASCII characters
@@ -30,4 +30,20 @@ def decode(data):
         text += Bits(byte).ascii
     return text
 
+def register_codec(encode, decode, name):
+    """Registers a codec so that it can later be used to encode
+    or decode strings and bytes.
+    """
+    def encode_wrapper(text):
+        return encode(text), len(text)
+
+    def decode_wrapper(data):
+        return decode(data), len(data)
+
+    def search_for_codec(query):
+        if query == name:
+            return codecs.CodecInfo(encode_wrapper, decode_wrapper, name=name)
+
+    codecs.register(search_for_codec)
+
 register_codec(encode, decode, "ascii7")