srv_utils.py

#coding:utf-8
# Copyright (c) 2019  PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import requests
import time
import paddle
import socket

from random import randint, seed

from paddlehub import version
from paddlehub.common.server_config import default_stat_config


def get_stat_server():
    seed(int(time.time()))
    stat_env = os.environ.get('HUB_SERVER_STAT_SRV')
    if stat_env:
        server_list = stat_env.split(';')
    else:
        server_list = default_stat_config['server_list']
    return server_list[randint(0, len(server_list) - 1)]


def hub_stat(argv):
    try:
        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        s.connect(('bj.bcebos.com', 80))
        ip_addr = s.getsockname()[0]
        params = {
            'command': ' '.join(argv),
            'hub_version': version.hub_version,
            'paddle_version': paddle.__version__,
            'ip_addr': ip_addr
        }
        stat_api = get_stat_server()
        r = requests.get(stat_api, params=params, timeout=0.5)
    except:
        pass


def uri_path(server_url, api):
    srv = server_url
    if server_url.endswith('/'):
        srv = server_url[:-1]
    if api.startswith('/'):
        srv += api
    else:
        api = '/' + api
        srv += api
    return srv


def hub_request(api, params):
    params['hub_version'] = version.hub_version
    params['paddle_version'] = paddle.__version__
    r = requests.get(api, params)
    return r.json()