商品图片搜索

特征服务，一般不开放接口
same_private_example.py
#!/usr/bin/env python
#coding:utf-8
import os
import os.path
import json
import sys
import urllib
import urllib2
import base64
from proto import imgfeature_pb2
import logging
import time
import subprocess
import random
logger = logging.getLogger()


class CalFeatureError(Exception):
    """when call remote server error"""
    pass

def calfeature_idlapi_remote(serverurl, imagedata, infos, logid=0):
    """ cropmethod will pass to server as alignflag parameter"""

    def _createprotobufdata(imagestr, info):
        """create protobuf from image and info"""
        proto_data = imgfeature_pb2.FeatureRequest()
        proto_data.image = imagestr
        proto_data.query_sign = '0,0'

        def addinfo(key, value):
            """add info to proto"""
            infoele = proto_data.info.texts.add()
            infoele.key = key
            infoele.value = value

        if info is not None:
            for key, value in info.iteritems():
                addinfo(key, value)
        data = proto_data.SerializeToString()
        return data

    protodata = _createprotobufdata(imagedata, infos)
    protodata = base64.b64encode(protodata)
    req_json = {
        'appid': "123456",
        'logid': logid,
        'format': 'json',
        'from': 'test-python',
        'cmdid': '123',
        'clientip': '0.0.0.0',
        'data': protodata,
    }

    conn = urllib2.urlopen(serverurl, data=json.dumps(req_json))
    resultjson = conn.read()
    result = json.loads(resultjson)
    if result['err_no'] != 0:
        errorinfo = '%s:%s' % (result['err_no'], result['err_msg'])
        logger.warn('got error from server:%s', errorinfo)
        raise CalFeatureError(errorinfo)

    res_data = result['result'].decode('base64')
    proto_result = imgfeature_pb2.FeatureResponse()
    proto_result.ParseFromString(res_data)
    features = {fea.fea_id: fea.fea_data for fea in proto_result.features}
    return features

def calfeature_remote(filename, feaaddr, featureids=''):
    import urllib2
    feaserverurl = 'http://%s/GeneralClassifyService/classify' % (feaaddr)
    param = {"detect": "1", "usedb": "0"}
    imagedata = open(filename, 'rb').read()
    start = time.time()
    features = calfeature_idlapi_remote(feaserverurl, imagedata, params)
    print "time cost: %.2f ms" % ((time.time() - start)*1000)
    for feaname in features:
        print "-------------"*5
        print feaname, features[feaname], len(features[feaname])

def main():
    img_path = 'query.jpg'
    calfeature_remote(img_path,'127.0.0.1:8120','')
    pass

if __name__ == '__main__':
    if len(sys.argv)>1 :
        func = getattr(sys.modules[__name__], sys.argv[1])
        func(*sys.argv[2:])
    else:
        print >> sys.stderr,'tools.py command'

same_private_example_add_remove.py:
#coding:utf-8
import urllib2
import base64
import json
import os
import sys
import logging
logger = logging.getLogger()

AS_SERVER = 'http://10.255.140.17:8025'

ADD_IMG = 'query.jpg'
APPID='test'  


def rts_as_add(filename=ADD_IMG, appid=APPID):
    params = {
        "appid": appid,
        "format": "json",
        "from": "from_xx",
        "logid": 666,
        "brief": "brief",
        "image": base64.b64encode(open(filename,'rb').read()),
        "features": [{"fea_id":1000, "fea_data" : base64.b64encode("addtest")}, {"fea_id":999, "fea_data": base64.b64encode(struct.pack('II', 0, 4))}]
    }
    conn = urllib2.urlopen(AS_SERVER+'/RTSAsService/add', data=json.dumps(params)).read()
    print conn

def rts_as_update(filename=IMG, appid=APPID):
    params = {
        "appid": appid,
        "format": "json",
        "from": "from_xx",
        "logid": 666,
        "image": base64.b64encode(open(filename,'rb').read()),
        "features": [{"fea_id":1000, "fea_data" : base64.b64encode("addtest")}]
    }
    conn = urllib2.urlopen(AS_SERVER+'/RTSAsService/update', data=json.dumps(params)).read()
    print conn

def rts_as_search(filename=IMG, appid=APPID):
    params = {
        "appid": appid,
        "image": base64.b64encode(open(filename,'rb').read()),
        "format": "json",
        "from": "from_yyy",
        "logid": 44,
    }
    conn = urllib2.urlopen(AS_SERVER+'/RTSAsService/search', data=json.dumps(params)).read()
    print conn

def rts_as_delete(contsign, appid=APPID):
    sign1, sign2 = map(int,contsign.split(','))

    params = {
        "appid": appid,
        "format": "json",
        "from": "from_xx",
        "logid": 666,
        #"image": base64.b64encode(open('1.jpg','rb').read()),
        "contsign" : {'sign1':sign1,"sign2":sign2},
    }
    conn = urllib2.urlopen(AS_SERVER+'/RTSAsService/remove', data=json.dumps(params)).read()
    print conn

def rts_as_snapshot(appid=APPID):
    params = {
        "appid": appid,
        "from": "from_xx",
        "logid": 666,
    }
    conn = urllib2.urlopen(AS_SERVER+'/RTSAsService/snapshot', data=json.dumps(params)).read()
    print conn;

def signup(appid=APPID):
    params = {
        "appid": appid,
        "from": "from_xx",
        "logid": 666,
    }
    conn = urllib2.urlopen(AS_SERVER+'/RTSAsService/signup', data=json.dumps(params)).read()
    print conn;

if __name__ == '__main__':
    if len(sys.argv) > 1:
        func = getattr(sys.modules[__name__], sys.argv[1])
        func(*sys.argv[2:])
    else:
        print >> sys.stderr, 'api.py {rts_as_add,rts_as_search,rts_as_delete}'