contentmonster/classes/file.py
2021-11-30 07:25:24 +01:00

96 lines
3.1 KiB
Python

from classes.chunk import Chunk
from classes.database import Database
from typing import Optional
import hashlib
import os.path
class File:
"""Object representing a file found in a local Directory
"""
def __init__(self, name: str, directory, uuid: Optional[str] = None) -> None:
"""Initialize new File object
Args:
name (str): Filename (basename without path) of the File to create
directory (classes.directory.Directory): Directory object the File
is located within
uuid (str, optional): Unique identifier of this File object. Will
be retrieved from database if None. Defaults to None.
Raises:
FileNotFoundError: Raised if the specified File does not exist
"""
self.name = name
self.directory = directory
if not self.exists():
raise FileNotFoundError(f"File {self.name} does not exist in {self.directory.name}!")
self.uuid = uuid or self.getUUID()
def exists(self) -> bool:
"""Check if the File exists on the local file system
Returns:
bool: True if the File exists, else False
"""
return os.path.isfile(self.directory.location / self.name)
def moveCompleted(self) -> None:
self.getFullPath().rename(self.directory.completeddir / self.name)
def getUUID(self) -> str:
"""Return unique identifier for this File object
Returns:
str: File object's UUID retrieved from Database
"""
db = Database()
return db.getFileUUID(self)
def getFullPath(self) -> str:
"""Get the full path of the File
Returns:
str: Full path of the File on the local file system
"""
return self.directory.location / self.name
def getHash(self) -> str:
"""Get hash for this File
Returns:
str: SHA256 for the full content of this File
"""
return self.getChunk(-1).getHash()
def getChunk(self, count: int, size: Optional[int] = None) -> Chunk:
"""Get a Chunk of this File
Args:
count (int): Position of the Chunk in a list of equally large
Chunks (first index: 0). -1 represents a Chunk containing the
complete file.
size (int, optional): Size of the Chunk to create in Bytes. Must
be set if count is not -1. Defaults to None.
Returns:
classes.chunk.Chunk: Chunk object containing File content from
(count * size) bytes to ((count + 1) * size - 1) bytes
Raises:
ValueError: Raised if size is not set for a count that is not -1
"""
if count != -1 and not size:
raise ValueError(
"A Chunk size needs to be passed to getChunk() if not using the complete file (-1)!")
with open(self.getFullPath(), "rb") as binary:
binary.seek((count * size) if count > 0 else 0)
data = binary.read(size if count >= 0 else None)
return Chunk(self, count, data)