get adult score by Microsoft Computer Vision API

get_adult_score.py

import sys
import os
import json
import http.client
from urllib.parse import urlencode


# Computer Vision API
# https://azure.microsoft.com/en-us/services/cognitive-services/computer-vision/


MICROSOFT_COMPUTER_VISION_API_KEY = os.environ['MICROSOFT_COMPUTER_VISION_API_KEY']


def get_ms_vision_score(jpeg_path):
    headers = {
       'Content-Type': 'application/octet-stream',
       'Ocp-Apim-Subscription-Key': MICROSOFT_COMPUTER_VISION_API_KEY,
    }

    params = urlencode({
       'visualFeatures': 'Adult',
    })

    conn = http.client.HTTPSConnection('southeastasia.api.cognitive.microsoft.com')
    conn.request("POST", "/vision/v1.0/analyze?%s" % params, open(jpeg_path, "br").read(), headers)
    return conn.getresponse()


if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("{} <image_file>".format(sys.argv[0]))
        sys.exit()
    response = get_ms_vision_score(sys.argv[1])
    print(json.dumps(json.loads(response.read().decode('utf-8')), indent=4))