Mercurial > repos > guerler > dbkit
diff dbkit_package/DBKit.py @ 2:81c7d4668a7e draft
"planemo upload commit bd03b7888eab0b010acfc3affd38bf4d4e2bb1ef-dirty"
author | guerler |
---|---|
date | Wed, 16 Dec 2020 12:02:48 +0000 |
parents | |
children | 03e124ff7e26 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/dbkit_package/DBKit.py Wed Dec 16 12:02:48 2020 +0000 @@ -0,0 +1,32 @@ +class DBKit: + def __init__(self, indexFile, databaseFile): + self.databaseFile = databaseFile + self.index = dict() + with open(indexFile) as file: + for line in file: + cols = line.split() + try: + identifier = cols[0] + start = int(cols[1]) + size = int(cols[2]) + self.index[identifier] = [start, size] + except Exception: + raise Exception("Invalid DBKit Index file format: %s." % line) + + def createFile(self, identifier, outputName): + if identifier in self.index: + entry = self.index[identifier] + start = entry[0] + size = entry[1] + with open(self.databaseFile) as file: + file.seek(start) + content = file.read(size) + outputFile = open(outputName, "w") + outputFile.write(content) + outputFile.close() + return True + else: + return False + + def getIndex(self): + return self.index