view util/index/TrixIndex.py @ 1:85195e0d4b71 draft

planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit b1ae7349f118a0fe7923d765020dfc684cf84116-dirty

#!/usr/bin/python

import os
import collections
import shutil
import logging
from ExternIndex import ExternIndex

class TrixIndex(ExternIndex):
    def __init__(self, indexIx, indexIxx, trackName, mySpecieFolderPath, trixId, **args):
        self.indexIx = indexIx
        self.indexIxx = indexIxx
        self.trackName = trackName
        self.mySpecieFolderPath = mySpecieFolderPath
        self.trixId = trixId
        self.index_settings = collections.OrderedDict()
        if "default_index" in args:
            self.default_index = args["default_index"]
        else:
            self.default_index = None

    def setExtLink(self):
        self.setSearchIndex()
        self.moveIndexFile()
        self.index_settings["searchTrix"] = "trix/%s" % self.indexIxName
        return self.index_settings

    def moveIndexFile(self):
        indexFolder = os.path.join(self.mySpecieFolderPath, 'trix')
        self.indexIxName = "".join( ( self.trackName, ".ix") )
        self.indexIxxName = "".join( ( self.trackName, ".ixx") )
        if not os.path.exists(indexFolder):
            os.makedirs(indexFolder)
        
        # Move index files to the index folder
        self.indexIxPath = os.path.join(indexFolder, self.indexIxName)
        shutil.copyfile(self.indexIx, self.indexIxPath)
        self.indexIxxPath = os.path.join(indexFolder, self.indexIxxName)
        shutil.copyfile(self.indexIxx, self.indexIxxPath)

    def setSearchIndex(self):
        if self.default_index:
            set_index = set()
            set_index.add(self.trixId)
            set_index.add(self.default_index)
            search_index = ",".join(set_index)
        else:
            search_index = self.trixId
        logging.debug("trixId= %s, searchIndex= %s", self.trixId, search_index)
        self.index_settings["searchIndex"] = search_index