mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-07 07:55:16 +02:00
96 lines
3 KiB
Python
96 lines
3 KiB
Python
|
import asyncio
|
||
|
|
import os
|
||
|
|
from datetime import datetime
|
||
|
|
from typing import BinaryIO, Optional
|
||
|
|
|
||
|
|
import aiofiles
|
||
|
|
|
||
|
|
from .base import BaseFileSystem
|
||
|
|
|
||
|
|
|
||
|
|
class LocalFileSystem(BaseFileSystem):
|
||
|
|
"""Local filesystem implementation."""
|
||
|
|
|
||
|
|
def __init__(self, base_path: str):
|
||
|
|
"""Initialize local filesystem.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
base_path: Base directory path for file operations
|
||
|
|
"""
|
||
|
|
self.base_path = base_path
|
||
|
|
os.makedirs(base_path, exist_ok=True)
|
||
|
|
|
||
|
|
def _get_full_path(self, file_path: str) -> str:
|
||
|
|
"""Get the full path by joining with base path."""
|
||
|
|
return os.path.join(self.base_path, file_path)
|
||
|
|
|
||
|
|
async def acreate_file(self, file_path: str, content: BinaryIO) -> bool:
|
||
|
|
try:
|
||
|
|
full_path = self._get_full_path(file_path)
|
||
|
|
os.makedirs(os.path.dirname(full_path), exist_ok=True)
|
||
|
|
|
||
|
|
async with aiofiles.open(full_path, "wb") as f:
|
||
|
|
await f.write(await content.read())
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
async def create_temp_file(self, file_path: str) -> bool:
|
||
|
|
try:
|
||
|
|
full_path = self._get_full_path(file_path)
|
||
|
|
os.makedirs(os.path.dirname(full_path), exist_ok=True)
|
||
|
|
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
async def aupload_file(self, local_path: str, destination_path: str) -> bool:
|
||
|
|
try:
|
||
|
|
full_dest_path = self._get_full_path(destination_path)
|
||
|
|
os.makedirs(os.path.dirname(full_dest_path), exist_ok=True)
|
||
|
|
|
||
|
|
async with (
|
||
|
|
aiofiles.open(local_path, "rb") as src,
|
||
|
|
aiofiles.open(full_dest_path, "wb") as dst,
|
||
|
|
):
|
||
|
|
await dst.write(await src.read())
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
async def aget_signed_url(
|
||
|
|
self, file_path: str, expiration: int = 3600
|
||
|
|
) -> Optional[str]:
|
||
|
|
# For local filesystem, we'll create a temporary symlink with expiration
|
||
|
|
try:
|
||
|
|
full_path = self._get_full_path(file_path)
|
||
|
|
if not os.path.exists(full_path):
|
||
|
|
return None
|
||
|
|
|
||
|
|
# Create a temporary directory for symlinks
|
||
|
|
temp_dir = os.path.join(self.base_path, ".temp_links")
|
||
|
|
os.makedirs(temp_dir, exist_ok=True)
|
||
|
|
|
||
|
|
# Generate a unique temporary filename
|
||
|
|
temp_filename = (
|
||
|
|
f"{datetime.now().timestamp()}_{os.path.basename(file_path)}"
|
||
|
|
)
|
||
|
|
temp_path = os.path.join(temp_dir, temp_filename)
|
||
|
|
|
||
|
|
# Create symlink
|
||
|
|
os.symlink(full_path, temp_path)
|
||
|
|
|
||
|
|
# Schedule deletion after expiration
|
||
|
|
async def delete_after_expiration():
|
||
|
|
await asyncio.sleep(expiration)
|
||
|
|
try:
|
||
|
|
os.remove(temp_path)
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
|
||
|
|
asyncio.create_task(delete_after_expiration())
|
||
|
|
|
||
|
|
return f"/files/{temp_filename}"
|
||
|
|
except Exception:
|
||
|
|
return None
|