FrederikBaerentsen/BrickTracker
4
8
Fork 6
Code Issues 7 Pull Requests 1 Actions Packages Projects 1 Releases 4 Wiki Activity

Fix None being submitted to a metadata get()

Browse Source
This commit is contained in:
Gregoo 2025-02-08 10:15:54 +01:00
parent fa3a321b9e
commit 3496143962
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bricktracker/metadata_list.py
View File

@ -3,7 +3,7 @@ from typing import List, overload, Self, Type, TypeVar
from flask import url_for
from .exceptions import NotFoundException
from .exceptions import ErrorException, NotFoundException
from .fields import BrickRecordFields
from .record_list import BrickRecordList
from .set_owner import BrickSetOwner
@ -113,12 +113,17 @@ class BrickMetadataList(BrickRecordList[T]):
# Grab a specific status
@classmethod
def get(cls, id: str, /, *, allow_none: bool = False) -> T:
def get(cls, id: str | None, /, *, allow_none: bool = False) -> T:
new = cls.new()
if allow_none and id == '':
if allow_none and (id == '' or id is None):
return new.model()
if id is None:
raise ErrorException('Cannot get {kind} with no ID'.format(
kind=new.kind.capitalize()
))
if id not in new.mapping:
raise NotFoundException(
'{kind} with ID {id} was not found in the database'.format(