view taxtastic.py @ 6:3c50a937d7c1 draft

Uploaded

import os
import zipfile
from galaxy.datatypes.binary import Binary
from galaxy.datatypes.data import Text

class Refpkg(Text):
    composite_type = "basic"

    def __init__(self, **kwd):
        Text.__init__(self, **kwd)
        self.add_composite_file("CONTENTS.json")

    def get_mime(self):
        return "application/json"

class RefpkgZip(Binary):
    file_ext = "refpkg.zip"

    def __init__(self, **kwd):
        Binary.__init__(self, **kwd)

    def sniff(self, filename):
        if not zipfile.is_zipfile(filename):
            return False
        contains_contents_file = False
        zip_file = zipfile.ZipFile(filename, "r")
        for name in zip_file.namelist():
            if os.path.basename(name) == "CONTENTS.json":
                contains_contents_file = True
                break
        zip_file.close()
        if not contains_contents_file:
            return False
        return True

    def get_mime(self):
        return "application/zip"

class OfficeXlsx(Binary):
    file_ext = "xlsx"

    def __init__(self, **kwd):
        Binary.__init__(self, **kwd)

    # TODO: this should check for an xl/ directory also
    def sniff(self, filename):
        if not zipfile.is_zipfile(filename):
            return False
        contains_contents_file = False
        zip_file = zipfile.ZipFile(filename, "r")
        for name in zip_file.namelist():
            if os.path.basename(name) == "[Content_Types].xml":
                contains_contents_file = True
                break
        zip_file.close()
        if not contains_contents_file:
            return False
        return True

    def get_mime(self):
        return "application/zip"

Binary.register_sniffable_binary_format("refpkg.zip", "refpkg.zip", RefpkgZip)
Binary.register_sniffable_binary_format("xlsx", "xlsx", OfficeXlsx)