mirror of
https://github.com/miaowware/qrm2.git
synced 2024-10-31 14:27:11 -04:00
64 lines
1.4 KiB
Python
64 lines
1.4 KiB
Python
"""
|
|
Resource index models for qrm2.
|
|
---
|
|
Copyright (C) 2021 classabbyamp, 0x5c
|
|
|
|
This file is part of qrm2 and is released under the terms of
|
|
the GNU General Public License, version 2.
|
|
"""
|
|
|
|
|
|
from collections.abc import Mapping
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class File(BaseModel):
|
|
filename: str
|
|
hash: str
|
|
|
|
# For some reason those were not the same???
|
|
def __str__(self) -> str:
|
|
return repr(self)
|
|
|
|
|
|
class Resource(BaseModel, Mapping):
|
|
# 'A Beautiful Hack' https://github.com/samuelcolvin/pydantic/issues/1802
|
|
__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)
|