File: Record.py

package info (click to toggle)
python-biopython 1.68%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 46,860 kB
  • ctags: 13,237
  • sloc: python: 160,306; xml: 93,216; ansic: 9,118; sql: 1,208; makefile: 155; sh: 63
file content (90 lines) | stat: -rw-r--r-- 2,946 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# Copyright 2001 by Katharine Lindner.  All rights reserved.
# Copyright 2006 by PeterC.  All rights reserved.
# This code is part of the Biopython distribution and governed by its
# license.  Please see the LICENSE file that should have been included
# as part of this package.

"""Hold GEO data in a straightforward format.

classes:
o Record - All of the information in an GEO record.

See http://www.ncbi.nlm.nih.gov/geo/
"""

from __future__ import print_function


class Record(object):
    """Hold GEO information in a format similar to the original record.

    The Record class is meant to make data easy to get to when you are
    just interested in looking at GEO data.

    Attributes:
    entity_type
    entity_id
    entity_attributes
    col_defs
    table_rows

    """
    def __init__(self):
        self.entity_type = ''
        self.entity_id = ''
        self.entity_attributes = {}
        self.col_defs = {}
        self.table_rows = []

    def __str__(self):
        output = ''
        output += 'GEO Type: %s\n' % self.entity_type
        output += 'GEO Id: %s\n' % self.entity_id
        att_keys = sorted(self.entity_attributes)
        for key in att_keys:
            contents = self.entity_attributes[key]
            if isinstance(contents, list):
                for item in contents:
                    try:
                        output += '%s: %s\n' % (key, item[:40])
                        output += out_block(item[40:])
                    except Exception:  # TODO: IndexError?
                        pass
            elif isinstance(contents, str):
                output += '%s: %s\n' % (key, contents[:40])
                output += out_block(contents[40:])
            else:
                print(contents)
                output += '%s: %s\n' % (key, contents[:40])
                output += out_block(contents[40:])
        col_keys = sorted(self.col_defs)
        output += 'Column Header Definitions\n'
        for key in col_keys:
            val = self.col_defs[key]
            output += '    %s: %s\n' % (key, val[:40])
            output += out_block(val[40:], '    ')
        # May have to display VERY large tables,
        # so only show the first 20 lines of data
        MAX_ROWS = 20 + 1  # include header in count
        for row in self.table_rows[0:MAX_ROWS]:
            output += '%s: ' % self.table_rows.index(row)
            for col in row:
                output += '%s\t' % col
            output += '\n'
        if len(self.table_rows) > MAX_ROWS:
            output += '...\n'
            row = self.table_rows[-1]
            output += '%s: ' % self.table_rows.index(row)
            for col in row:
                output += '%s\t' % col
            output += '\n'

        return output


def out_block(text, prefix=''):
    output = ''
    for j in range(0, len(text), 80):
        output += '%s%s\n' % (prefix, text[j:j + 80])
    output += '\n'
    return output