import os
import sys

rekursioon=0
valjundikaust="."
if len(sys.argv)>1:
  for arg in sys.argv:
    if arg=="koos_alamkataloogidega=1":rekursioon=1
    if arg.find("valjundikaust")==0:
      m=arg.split("=")
      if len(m)==2:valjundikaust=m[1]

f1=open("marksonad.txt", "r")
otsitavad=f1.readline().strip().split()
f1.close()

def leiaVasted(sisend):
 if sisend.find("_valjund")!=-1 or sisend.find("marksonad.txt")!=-1: return
 if len(sisend)>4 and sisend[-4:]==".txt": valjund=valjundikaust+sisend[1:-4]+"_valjund.txt"
 else: return 
 f2=open(sisend, "r")
 try:
   os.makedirs(valjund[0: valjund.rfind("/")])
 except:
   pass #juba olemas
 f3=open(valjund, "w")
 loend={}
 for rida in f2:
  leitud=[]
  for sona in otsitavad:
     if rida.find(sona)!=-1:
       leitud.append(sona)
       if sona in loend:
         loend[sona]=loend[sona]+1
       else:
         loend[sona]=1
  if leitud:
     f3.write(", ".join(leitud)+"\n")
     f3.write("******\n")
     f3.write(rida+"\n")
 f2.close()
 f3.write("\n------\n")
 for sona in loend.keys():
  f3.write(sona+": "+str(loend[sona])+"\n")
 f3.close()

def tootleFailid(kataloog):
 print kataloog
 failid=os.listdir(kataloog)
 for fail in failid:
   print fail
   if os.path.isdir(kataloog+"/"+fail):
    if rekursioon:
     tootleFailid(kataloog+"/"+fail)
   else: leiaVasted(kataloog+"/"+fail)
 
tootleFailid(".")

#kaivitamiseks nt
#python uuring3.py valjundikaust=tulemused koos_alamkataloogidega=1