def separate_tiers(args): tiers = set(args.tiers) # assuming XML for now with open(args.infile, 'r') as instream: src_xc = xigtxml.load(instream) sep_xc = XigtCorpus(attributes=src_xc.attributes, metadata=src_xc.metadata) for igt in src_xc.igts: sep_xc.add( Igt(id=igt.id, type=igt.type, attributes=igt.attributes, metadata=igt.metadata, tiers=[t for t in igt.tiers if t.type in tiers])) xigtxml.dump(open(args.outfile, 'w'), sep_xc) if not args.remainder: return with open(args.infile, 'r') as instream: src_xc = xigtxml.load(instream) rem_xc = XigtCorpus(attributes=src_xc.attributes, metadata=src_xc.metadata) for igt in src_xc.igts: rem_xc.add( Igt(id=igt.id, type=igt.type, attributes=igt.attributes, metadata=igt.metadata, tiers=[t for t in igt.tiers if t.type not in tiers])) xigtxml.dump(open(args.remainder, 'w'), rem_xc)
def separate_tiers(args): tiers = set(args.tiers) # assuming XML for now with open(args.infile,'r') as instream: src_xc = xigtxml.load(instream) sep_xc = XigtCorpus(attributes=src_xc.attributes, metadata=src_xc.metadata) for igt in src_xc.igts: sep_xc.add(Igt(id=igt.id, type=igt.type, attributes=igt.attributes, metadata=igt.metadata, tiers=[t for t in igt.tiers if t.type in tiers])) xigtxml.dump(open(args.outfile, 'w'), sep_xc) if not args.remainder: return with open(args.infile,'r') as instream: src_xc = xigtxml.load(instream) rem_xc = XigtCorpus(attributes=src_xc.attributes, metadata=src_xc.metadata) for igt in src_xc.igts: rem_xc.add(Igt(id=igt.id, type=igt.type, attributes=igt.attributes, metadata=igt.metadata, tiers=[t for t in igt.tiers if t.type not in tiers])) xigtxml.dump(open(args.remainder, 'w'), rem_xc)