lazy-player/lazy_player/file_model.py

100 lines
2.6 KiB
Python
Raw Normal View History

2025-03-08 23:34:19 +01:00
from __future__ import annotations
import os
from enum import Enum, auto
2025-03-09 11:49:42 +01:00
from pathlib import Path
from typing import Optional, overload
2025-03-08 23:34:19 +01:00
import gi
gi.require_version("GObject", "2.0")
from gi.repository import Gio, GObject # NOQA: E402
2025-03-08 23:34:19 +01:00
class FileType(Enum):
DIRECTORY = auto()
VIDEO = auto()
class FileItem(GObject.Object):
2025-03-09 09:51:10 +01:00
file_type: FileType
2025-03-09 11:49:42 +01:00
full_path: Path
2025-03-09 09:51:10 +01:00
2025-03-08 23:34:19 +01:00
__gtype_name__ = "FileItem"
2025-03-09 10:52:34 +01:00
attrs_changed = GObject.Property(type=int, default=0)
2025-03-09 11:49:42 +01:00
def __init__(self, name: str, file_type: FileType, full_path: Path):
2025-03-08 23:34:19 +01:00
super().__init__()
2025-03-09 10:52:34 +01:00
self.attrs_changed = 0
2025-03-08 23:34:19 +01:00
self.name = name
self.file_type = file_type
self.full_path = full_path
2025-03-09 09:57:24 +01:00
@overload
def load_attribute(self, name: str, dfl: str) -> str: ...
@overload
def load_attribute(self, name: str, dfl: int) -> int: ...
def load_attribute(self, name: str, dfl: str | int) -> str | int:
try:
strval = os.getxattr(self.full_path, f"user.lazy_player.{name}")
return type(dfl)(strval)
2025-03-09 10:08:26 +01:00
except OSError:
2025-03-09 09:57:24 +01:00
return dfl
def save_attribute(self, name: str, value: str | float | int | None) -> None:
try:
if value is None:
os.removexattr(self.full_path, f"user.lazy_player.{name}")
else:
os.setxattr(self.full_path, f"user.lazy_player.{name}", str(value).encode("utf8"))
2025-03-09 10:08:26 +01:00
except OSError:
pass
2025-03-09 10:52:34 +01:00
self.notify("attrs-changed")
class FileListModel(GObject.Object, Gio.ListModel):
"""A ListModel implementation for FileItems"""
__gtype_name__ = "FileListModel"
items: list[FileItem]
def __init__(self):
super().__init__()
self.items = []
def do_get_item_type(self) -> GObject.GType:
return GObject.type_from_name("FileItem")
def do_get_n_items(self) -> int:
return len(self.items)
def do_get_item(self, position: int) -> Optional[FileItem]:
if 0 <= position < len(self.items):
return self.items[position]
return None
def remove_all(self) -> None:
removed = len(self.items)
self.items = []
self.items_changed(0, removed, 0)
def set_items(self, items: list[FileItem]):
removed = len(self.items)
self.items = list(items)
self.items_changed(0, removed, len(self.items))
def append(self, item: FileItem) -> None:
pos = len(self.items)
self.items.append(item)
self.items_changed(pos, 0, 1)
def remove(self, position: int) -> None:
if 0 <= position < len(self.items):
self.items.pop(position)
self.items_changed(position, 1, 0)