41 lines
988 B
Python
41 lines
988 B
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from enum import Enum, auto
|
|
|
|
import gi
|
|
|
|
gi.require_version("GObject", "2.0")
|
|
from gi.repository import GObject # NOQA: E402
|
|
|
|
|
|
class FileType(Enum):
|
|
DIRECTORY = auto()
|
|
VIDEO = auto()
|
|
|
|
|
|
class FileItem(GObject.Object):
|
|
file_type: FileType
|
|
full_path: str
|
|
|
|
__gtype_name__ = "FileItem"
|
|
|
|
def __init__(self, name: str, file_type: FileType, full_path: str):
|
|
super().__init__()
|
|
self.name = name
|
|
self.file_type = file_type
|
|
self.full_path = full_path
|
|
|
|
@staticmethod
|
|
def from_path(path: str) -> FileItem:
|
|
name = os.path.basename(path)
|
|
if path.endswith("/"):
|
|
return FileItem(name, FileType.DIRECTORY, path)
|
|
|
|
parts = name.split(".")
|
|
suffix = parts[-1].lower() if len(parts) >= 2 else ""
|
|
|
|
if suffix in ("mkv", "mp4", "avi"):
|
|
return FileItem(name, FileType.VIDEO, path)
|
|
|
|
raise ValueError(f"Unsupported file type: {path}")
|