apiServer.py

#coding:utf-8
'''
定义几个关键字，count type,protocol,country,area,
'''
import urllib
from config import API_PORT
from db.SQLiteHelper import SqliteHelper

__author__ = 'Xaxdus'

import BaseHTTPServer
import json
import urlparse
import logging
logger = logging.getLogger('api')

# keylist=['count', 'types','protocol','country','area']
class WebRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):

    def do_GET(self):
        """
        """
        dict={}

        parsed_path = urlparse.urlparse(self.path)
        try:
            query = urllib.unquote(parsed_path.query)
            logger.info("query %s" %query)
            if query.find('&')!=-1:
                params = query.split('&')
                for param in params:
                    dict[param.split('=')[0]]=param.split('=')[1]
            else:
                    dict[query.split('=')[0]]=query.split('=')[1]

            sqlHelper = SqliteHelper()
            # 处理删除代理的请求
            if dict.has_key('delete'):
                condition="ip='" + dict['ip'] + "' AND port=" + dict['port']
                sqlHelper.delete(SqliteHelper.tableName, condition)
                self.send_response(200)
                self.end_headers()
                self.wfile.write("Success delete proxy: " + dict['ip'] + ":" + dict['port'])
            else:
                str_count=''
                conditions=[]
                for key in dict:
                    if key =='count':
                        str_count = 'LIMIT 0,%s'% dict[key]
                    if key =='country' or key =='area':
                        conditions .append(key+" LIKE '"+dict[key]+"%'")
                    elif key =='types' or key =='protocol' or key =='country' or key =='area':
                        conditions .append(key+"="+dict[key])
                if len(conditions)>1:
                    conditions = ' AND '.join(conditions)
                else:
                    conditions =conditions[0]
                result = sqlHelper.select(sqlHelper.tableName,conditions,str_count)
                # print type(result)
                # for r in  result:
                #     print r
                data = [{'ip':item[0], 'port': item[1]} for item in result]
                data = json.dumps(data)
                self.send_response(200)
                self.end_headers()
                self.wfile.write(data)
        except Exception,e:
            logger.warning(str(e))
            self.send_response(404)

if __name__=='__main__':
    server = BaseHTTPServer.HTTPServer(('0.0.0.0',API_PORT), WebRequestHandler)
    server.serve_forever()