Function upload_completion

src/opencdeserver/api/app/api/documents.py:281–344

(
    upload_session: str, current_user: User = Depends(get_current_active_user)
)

@router.post("/documents/1.0/upload-completion", tags=[""])
def upload_completion(
    upload_session: str, current_user: User = Depends(get_current_active_user)
) -> Union[DocumentVersion, bool]:

    # check if all parts really are marked as uploaded in database
    # retrieve document_id, file_type, file_ending, and parts_id (in order)

    if not doc_db.all_parts_uploaded(upload_session, current_user):
        raise HTTPException(status_code=400, detail="All parts not uploaded.")

    parts = doc_db.retrieve_uploaded_parts(upload_session, current_user)
    print("Number of parts: " + str(len(parts)))
    document = doc_db.get_document_from_session(upload_session, current_user)

    # check if all parts really are uploaded to document_id-dir
    document_name = doc_db.safe_path(document.document_id)

    path = "./data/document_parts/" + document_name + "/"

    for part in parts:
        part = doc_db.safe_path(part)
        print("Checking for part " + part + " in dir " + path)

        if not os.path.isfile(path + part):
            print(part + " is not in dir " + path)
            raise HTTPException(status_code=400, detail="All parts not in dir.")
        else:
            print(part + " is in dir " + path)

    # merge parts to a new temporary document
    temp_doc_path = path
    temp_doc_file_name = path + document_name
    if not os.path.exists(temp_doc_path):
        os.makedirs(temp_doc_path)

    new_doc_path = "./data/documents/"
    new_doc_path_name = new_doc_path + document.file_description.name

    # Read parts and write to temp doc.
    with open(temp_doc_file_name, "ab") as temp_doc:
        for part in parts:
            part = doc_db.safe_path(part)
            with open(temp_doc_path + part, "rb") as part_doc:
                temp_doc.write(part_doc.read())

    # move document to new location in documents dir
    os.rename(temp_doc_file_name, new_doc_path_name)

    # remove temp parts
    for part in parts:
        os.remove(temp_doc_path + part)

    # remove temp path
    os.rmdir(temp_doc_path)

    # clean database and create new document node under correct project
    if doc_db.document_upload_finish(upload_session, current_user, document.project):
        # get DocumentVersion