import urllib.request

import json

import os

def login():

        username = input("username:")

        password = input("password:")

        url = "https://api.zoomeye.org/user/login"

        data = json.dumps({'username':username,'password':password})

        data_bytes = bytes(data,'utf8')

        try:

                req = urllib.request.Request(url,data_bytes)

                response = urllib.request.urlopen(req)

                html = response.read().decode('utf-8')

                target = json.loads(html)

                access_token = target['access_token']

                with open('access_token.txt','w') as f:

                        f.write(access_token)

                        f.close

                        print('login seccess !!!')

                #print('your access_token is:%s'%(access_token))

        except Exception as err:

                print('[info]:username or password is wrong !')

def apiget():

        while True:

                host_or_web = input("search for host or web:(‘q!’Sign out):")

                if host_or_web == 'q!':

                        break

                query = input("input your keyword(-r for your resources-info):")  #查看自己的套餐剩余量

                if query == '-r':

                        req = urllib.request.Request('https://api.zoomeye.org/resources-info')

                        ida = open('access_token.txt').read() 

                        req.add_header('Authorization','JWT %s'%(ida))

                        re = urllib.request.urlopen(req)

                        ae = (re.read().decode('utf-8'))

                        ae = target = json.loads(ae)

                        dict_web = ae['resources']['web-search']

                        dict_host = ae['resources']['host-search']

                        print('your web search:%s'%(dict_web))

                        print('your host search:%s'%(dict_host))

                else:

                        try:
　　　　　　　　　　　　　　　　　　　req = urllib.request.Request('https://api.zoomeye.org/%s/search?query=%s'%(host_or_web,query))

                                idb = open('access_token.txt').read()

                                req.add_header('Authorization','JWT %s'%(idb))

                                r = urllib.request.urlopen(req)

                                a = (r.read().decode('utf-8'))

                                a = target = json.loads(a)

                                print('tatal:%s'%(a['total']))

                                for i in range(len(a['matches'])):

                                        print (a['matches'][i]['ip'])

                        except Exception as err:

                                print('[erro]:Please enter the correct syntax !')

def start():

        if not os.path.isfile('access_token.txt'): #首次使用会检查脚本目录下access_token.txt文件，没有的话登录会创建

                print('[info]:you need login')

                login()

        apiget()

if __name__ == '__main__':

    start()