lfss 0.8.1__tar.gz → 0.8.2__tar.gz

{lfss-0.8.1 → lfss-0.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lfss
-Version: 0.8.1
+Version: 0.8.2
 Summary: Lightweight file storage service
 Home-page: https://github.com/MenxLi/lfss
 Author: li, mengxun

{lfss-0.8.1 → lfss-0.8.2}/lfss/api/__init__.py

@@ -16,6 +16,11 @@ def upload_file(
     ) -> tuple[bool, str]:
     this_try = 0
     error_msg = ""
+    assert not file_path.endswith('/'), "File path must not end with a slash."
+    if dst_url.endswith('/'):
+        fname = file_path.split('/')[-1]
+        dst_url = f"{dst_url}{fname}"
+
     while this_try <= n_retries:
         try:
             fsize = os.path.getsize(file_path)
@@ -31,7 +36,9 @@ def upload_file(
                 raise e
             if verbose:
                 print(f"Error uploading {file_path}: {e}, retrying...")
-                error_msg = str(e)
+            error_msg = str(e)
+            if hasattr(e, 'response'):
+                error_msg = f"{error_msg}, {e.response.text}"   # type: ignore
             this_try += 1
         finally:
             time.sleep(interval)
@@ -95,7 +102,12 @@ def download_file(
     ) -> tuple[bool, str]:
     this_try = 0
     error_msg = ""
+    assert not src_url.endswith('/'), "Source URL must not end with a slash."
     while this_try <= n_retries:
+        if os.path.isdir(file_path):
+            fname = src_url.split('/')[-1]
+            file_path = os.path.join(file_path, fname)
+
         if not overwrite and os.path.exists(file_path):
             if verbose:
                 print(f"File {file_path} already exists, skipping download.")
@@ -124,7 +136,9 @@ def download_file(
                 raise e
             if verbose:
                 print(f"Error downloading {src_url}: {e}, retrying...")
-                error_msg = str(e)
+            error_msg = str(e)
+            if hasattr(e, 'response'):
+                error_msg = f"{error_msg}, {e.response.text}"   # type: ignore
             this_try += 1
         finally:
             time.sleep(interval)

lfss-0.8.2/lfss/cli/__init__.py

@@ -0,0 +1,27 @@
+from contextlib import contextmanager
+from typing import Iterable, TypeVar, Generator
+import requests, os
+
+@contextmanager
+def catch_request_error():
+    try:
+        yield
+    except requests.RequestException as e:
+        print(f"\033[31m[Request error]: {e}\033[0m")
+        if e.response is not None:
+            print(f"\033[91m[Error message]: {e.response.text}\033[0m")
+
+T = TypeVar('T')
+def line_sep(iter: Iterable[T], enable=True, start=True, end=True, color="\033[90m") -> Generator[T, None, None]:
+    screen_width = os.get_terminal_size().columns
+    def print_ln():
+        print(color + "-" * screen_width + "\033[0m")
+
+    if start and enable:
+        print_ln()
+    for i, line in enumerate(iter):
+        if enable and i > 0:
+            print_ln()
+        yield line
+    if end and enable:
+        print_ln()

{lfss-0.8.1 → lfss-0.8.2}/lfss/cli/cli.py

@@ -1,7 +1,9 @@
 from lfss.api import Connector, upload_directory, upload_file, download_file, download_directory
 from pathlib import Path
-import argparse
-from lfss.src.datatype import FileReadPermission
+import argparse, typing
+from lfss.src.datatype import FileReadPermission, FileSortKey, DirSortKey
+from lfss.src.utils import decode_uri_compnents
+from . import catch_request_error, line_sep
 
 def parse_permission(s: str) -> FileReadPermission:
     if s.lower() == "public":
@@ -39,6 +41,29 @@ def parse_arguments():
     sp_download.add_argument("--overwrite", action="store_true", help="Overwrite existing files")
     sp_download.add_argument("--retries", type=int, default=0, help="Number of retries")
 
+    # query
+    sp_query = sp.add_parser("query", help="Query files or directories metadata from the server")
+    sp_query.add_argument("path", help="Path to query", nargs="*", type=str)
+
+    # list directories
+    sp_list_d = sp.add_parser("list-dirs", help="List directories of a given path")
+    sp_list_d.add_argument("path", help="Path to list", type=str)
+    sp_list_d.add_argument("--offset", type=int, default=0, help="Offset of the list")
+    sp_list_d.add_argument("--limit", type=int, default=100, help="Limit of the list")
+    sp_list_d.add_argument("-l", "--long", action="store_true", help="Detailed list, including all metadata")
+    sp_list_d.add_argument("--order", "--order-by", type=str, help="Order of the list", default="", choices=typing.get_args(DirSortKey))
+    sp_list_d.add_argument("--reverse", "--order-desc", action="store_true", help="Reverse the list order")
+
+    # list files
+    sp_list_f = sp.add_parser("list-files", help="List files of a given path")
+    sp_list_f.add_argument("path", help="Path to list", type=str)
+    sp_list_f.add_argument("--offset", type=int, default=0, help="Offset of the list")
+    sp_list_f.add_argument("--limit", type=int, default=100, help="Limit of the list")
+    sp_list_f.add_argument("-r", "--recursive", "--flat", action="store_true", help="List files recursively")
+    sp_list_f.add_argument("-l", "--long", action="store_true", help="Detailed list, including all metadata")
+    sp_list_f.add_argument("--order", "--order-by", type=str, help="Order of the list", default="", choices=typing.get_args(FileSortKey))
+    sp_list_f.add_argument("--reverse", "--order-desc", action="store_true", help="Reverse the list order")
+
     return parser.parse_args()
 
 def main():
@@ -57,11 +82,11 @@ def main():
                 permission=args.permission
             )
             if failed_upload:
-                print("Failed to upload:")
+                print("\033[91mFailed to upload:\033[0m")
                 for path in failed_upload:
                     print(f"  {path}")
         else:
-            success = upload_file(
+            success, msg = upload_file(
                 connector, 
                 file_path = args.src, 
                 dst_url = args.dst, 
@@ -72,7 +97,7 @@ def main():
                 permission=args.permission
             )
             if not success:
-                print("Failed to upload.")
+                print("\033[91mFailed to upload: \033[0m", msg)
     
     elif args.command == "download":
         is_dir = args.src.endswith("/")
@@ -86,11 +111,11 @@ def main():
                 overwrite=args.overwrite
             )
             if failed_download:
-                print("Failed to download:")
+                print("\033[91mFailed to download:\033[0m")
                 for path in failed_download:
                     print(f"  {path}")
         else:
-            success = download_file(
+            success, msg = download_file(
                 connector, 
                 src_url = args.src, 
                 file_path = args.dst, 
@@ -100,7 +125,51 @@ def main():
                 overwrite=args.overwrite
             )
             if not success:
-                print("Failed to download.")
+                print("\033[91mFailed to download: \033[0m", msg)
+    
+    elif args.command == "query":
+        for path in args.path:
+            with catch_request_error():
+                res = connector.get_metadata(path)
+                if res is None:
+                    print(f"\033[31mNot found\033[0m ({path})")
+                else:
+                    print(res)
+    
+    elif args.command == "list-files":
+        with catch_request_error():
+            res = connector.list_files(
+                args.path, 
+                offset=args.offset, 
+                limit=args.limit, 
+                flat=args.recursive, 
+                order_by=args.order,
+                order_desc=args.reverse,
+            )
+            for i, f in enumerate(line_sep(res)):
+                f.url = decode_uri_compnents(f.url)
+                print(f"[{i+1}] {f if args.long else f.url}")
+
+            if len(res) == args.limit:
+                print(f"\033[33m[Warning] List limit reached, use --offset and --limit to list more files.")
+        
+    elif args.command == "list-dirs":
+        with catch_request_error():
+            res = connector.list_dirs(
+                args.path, 
+                offset=args.offset, 
+                limit=args.limit, 
+                skim=not args.long, 
+                order_by=args.order,
+                order_desc=args.reverse,
+            )
+            for i, d in enumerate(line_sep(res)):
+                d.url = decode_uri_compnents(d.url)
+                print(f"[{i+1}] {d if args.long else d.url}")
+
+            if len(res) == args.limit:
+                print(f"\033[33m[Warning] List limit reached, use --offset and --limit to list more directories.")
+    
     else:
         raise NotImplementedError(f"Command {args.command} not implemented.")
     

{lfss-0.8.1 → lfss-0.8.2}/lfss/cli/user.py

@@ -1,8 +1,8 @@
-import argparse, asyncio
+import argparse, asyncio, os
 from contextlib import asynccontextmanager
 from .cli import parse_permission, FileReadPermission
-from ..src.utils import parse_storage_size
-from ..src.database import Database, FileReadPermission, transaction, UserConn
+from ..src.utils import parse_storage_size, fmt_storage_size
+from ..src.database import Database, FileReadPermission, transaction, UserConn, unique_cursor, FileConn
 from ..src.connection_pool import global_entrance
 
 @global_entrance(1)
@@ -33,6 +33,7 @@ async def _main():
     sp_set.add_argument('--max-storage', type=parse_storage_size, default=None)
     
     sp_list = sp.add_parser('list')
+    sp_list.add_argument("username", nargs='*', type=str, default=None)
     sp_list.add_argument("-l", "--long", action="store_true")
     
     args = parser.parse_args()
@@ -73,10 +74,18 @@ async def _main():
     
     if args.subparser_name == 'list':
         async with get_uconn() as uconn:
+            term_width = os.get_terminal_size().columns
             async for user in uconn.all():
+                if args.username and not user.username in args.username:
+                    continue
+                print("\033[90m-\033[0m" * term_width)
                 print(user)
                 if args.long:
-                    print('  ', user.credential)
+                    async with unique_cursor() as c:
+                        fconn = FileConn(c)
+                        user_size_used = await fconn.user_size(user.id)
+                    print('- Credential: ', user.credential)
+                    print(f'- Storage: {fmt_storage_size(user_size_used)} / {fmt_storage_size(user.max_storage)}')
         
 def main():
     asyncio.run(_main())

{lfss-0.8.1 → lfss-0.8.2}/lfss/src/database.py

@@ -21,6 +21,12 @@ from .utils import decode_uri_compnents, hash_credential, concurrent_wrap, debou
 from .error import *
 
 class DBObjectBase(ABC):
+    """
+    NOTE: 
+    The object of this class should hold a cursor to the database. 
+    The methods calling the cursor should not be called concurrently. 
+    """
+
     logger = get_logger('database', global_instance=True)
     _cur: aiosqlite.Cursor
 
@@ -206,7 +212,7 @@ class FileConn(DBObjectBase):
                 return DirectoryRecord(dir_url)
             else:
                 return await self.get_path_record(dir_url)
-        dirs = await asyncio.gather(*[get_dir(url + d) for d in dirs_str])
+        dirs = [await get_dir(url + d) for d in dirs_str]
         return dirs
     
     async def count_path_files(self, url: str, flat: bool = False):
@@ -308,18 +314,14 @@ class FileConn(DBObjectBase):
         return res[0] or 0
     
     async def update_file_record(
-        self, url, owner_id: Optional[int] = None, permission: Optional[FileReadPermission] = None
+        self, url, 
+        permission: Optional[FileReadPermission] = None, 
+        mime_type: Optional[str] = None
         ):
-        old = await self.get_file_record(url)
-        assert old is not None, f"File {url} not found"
-        if owner_id is None:
-            owner_id = old.owner_id
-        if permission is None:
-            permission = old.permission
-        await self.cur.execute(
-            "UPDATE fmeta SET owner_id = ?, permission = ? WHERE url = ?", 
-            (owner_id, int(permission), url)
-            )
+        if permission is not None:
+            await self.cur.execute("UPDATE fmeta SET permission = ? WHERE url = ?", (int(permission), url))
+        if mime_type is not None:
+            await self.cur.execute("UPDATE fmeta SET mime_type = ? WHERE url = ?", (mime_type, url))
         self.logger.info(f"Updated file {url}")
     
     async def set_file_record(
@@ -392,7 +394,7 @@ class FileConn(DBObjectBase):
         self.logger.info(f"Deleted {len(ret)} file records for user {owner_id}") # type: ignore
         return ret
     
-    async def delete_path_records(self, path: str, under_user_id: Optional[int] = None) -> list[FileRecord]:
+    async def delete_path_records(self, path: str, under_owner_id: Optional[int] = None) -> list[FileRecord]:
         """Delete all records with url starting with path"""
         # update user size
         cursor = await self.cur.execute("SELECT DISTINCT owner_id FROM fmeta WHERE url LIKE ?", (path + '%', ))
@@ -406,10 +408,10 @@ class FileConn(DBObjectBase):
         # if any new records are created here, the size update may be inconsistent
         # but it's not a big deal... we should have only one writer
         
-        if under_user_id is None:
+        if under_owner_id is None:
             res = await self.cur.execute("DELETE FROM fmeta WHERE url LIKE ? RETURNING *", (path + '%', ))
         else:
-            res = await self.cur.execute("DELETE FROM fmeta WHERE url LIKE ? AND owner_id = ? RETURNING *", (path + '%', under_user_id))
+            res = await self.cur.execute("DELETE FROM fmeta WHERE url LIKE ? AND owner_id = ? RETURNING *", (path + '%', under_owner_id))
         all_f_rec = await res.fetchall()
         self.logger.info(f"Deleted {len(all_f_rec)} file(s) for path {path}") # type: ignore
         return [self.parse_record(r) for r in all_f_rec]
@@ -521,15 +523,16 @@ class Database:
             await execute_sql(conn, 'init.sql')
         return self
     
-    async def update_file_record(self, user: UserRecord, url: str, permission: FileReadPermission):
+    async def update_file_record(self, url: str, permission: FileReadPermission, op_user: Optional[UserRecord] = None):
         validate_url(url)
         async with transaction() as conn:
             fconn = FileConn(conn)
             r = await fconn.get_file_record(url)
             if r is None:
                 raise PathNotFoundError(f"File {url} not found")
-            if r.owner_id != user.id and not user.is_admin:
-                raise PermissionDeniedError(f"Permission denied: {user.username} cannot update file {url}")
+            if op_user is not None:
+                if r.owner_id != op_user.id and not op_user.is_admin:
+                    raise PermissionDeniedError(f"Permission denied: {op_user.username} cannot update file {url}")
             await fconn.update_file_record(url, permission=permission)
     
     async def save_file(
@@ -561,8 +564,7 @@ class Database:
                 
                 # check mime type
                 if mime_type is None:
-                    fname = url.split('/')[-1]
-                    mime_type, _ = mimetypes.guess_type(fname)
+                    mime_type, _ = mimetypes.guess_type(url)
                 if mime_type is None:
                     await f.seek(0)
                     mime_type = mimesniff.what(await f.read(1024))
@@ -591,8 +593,6 @@ class Database:
                         await FileConn(w_cur).set_file_record(
                             url, owner_id=user.id, file_id=f_id, file_size=file_size, 
                             permission=permission, external=True, mime_type=mime_type)
-
-        await delayed_log_activity(user.username)
         return file_size
 
     async def read_file_stream(self, url: str) -> AsyncIterable[bytes]:
@@ -605,11 +605,8 @@ class Database:
             if not r.external:
                 raise ValueError(f"File {url} is not stored externally, should use read_file instead")
             ret = fconn.get_file_blob_external(r.file_id)
-
-        await delayed_log_access(url)
         return ret
 
-
     async def read_file(self, url: str) -> bytes:
         validate_url(url)
 
@@ -625,11 +622,9 @@ class Database:
             blob = await fconn.get_file_blob(f_id)
             if blob is None:
                 raise FileNotFoundError(f"File {url} data not found")
-
-        await delayed_log_access(url)
         return blob
 
-    async def delete_file(self, url: str, assure_user: Optional[UserRecord] = None) -> Optional[FileRecord]:
+    async def delete_file(self, url: str, op_user: Optional[UserRecord] = None) -> Optional[FileRecord]:
         validate_url(url)
 
         async with transaction() as cur:
@@ -637,10 +632,10 @@ class Database:
             r = await fconn.delete_file_record(url)
             if r is None:
                 return None
-            if assure_user is not None:
-                if r.owner_id != assure_user.id:
+            if op_user is not None:
+                if r.owner_id != op_user.id and not op_user.is_admin:
                     # will rollback
-                    raise PermissionDeniedError(f"Permission denied: {assure_user.username} cannot delete file {url}")
+                    raise PermissionDeniedError(f"Permission denied: {op_user.username} cannot delete file {url}")
             f_id = r.file_id
             if r.external:
                 await fconn.delete_file_blob_external(f_id)
@@ -648,7 +643,7 @@ class Database:
                 await fconn.delete_file_blob(f_id)
             return r
     
-    async def move_file(self, old_url: str, new_url: str, ensure_user: Optional[UserRecord] = None):
+    async def move_file(self, old_url: str, new_url: str, op_user: Optional[UserRecord] = None):
         validate_url(old_url)
         validate_url(new_url)
 
@@ -657,12 +652,16 @@ class Database:
             r = await fconn.get_file_record(old_url)
             if r is None:
                 raise FileNotFoundError(f"File {old_url} not found")
-            if ensure_user is not None:
-                if r.owner_id != ensure_user.id:
-                    raise PermissionDeniedError(f"Permission denied: {ensure_user.username} cannot move file {old_url}")
+            if op_user is not None:
+                if r.owner_id != op_user.id:
+                    raise PermissionDeniedError(f"Permission denied: {op_user.username} cannot move file {old_url}")
             await fconn.move_file(old_url, new_url)
+
+            new_mime, _ = mimetypes.guess_type(new_url)
+            if not new_mime is None:
+                await fconn.update_file_record(new_url, mime_type=new_mime)
     
-    async def move_path(self, user: UserRecord, old_url: str, new_url: str):
+    async def move_path(self, old_url: str, new_url: str, op_user: UserRecord):
         validate_url(old_url, is_file=False)
         validate_url(new_url, is_file=False)
 
@@ -676,20 +675,20 @@ class Database:
 
         async with transaction() as cur:
             first_component = new_url.split('/')[0]
-            if not (first_component == user.username or user.is_admin):
-                raise PermissionDeniedError(f"Permission denied: path must start with {user.username}")
-            elif user.is_admin:
+            if not (first_component == op_user.username or op_user.is_admin):
+                raise PermissionDeniedError(f"Permission denied: path must start with {op_user.username}")
+            elif op_user.is_admin:
                 uconn = UserConn(cur)
                 _is_user = await uconn.get_user(first_component)
                 if not _is_user:
                     raise PermissionDeniedError(f"Invalid path: {first_component} is not a valid username")
             
             # check if old path is under user's directory (non-admin)
-            if not old_url.startswith(user.username + '/') and not user.is_admin:
-                raise PermissionDeniedError(f"Permission denied: {user.username} cannot move path {old_url}")
+            if not old_url.startswith(op_user.username + '/') and not op_user.is_admin:
+                raise PermissionDeniedError(f"Permission denied: {op_user.username} cannot move path {old_url}")
 
             fconn = FileConn(cur)
-            await fconn.move_path(old_url, new_url, 'overwrite', user.id)
+            await fconn.move_path(old_url, new_url, 'overwrite', op_user.id)
 
     async def __batch_delete_file_blobs(self, fconn: FileConn, file_records: list[FileRecord], batch_size: int = 512):
         # https://github.com/langchain-ai/langchain/issues/10321
@@ -709,13 +708,13 @@ class Database:
                 await fconn.delete_file_blob_external(external_ids[i])
         await asyncio.gather(del_internal(), del_external())
 
-    async def delete_path(self, url: str, under_user: Optional[UserRecord] = None) -> Optional[list[FileRecord]]:
+    async def delete_path(self, url: str, op_user: Optional[UserRecord] = None) -> Optional[list[FileRecord]]:
         validate_url(url, is_file=False)
-        user_id = under_user.id if under_user is not None else None
+        from_owner_id = op_user.id if op_user is not None and not op_user.is_admin else None
 
         async with transaction() as cur:
             fconn = FileConn(cur)
-            records = await fconn.delete_path_records(url, user_id)
+            records = await fconn.delete_path_records(url, from_owner_id)
             if not records:
                 return None
             await self.__batch_delete_file_blobs(fconn, records)

{lfss-0.8.1 → lfss-0.8.2}/lfss/src/datatype.py

@@ -1,6 +1,5 @@
 from enum import IntEnum
-from typing import Literal
-import dataclasses
+import dataclasses, typing
 
 class FileReadPermission(IntEnum):
     UNSET = 0           # not set
@@ -55,7 +54,7 @@ class PathContents:
     dirs: list[DirectoryRecord] = dataclasses.field(default_factory=list)
     files: list[FileRecord] = dataclasses.field(default_factory=list)
     
-FileSortKey = Literal['', 'url', 'file_size', 'create_time', 'access_time', 'mime_type']
-isValidFileSortKey = lambda x: x in ['', 'url', 'file_size', 'create_time', 'access_time', 'mime_type']
-DirSortKey = Literal['', 'dirname']
-isValidDirSortKey = lambda x: x in ['', 'dirname']
+FileSortKey = typing.Literal['', 'url', 'file_size', 'create_time', 'access_time', 'mime_type']
+isValidFileSortKey = lambda x: x in typing.get_args(FileSortKey)
+DirSortKey = typing.Literal['', 'dirname']
+isValidDirSortKey = lambda x: x in typing.get_args(DirSortKey)

{lfss-0.8.1 → lfss-0.8.2}/lfss/src/server.py

@@ -6,7 +6,6 @@ from fastapi.responses import StreamingResponse
 from fastapi.exceptions import HTTPException 
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 from fastapi.middleware.cors import CORSMiddleware
-import mimesniff
 
 import asyncio, json, time
 from contextlib import asynccontextmanager
@@ -14,10 +13,11 @@ from contextlib import asynccontextmanager
 from .error import *
 from .log import get_logger
 from .stat import RequestDB
-from .config import MAX_BUNDLE_BYTES, MAX_MEM_FILE_BYTES, LARGE_FILE_BYTES, CHUNK_SIZE
+from .config import MAX_BUNDLE_BYTES, CHUNK_SIZE
 from .utils import ensure_uri_compnents, format_last_modified, now_stamp, wait_for_debounce_tasks
 from .connection_pool import global_connection_init, global_connection_close, unique_cursor
-from .database import Database, DECOY_USER, check_user_permission, UserConn, FileConn, delayed_log_activity, get_user
+from .database import Database, DECOY_USER, check_user_permission, UserConn, FileConn
+from .database import delayed_log_activity, delayed_log_access
 from .datatype import (
     FileReadPermission, FileRecord, UserRecord, PathContents, 
     FileSortKey, DirSortKey
@@ -80,6 +80,10 @@ async def get_current_user(
 
     if not user:
         raise HTTPException(status_code=401, detail="Invalid token")
+    
+    if not user.id == 0:
+        await delayed_log_activity(user.username)
+
     return user
 
 async def registered_user(user: UserRecord = Depends(get_current_user)):
@@ -165,6 +169,8 @@ async def emit_file(
         media_type = file_record.mime_type
     path = file_record.url
     fname = path.split("/")[-1]
+
+    await delayed_log_access(path)
     if not file_record.external:
         fblob = await db.read_file(path)
         return Response(
@@ -360,11 +366,10 @@ async def delete_file(path: str, user: UserRecord = Depends(registered_user)):
     logger.info(f"DELETE {path}, user: {user.username}")
 
     if path.endswith("/"):
-        res = await db.delete_path(path, user if not user.is_admin else None)
+        res = await db.delete_path(path, user)
     else:
-        res = await db.delete_file(path, user if not user.is_admin else None)
+        res = await db.delete_file(path, user)
 
-    await delayed_log_activity(user.username)
     if res:
         return Response(status_code=200, content="Deleted")
     else:
@@ -456,16 +461,15 @@ async def update_file_meta(
     path = ensure_uri_compnents(path)
     if path.startswith("/"):
         path = path[1:]
-    await delayed_log_activity(user.username)
 
     # file
     if not path.endswith("/"):
         if perm is not None:
             logger.info(f"Update permission of {path} to {perm}")
             await db.update_file_record(
-                user = user,
                 url = path, 
-                permission = FileReadPermission(perm)
+                permission = FileReadPermission(perm), 
+                op_user = user,
             )
     
         if new_path is not None:
@@ -480,7 +484,7 @@ async def update_file_meta(
             new_path = ensure_uri_compnents(new_path)
             logger.info(f"Update path of {path} to {new_path}")
             # currently only move own file, with overwrite
-            await db.move_path(user, path, new_path)
+            await db.move_path(path, new_path, user)
 
     return Response(status_code=200, content="OK")
 

{lfss-0.8.1 → lfss-0.8.2}/lfss/src/utils.py

@@ -109,6 +109,17 @@ def parse_storage_size(s: str) -> int:
         case 'g': return int(s[:-1]) * 1024**3
         case 't': return int(s[:-1]) * 1024**4
         case _: raise ValueError(f"Invalid file size string: {s}")
+def fmt_storage_size(size: int) -> str:
+    """ Format the file size to human-readable format """
+    if size < 1024:
+        return f"{size}B"
+    if size < 1024**2:
+        return f"{size/1024:.2f}K"
+    if size < 1024**3:
+        return f"{size/1024**2:.2f}M"
+    if size < 1024**4:
+        return f"{size/1024**3:.2f}G"
+    return f"{size/1024**4:.2f}T"
 
 _FnReturnT = TypeVar('_FnReturnT')
 _AsyncReturnT = Awaitable[_FnReturnT]
@@ -136,4 +147,12 @@ def concurrent_wrap(executor=None):
             loop = asyncio.new_event_loop()
             return loop.run_until_complete(func(*args, **kwargs))
         return sync_fn
-    return _concurrent_wrap
+    return _concurrent_wrap
+
+# https://stackoverflow.com/a/279586/6775765
+def static_vars(**kwargs):
+    def decorate(func):
+        for k in kwargs:
+            setattr(func, k, kwargs[k])
+        return func
+    return decorate

{lfss-0.8.1 → lfss-0.8.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lfss"
-version = "0.8.1"
+version = "0.8.2"
 description = "Lightweight file storage service"
 authors = ["li, mengxun <limengxun45@outlook.com>"]
 readme = "Readme.md"