fix: intermediate transport commit

This commit is contained in:
Hazel 2024-04-16 17:50:01 +02:00
parent 71d582a17c
commit 0b216b7d40
4 changed files with 29 additions and 31 deletions

View File

@ -26,6 +26,7 @@
"OKBLUE", "OKBLUE",
"Referer", "Referer",
"tracksort", "tracksort",
"unmap",
"youtube" "youtube"
] ]
} }

View File

@ -6,8 +6,8 @@ logging.getLogger().setLevel(logging.DEBUG)
if __name__ == "__main__": if __name__ == "__main__":
commands = [ commands = [
"s: #a And End...", "s: #a Ghost Bath",
"d: 10", "4",
] ]

View File

@ -110,7 +110,7 @@ class Collection(Generic[T]):
if self._contained_in_self(__object): if self._contained_in_self(__object):
return [self] return [self]
for collection in (*self.children, *self.parents): for collection in self.children:
results.extend(collection._contained_in_sub(__object, break_at_first=break_at_first)) results.extend(collection._contained_in_sub(__object, break_at_first=break_at_first))
if break_at_first: if break_at_first:
@ -198,7 +198,7 @@ class Collection(Generic[T]):
if value in self._indexed_values[name]: if value in self._indexed_values[name]:
return self._indexed_to_objects[value][0] return self._indexed_to_objects[value][0]
def _find_object(self, __object: T) -> Tuple[Collection[T], Optional[T]]: def _find_object(self, __object: T, no_sibling: bool = False) -> Tuple[Collection[T], Optional[T]]:
other_object = self._find_object_in_self(__object) other_object = self._find_object_in_self(__object)
if other_object is not None: if other_object is not None:
return self, other_object return self, other_object
@ -208,6 +208,19 @@ class Collection(Generic[T]):
if other_object is not None: if other_object is not None:
return o, other_object return o, other_object
if no_sibling:
return self, None
# find in siblings and all children of siblings
for parent in self.parents:
for sibling in parent.children:
if sibling is self:
continue
o, other_object = sibling._find_object(__object, no_sibling=True)
if other_object is not None:
return o, other_object
return self, None return self, None
def append(self, __object: Optional[T], already_is_parent: bool = False, from_map: bool = False): def append(self, __object: Optional[T], already_is_parent: bool = False, from_map: bool = False):
@ -229,33 +242,14 @@ class Collection(Generic[T]):
if existing_object is None: if existing_object is None:
# append # append
# print("appending", existing_object, __object)
append_to._data.append(__object) append_to._data.append(__object)
append_to._map_element(__object, from_map=from_map)
else: else:
# merge # merge
append_to._unmap_element(existing_object) append_to._unmap_element(existing_object)
existing_object.merge(__object) existing_object.merge(__object)
append_to._map_element(existing_object, from_map=from_map)
append_to._map_element(__object, from_map=from_map)
"""
exists_in_collection = self._contained_in_sub(__object)
if len(exists_in_collection) and self is exists_in_collection[0]:
# assuming that the object already is contained in the correct collections
if not already_is_parent:
self.merge_into_self(__object, from_map=from_map)
return
if not len(exists_in_collection):
self._append(__object, from_map=from_map)
else:
exists_in_collection[0].merge_into_self(__object, from_map=from_map)
if not already_is_parent or not self._is_root:
for parent_collection in self._get_parents_of_multiple_contained_children(__object):
pass
parent_collection.append(__object, already_is_parent=True, from_map=from_map)
"""
def extend(self, __iterable: Optional[Iterable[T]], from_map: bool = False): def extend(self, __iterable: Optional[Iterable[T]], from_map: bool = False):
if __iterable is None: if __iterable is None:
@ -296,8 +290,7 @@ class Collection(Generic[T]):
@property @property
def data(self) -> List[T]: def data(self) -> List[T]:
return [*self._data, return list(i for i in self.__iter__())
*(__object for collection in self.children for __object in collection.shallow_list)]
def __len__(self) -> int: def __len__(self) -> int:
return len(self._data) + sum(len(collection) for collection in self.children) return len(self._data) + sum(len(collection) for collection in self.children)
@ -306,13 +299,17 @@ class Collection(Generic[T]):
def empty(self) -> bool: def empty(self) -> bool:
return self.__len__() <= 0 return self.__len__() <= 0
def __iter__(self) -> Iterator[T]: def __iter__(self, finished_ids: set = None) -> Iterator[T]:
_finished_ids = finished_ids or set()
for element in self._data: for element in self._data:
if element.id in _finished_ids:
continue
_finished_ids.add(element.id)
yield element yield element
for c in self.children: for c in self.children:
for element in c: yield from c.__iter__(finished_ids=finished_ids)
yield element
def __merge__(self, __other: Collection, override: bool = False): def __merge__(self, __other: Collection, override: bool = False):
self.extend(__other._data, from_map=True) self.extend(__other._data, from_map=True)

View File

@ -299,7 +299,7 @@ class Album(Base):
@property @property
def option_string(self) -> str: def option_string(self) -> str:
return f"{self.__repr__()} " \ return f"{self.__repr__()} " \
f"by Artist({OPTION_STRING_DELIMITER.join([artist.name for artist in self.artist_collection])}) " \ f"by Artist({OPTION_STRING_DELIMITER.join([artist.name + str(artist.id) for artist in self.artist_collection])}) " \
f"under Label({OPTION_STRING_DELIMITER.join([label.name for label in self.label_collection])})" f"under Label({OPTION_STRING_DELIMITER.join([label.name for label in self.label_collection])})"
@property @property