made my codecs but was having trouble testing it

This commit is contained in:
tsmith372
2026-04-06 19:46:50 -04:00
parent 7e31ef1cd7
commit c71f841088
5 changed files with 89 additions and 2 deletions

View File

@@ -1,7 +1,25 @@
import string
import codecs
from custom_codecs.register import register_codec
from easybits import Bits
import codecs
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)
allowed_characters = string.ascii_letters + string.digits