#!/usr/local/bin/python
# programmer : Bo
# usage:

import sys
import re
import random
import string
import time

def main(X):
    try:
        print 'opening file :',X
        infile = open(X,"r+").readlines()
        print 'Total ',len(infile),' lines.'
        return infile
    except IOError,message:
        print >> sys.stderr, "cannot open file",message
        sys.exit(1)

if __name__=="__main__":
    tS = time.time()
    T = main('mouse_TF_list')
    tf = []
    for ea in T:
        tf.append(ea[:-1])
    P = main(sys.argv[1])
    o = file('TF_'+sys.argv[1],'w')
    o.write(P[0])
    for i in range(1,len(P)):
        te = P[i].split('\t')
        if te[0] in tf:
            o.write(P[i])
















    tE = time.time()
    print 'Cost ',(tE-tS),' sec'

