mirror of https://github.com/miaowware/qrm2.git
62 lines
1.3 KiB
Python
62 lines
1.3 KiB
Python
"""
|
|
Resource index models for qrm2.
|
|
---
|
|
Copyright (C) 2021-2023 classabbyamp, 0x5c
|
|
|
|
SPDX-License-Identifier: LiLiQ-Rplus-1.1
|
|
"""
|
|
|
|
|
|
from collections.abc import Mapping
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, RootModel
|
|
|
|
|
|
class File(BaseModel):
|
|
filename: str
|
|
hash: str
|
|
|
|
# For some reason those were not the same???
|
|
def __str__(self) -> str:
|
|
return repr(self)
|
|
|
|
|
|
class Resource(RootModel, Mapping):
|
|
root: dict[str, list[File]]
|
|
|
|
def __getitem__(self, key: str) -> list[File]:
|
|
return self.root[key]
|
|
|
|
def __iter__(self):
|
|
return iter(self.root)
|
|
|
|
def __len__(self) -> int:
|
|
return len(self.root)
|
|
|
|
# For some reason those were not the same???
|
|
def __str__(self) -> str:
|
|
return repr(self)
|
|
|
|
# Make the repr more logical (despite the technical inaccuracy)
|
|
def __repr_args__(self):
|
|
return self.root.items()
|
|
|
|
|
|
class Index(BaseModel, Mapping):
|
|
last_updated: datetime
|
|
resources: dict[str, Resource]
|
|
|
|
def __getitem__(self, key: str) -> Resource:
|
|
return self.resources[key]
|
|
|
|
def __iter__(self):
|
|
return iter(self.resources)
|
|
|
|
def __len__(self) -> int:
|
|
return len(self.resources)
|
|
|
|
# For some reason those were not the same???
|
|
def __str__(self) -> str:
|
|
return repr(self)
|