mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-16 01:55:15 +03:00
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import hashlib
|
|
|
|
from fastapi import UploadFile
|
|
|
|
|
|
def convert_bytes(bytes, precision=2):
|
|
"""Converts bytes into a human-friendly format."""
|
|
abbreviations = ["B", "KB", "MB"]
|
|
if bytes <= 0:
|
|
return "0 B"
|
|
size = bytes
|
|
index = 0
|
|
while size >= 1024 and index < len(abbreviations) - 1:
|
|
size /= 1024
|
|
index += 1
|
|
return f"{size:.{precision}f} {abbreviations[index]}"
|
|
|
|
|
|
def get_file_size(file: UploadFile):
|
|
# move the cursor to the end of the file
|
|
file.file._file.seek(0, 2) # pyright: ignore reportPrivateUsage=none
|
|
file_size = (
|
|
file.file._file.tell() # pyright: ignore reportPrivateUsage=none
|
|
) # Getting the size of the file
|
|
# move the cursor back to the beginning of the file
|
|
file.file.seek(0)
|
|
|
|
return file_size
|
|
|
|
|
|
def compute_sha1_from_file(file_path):
|
|
with open(file_path, "rb") as file:
|
|
bytes = file.read()
|
|
readable_hash = compute_sha1_from_content(bytes)
|
|
return readable_hash
|
|
|
|
|
|
def compute_sha1_from_content(content):
|
|
readable_hash = hashlib.sha1(content).hexdigest()
|
|
return readable_hash
|