def init_pdb_groups_relation():
    groups = get_manual_groups()
    group_rel = PdbGroupRelation(groups)

    return group_rel
import xlrd
import xlwt

from util.manual_classification import get_manual_groups

groups = get_manual_groups("166")
pdbs = []
for g in groups:
    for i in g:
        pdbs.append(i)

new_wb = xlwt.Workbook()
new_sh = new_wb.add_sheet("166 data")

wb = xlrd.open_workbook("antiginInfo-rich.xls")
sh = wb.sheet_by_index(0)
row_count = 0
for rownum in range(sh.nrows):
    row = sh.row_values(rownum)
    pdb = '_'.join(row[:2])
    if pdb in pdbs:
        for i in xrange(len(row)):
            new_sh.write(row_count,i,row[i])
        row_count += 1

print row_count
new_wb.save("antigen_info_166.xls")