demo reader

demo_reader___init__.py

print('hi')

demo_reader_compressed___init__.py

from demo_reader.compressed.bzipped import opener as bz2_opener
from demo_reader.compressed.gzipped import opener as gzip_opener


__all__ = ['bz2_opener', 'gzip_opener']

demo_reader_compressed_bzipped.py

import bz2

from ..util import writer

opener = bz2.open

if __name__ == '__main__':
    writer.main(opener)

demo_reader_compressed_gzipped.py

import gzip

from ..util import writer

opener = gzip.open

if __name__ == '__main__':
    writer.main(opener)

demo_reader_multireader.py

import os

from demo_reader.compressed import bzipped
from demo_reader.compressed import gzipped

extensions_map = {
    '.gz' : gzipped.opener,
    '.bz2' : bzipped.opener
}


class MultiReader:
    def __init__(self, filename):
        ext = os.path.splitext(filename)[1]
        opener = extensions_map.get(ext, open)
        self.f = opener(filename, 'rt')


    def close(self):
        self.f.close()

    def read(self):
        return self.f.read()

demo_reader_util_writer.py

import sys

def main(opener):
    f = opener(sys.argv[1], mode='wt')
    f.write(''.join(sys.argv[2:]))
    f.close()