fab_ui_upload_logo.py 文件源码

def upload(session: Session, token: str, file_path: str) -> Response:
    """Upload logo.

    :param session: Supplier session object
    :param token: CSRF token required to upload the file
    :param file_path: absolute path to the uploaded file
    :return: response object
    """
    headers = {"Referer": get_absolute_url("ui-buyer:upload-logo")}
    url = get_absolute_url("ui-buyer:upload-logo")
    data = {
        "csrfmiddlewaretoken": token,
        "company_profile_logo_edit_view-current_step": "logo",
    }
    with open(file_path, "rb") as f:
        picture = f.read()
    mime = mimetypes.MimeTypes().guess_type(file_path)[0]
    files = {"logo-logo": (os.path.basename(file_path), picture, mime)}
    response = make_request(
        Method.POST, url, session=session, headers=headers, data=data,
        files=files)
    return response