復活の呪文

fukkatsu.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

import re
import sys

class FukkatsuEncoder:

    __USABLE_CHARS = \
        'あいうえおかきくけこさしすせそたちつてとなにぬねのはひふへほ' \
        'まみむめもらりるれろがぎぐげござじずぜぞばびぶべぼぱぴぷぺぽ' \
        'やゆよわ'


    def encode(self, raw: bytes) -> str:
        b = ''.join(map(lambda c: format(c, '08b'), raw))

        # ６の倍数になるようゼロ埋めする
        b += '00000'
        b = b[0:-(len(b)%6)]

        # 連番が偶数となる要素は空文字なので除外する
        encoded = ''.join(map(lambda c: self.__USABLE_CHARS[int(c, 2)], \
                              re.split('(.{6})', b)[1::2] \
        ))
        return encoded


    def decode(self, encoded: str) -> bytes:
        b = ''.join(map(lambda c: format(self.__USABLE_CHARS.find(c), '06b'), \
                        encoded))

        # ８の倍数に満たない場合、末尾は捨てられてしまうので問題なし
        raw = bytes(map(lambda i: int(i, 2), \
                    re.split('(.{8})', b)[1::2]))
        return raw


def main():
    fe = FukkatsuEncoder()

    print(fe.encode('あいう'.encode()))
    # => ぴぴきうぴぴきおぴぴきき

    print(fe.decode('ぴぴきうぴぴきおぴぴきき').decode())
    # => あいう


if __name__ == '__main__':
    main()