edea54c5c3
## Description fix int unable find method .split, not sure why its a int ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin.
32 lines
794 B
Python
32 lines
794 B
Python
from typing import BinaryIO, TypedDict
|
|
import hashlib
|
|
from .guess_file_type import guess_file_type
|
|
from cognee.shared.utils import get_file_content_hash
|
|
|
|
|
|
class FileMetadata(TypedDict):
|
|
name: str
|
|
file_path: str
|
|
mime_type: str
|
|
extension: str
|
|
content_hash: str
|
|
|
|
|
|
def get_file_metadata(file: BinaryIO) -> FileMetadata:
|
|
"""Get metadata from a file"""
|
|
file.seek(0)
|
|
content_hash = get_file_content_hash(file)
|
|
file.seek(0)
|
|
|
|
file_type = guess_file_type(file)
|
|
|
|
file_path = file.name
|
|
file_name = str(file_path).split("/")[-1].split(".")[0] if file_path else None
|
|
|
|
return FileMetadata(
|
|
name=file_name,
|
|
file_path=file_path,
|
|
mime_type=file_type.mime,
|
|
extension=file_type.extension,
|
|
content_hash=content_hash,
|
|
)
|