search_aj

#!/usr/bin/env python
#coding=UTF-8
## 搜索Java资源，https://github.com/superhj1987/awesome-tech-collections/blob/master/awesome-java.md

import sys
import getopt 
import urllib2

reload(sys) 
sys.setdefaultencoding( "utf-8" )

TIPS_TPL = "\n" + '=' * 8 + "%s" + '=' * 8 + "\n"

def usage():
    print "usage: search_aj [catagory_name | project_name]"

def progress_report(bytes_so_far, chunk_size, total_size):
   percent = float(bytes_so_far) / total_size
   percent = round(percent*100, 2)
   sys.stdout.write("searching.... (%0.2f%%)\r" % 
       (percent))
   sys.stdout.flush()

   if bytes_so_far >= total_size:
      sys.stdout.write('\n')

def main():
    opts,args = getopt.getopt(sys.argv[1:],'',[])

    argc = len(args)
    if argc < 1:
        usage()
        sys.exit(1) 

    search_name = args[0]

    req = urllib2.Request("https://raw.githubusercontent.com/superhj1987/awesome-tech-collections/master/awesome-java.md")
    res = urllib2.urlopen(req)
    
    total_size = res.info().getheader('Content-Length').strip()
    total_size = int(total_size)
    chunk_size = 1024
    bytes_so_far = 0
    data = []
    while 1:
      chunk = res.read(chunk_size)
      bytes_so_far += len(chunk)
      if not chunk:
         break

      data += chunk

      progress_report(bytes_so_far, chunk_size, total_size)
    content = "".join(data)

    category = ""
    target_cat = None
    search_result = ""
    lines = content.split("\n")
    search_name = search_name.lower()
    for line in lines:
        if line.startswith("##") :
            target_cat = None
            category = line[line.index(" "):].strip().lower()
            if category.find(search_name) > 0:
                search_result +=  TIPS_TPL % ('finded in Categoty[' + category + ']')
                target_cat = category
        elif line.startswith("* ["):
            project = line[line.index("["):].strip().lower()
            if target_cat:
                search_result += project + "\n"
            elif project.find(search_name) > 0:
                search_result += TIPS_TPL % ('find matched projects in Category[' + category + ']') + project + "\n"

    print search_result

if __name__ == "__main__":
    main()