Mercurial > repos > mvdbeek > test_sra

annotate sra.py @ 18:fc784976527b draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
author mvdbeek
date Sat, 15 Aug 2015 07:44:28 -0400
parents ec8b334b5ebb
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
1 """
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
2 NCBI sra class
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
3 """
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
4 import logging
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
5 import binascii
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
6 from galaxy.datatypes.data import nice_size
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
7 from galaxy.datatypes.binary import Binary
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
8
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
9 log = logging.getLogger(__name__)
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
10
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
11 class Sra(Binary):
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
12 """ Sequence Read Archive (SRA) """
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
13 file_ext = 'sra'
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
14
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
15 def __init__( self, **kwd ):
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
16 Binary.__init__( self, **kwd )
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
17
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
18 def sniff( self, filename ):
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
19 """ The first 8 bytes of any NCBI sra file is 'NCBI.sra', and the file is binary.
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
20 For details about the format, see http://www.ncbi.nlm.nih.gov/books/n/helpsra/SRA_Overview_BK/#SRA_Overview_BK.4_SRA_Data_Structure
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
21 """
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
22 try:
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
23 header = open(filename).read(8)
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
24 if binascii.b2a_hex(header) == binascii.hexlify('NCBI.sra'):
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
25 return True
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
26 else:
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
27 return False
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
28 except:
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
29 return False
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
30
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
31 def set_peek(self, dataset, is_multi_byte=False):
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
32 if not dataset.dataset.purged:
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
33 dataset.peek = 'Binary sra file'
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
34 dataset.blurb = nice_size(dataset.get_size())
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
35 else:
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
36 dataset.peek = 'file does not exist'
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
37 dataset.blurb = 'file purged from disk'
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
38
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
39 def display_peek(self, dataset):
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
40 try:
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
41 return dataset.peek
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
42 except:
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
43 return 'Binary sra file (%s)' % (nice_size(dataset.get_size()))
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
44
ec8b334b5ebb planemo upload commit 96ecd86165525684766848f98a1c3be4353ec3b1-dirty
mvdbeek
parents:
diff changeset
45 Binary.register_sniffable_binary_format('sra', 'sra', Sra)