import logging

import cv2
import requests
from tenacity import retry, stop_after_attempt, wait_random


@retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
       after=lambda x: logging.warning('获取文档区域失败！'))
def request_book_areas(img_path):
    url = 'http://det_api:5000/det/books'
    response = requests.post(url, {'img_path': img_path})
    if response.status_code == 200:
        response_data = response.json()
        books = []
        for books_path in response_data:
            books.append(cv2.imread(books_path))
        return books
    else:
        return []