Mercurial > repos > yating-l > hubarchivecreator
annotate util/index/TrixIndex.py @ 83:0bc486356e2a draft
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
author | yating-l |
---|---|
date | Mon, 30 Oct 2017 12:01:23 -0400 |
parents | |
children | cb62314a6297 |
rev | line source |
---|---|
83
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
1 #!/usr/bin/python |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
2 |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
3 import os |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
4 import collections |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
5 import shutil |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
6 import logging |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
7 from ExternIndex import ExternIndex |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
8 |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
9 class TrixIndex(ExternIndex): |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
10 def __init__(self, indexIx, indexIxx, trackName, mySpecieFolderPath, **args): |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
11 self.logger = logging.getLogger(__name__) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
12 self.indexIx = indexIx |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
13 self.indexIxx = indexIxx |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
14 self.trackName = trackName |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
15 self.mySpecieFolderPath = mySpecieFolderPath |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
16 self.index_settings = collections.OrderedDict() |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
17 if "trixId" in args: |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
18 self.trixId = args["trixId"] |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
19 else: |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
20 self.trixId = None |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
21 |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
22 def setExtLink(self): |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
23 self.moveIndexFile() |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
24 self.index_settings["searchTrix"] = "trix/%s" % self.indexIxName |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
25 return self.index_settings |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
26 |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
27 def moveIndexFile(self): |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
28 indexFolder = os.path.join(self.mySpecieFolderPath, 'trix') |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
29 self.indexIxName = "".join( ( self.trackName, ".ix") ) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
30 self.indexIxxName = "".join( ( self.trackName, ".ixx") ) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
31 if not os.path.exists(indexFolder): |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
32 os.makedirs(indexFolder) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
33 |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
34 # Move index files to the index folder |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
35 self.indexIxPath = os.path.join(indexFolder, self.indexIxName) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
36 shutil.copyfile(self.indexIx, self.indexIxPath) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
37 self.indexIxxPath = os.path.join(indexFolder, self.indexIxxName) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
38 shutil.copyfile(self.indexIxx, self.indexIxxPath) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
39 |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
40 def setSearchIndex(self): |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
41 if self.trixId: |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
42 set_index = set() |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
43 set_index.add(self.trixId) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
44 set_index.add("name") |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
45 search_index = ",".join(set_index) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
46 else: |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
47 search_index = "name" |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
48 logging.debug("trixId= %s, searchIndex= %s", self.trixId, search_index) |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
49 self.index_settings["searchIndex"] = search_index |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
50 |
0bc486356e2a
planemo upload for repository https://github.com/goeckslab/hub-archive-creator commit ac73da32d59853ca563e7939d05016a6f3a6899e-dirty
yating-l
parents:
diff
changeset
|
51 |