allpy: 1fb830d7517e allpy/fileio.py

allpy

view allpy/fileio.py @ 554:1fb830d7517e

graph: improve expand_to_max_clique() Do not recalculate candidates and weights at each step

author	boris (kodomo) <bnagaev@gmail.com>
date	Wed, 16 Mar 2011 19:42:58 +0300
parents	7ebeae6eea65
children	f5af99997e0e 737b52785e5e

line source

1 import os

2 from tempfile import NamedTemporaryFile

4 import util

6 class BaseIo(object):

7 """ Base class providing alignment/sequence import and export

9 Data:

10 * file - file object

11 """

13 def __init__(self, file):

14 self.file = file

16 def save_string(self, string, name, description=''):

17 """ Saves given string to file

19 Splits long lines to substrings of length=long_line

20 To prevent this, set long_line=None

21 """

22 pass

24 def get_all_strings(self):

25 """Parse fasta file, remove spaces and newlines from sequence bodies.

27 Return a list of tuples (name, description, sequence_body).

28 """

29 pass

31 def get_string(self, name):

32 """ return tuple (name, description, string) for sequence with name name """

33 for name_test, description, body in self.get_all_strings():

34 if name_test == name:

35 return (name_test, description, body)

37 class FastaIo(BaseIo):

38 """ Fasta import and export

40 Additional data:

41 * long_line - max length of file line while export

42 Splits long lines to substrings of length=long_line

43 To prevent this, set long_line=None

44 """

46 def __init__(self, file, long_line=70):

47 BaseIo.__init__(self, file)

48 self.long_line = long_line

50 def save_string(self, string, name, description=''):

51 if description:

52 name += " " + description

53 self.file.write(">%s\n" % name)

54 if self.long_line:

55 for i in range(0, len(string) // self.long_line + 1):

56 start = i*self.long_line

57 end = i*self.long_line + self.long_line

58 self.file.write("%s\n" % string[start:end])

59 else:

60 self.file.write("%s\n" % string)

62 def get_all_strings(self):

63 for part in self.file.read().split("\n>"):

64 header, _, body = part.partition("\n")

65 header = header.lstrip(">").strip()

66 name, _, description = header.partition(" ")

67 name = name.strip()

68 description = description.strip()

69 body = util.remove_each(body, " \n\r\t\v")

70 yield (name, description, body)

72 def get_string(self, name):

73 for name_test, description, body in self.get_all_strings():

74 if name_test == name:

75 return (name_test, description, body)

77 class MsfIo(BaseIo):

78 """ Msf import and export """

80 def __init__(self, file):

81 BaseIo.__init__(self, file)

83 def save_string(self, string, name, description=''):

84 name = name.replace(':', '_') # seqret bug

85 tmp_fasta = NamedTemporaryFile('w', delete=False)

86 tmp_fasta.close()

87 os.system("seqret %(msf)s %(fasta)s" % \

88 {'msf': self.file.name, 'fasta': tmp_fasta.name})

89 tmp_fasta = open(tmp_fasta.name, 'a')

90 fasta = FastaIo(tmp_fasta)

91 fasta.save_string(string, name, description)

92 tmp_fasta.close()

93 self.file.close()

94 os.system("seqret %(fasta)s msf::%(msf)s" % \

95 {'msf': self.file.name, 'fasta': tmp_fasta.name})

96 os.unlink(tmp_fasta.name)

97 self.file = open(self.file.name)

99 def get_all_strings(self):

100 tmp_fasta = NamedTemporaryFile(delete=False)

101 os.system("seqret %(msf)s %(fasta)s" % \

102 {'msf': self.file.name, 'fasta': tmp_fasta.name})

103 fasta = FastaIo(tmp_fasta)

104 strings = list(fasta.get_all_strings())

105 os.unlink(tmp_fasta.name)

106 return strings

107