Zergling_man/moment
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Working python proxy/filter for room list

Browse Source
This commit is contained in:
miruka
2020-05-06 01:49:25 -04:00
parent eee198b238
commit 5432958121
7 changed files with 101 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/backend/backend.py
View File

@@ -7,16 +7,20 @@ import traceback
from pathlib import Path
from typing import Any, DefaultDict, Dict, List, Optional
import nio
from appdirs import AppDirs
import nio
from . import __app_name__
from .errors import MatrixError
from .matrix_client import MatrixClient
from .media_cache import MediaCache
from .models import SyncId
from .models.filters import FieldSubstringFilter
from .models.items import Account
from .models.model import Model
from .models.model_store import ModelStore
from .models.special_models import AllRooms
from .user_files import Accounts, History, Theme, UISettings, UIState
# Logging configuration
@@ -77,8 +81,9 @@ class Backend:
self.ui_state: UIState = UIState(self)
self.history: History = History(self)
self.models: ModelStore = ModelStore()
self.clients: Dict[str, MatrixClient] = {}
self.all_rooms: AllRooms = AllRooms()
self.models: ModelStore = ModelStore()
self.clients: Dict[str, MatrixClient] = {}
self.profile_cache: Dict[str, nio.ProfileGetResponse] = {}
self.get_profile_locks: DefaultDict[str, asyncio.Lock] = \
@@ -304,3 +309,12 @@ class Backend:
await asyncio.sleep(0.1)
failures += 1
async def set_substring_filter(self, model_id: SyncId, value: str) -> None:
model = Model.proxies[model_id]
if not isinstance(model, FieldSubstringFilter):
raise TypeError("model_id must point to a FieldSubstringFilter")
model.filter = value

4
src/backend/matrix_client.py
View File

@@ -871,7 +871,6 @@ class MatrixClient(nio.AsyncClient):
"""
self.models[self.user_id, "rooms"].pop(room_id, None)
self.models["every_room"].pop((self.user_id, room_id), None)
self.models.pop((self.user_id, room_id, "events"), None)
self.models.pop((self.user_id, room_id, "members"), None)
@@ -1248,8 +1247,7 @@ class MatrixClient(nio.AsyncClient):
unreads = unreads,
)
self.models[self.user_id, "rooms"][room.room_id] = room_item
self.models["every_room"][self.user_id, room.room_id] = room_item
self.models[self.user_id, "rooms"][room.room_id] = room_item
# List members that left the room, then remove them from our model
left_the_room = [

22
src/backend/models/filters.py
View File

@@ -1,6 +1,6 @@
# SPDX-License-Identifier: LGPL-3.0-or-later
from typing import TYPE_CHECKING, Collection, Dict, Optional, Tuple
from typing import TYPE_CHECKING, Any, Collection, Dict, Optional, Tuple
from . import SyncId
from .model import Model
@@ -20,10 +20,20 @@ class ModelFilter(ModelProxy):
return True
def source_item_set(self, source: Model, key, value: "ModelItem") -> None:
def source_item_set(
self,
source: Model,
key,
value: "ModelItem",
_changed_fields: Optional[Dict[str, Any]] = None,
) -> None:
if self.accept_source(source):
dct = self if self.accept_item(value) else self.filtered_out
dct[source.sync_id, key] = value
if self.accept_item(value):
self.__setitem__((source.sync_id, key), value, _changed_fields)
self.filtered_out.pop((source.sync_id, key), None)
else:
self.filtered_out[source.sync_id, key] = value
self.pop((source.sync_id, key), None)
def source_item_deleted(self, source: Model, key) -> None:
@@ -64,11 +74,11 @@ class ModelFilter(ModelProxy):
class FieldSubstringFilter(ModelFilter):
def __init__(self, fields: Collection[str], *args, **kwargs) -> None:
def __init__(self, sync_id: SyncId, fields: Collection[str]) -> None:
self.fields: Collection[str] = fields
self._filter: str = ""
super().__init__(*args, **kwargs)
super().__init__(sync_id)
@property

40
src/backend/models/model.py
View File

@@ -40,6 +40,11 @@ class Model(MutableMapping):
self._sorted_data: List["ModelItem"] = blist()
self._write_lock: RLock = RLock()
self.take_items_ownership: bool = True
if self.sync_id:
self.instances[self.sync_id] = self
def __repr__(self) -> str:
"""Provide a full representation of the model and its content."""
@@ -63,27 +68,34 @@ class Model(MutableMapping):
return self._data[key]
def __setitem__(self, key, value: "ModelItem") -> None:
def __setitem__(
self,
key,
value: "ModelItem",
_changed_fields: Optional[Dict[str, Any]] = None,
) -> None:
with self._write_lock:
existing = self._data.get(key)
new = value
# Collect changed fields
changed_fields = {}
changed_fields = _changed_fields or {}
for field in new.__dataclass_fields__: # type: ignore
changed = True
if not changed_fields:
for field in new.__dataclass_fields__: # type: ignore
changed = True
if existing:
changed = getattr(new, field) != getattr(existing, field)
if existing:
changed = \
getattr(new, field) != getattr(existing, field)
if changed:
changed_fields[field] = new.serialize_field(field)
if changed:
changed_fields[field] = new.serialize_field(field)
# Set parent model on new item
if self.sync_id:
if self.sync_id and self.take_items_ownership:
new.parent_model = self
# Insert into sorted data
@@ -101,6 +113,12 @@ class Model(MutableMapping):
self._data[key] = new
# Callbacks
for sync_id, proxy in self.proxies.items():
if sync_id != self.sync_id:
proxy.source_item_set(self, key, value)
# Emit PyOtherSide event
if self.sync_id and (index_then != index_now or changed_fields):
@@ -121,6 +139,10 @@ class Model(MutableMapping):
index = self._sorted_data.index(item)
del self._sorted_data[index]
for sync_id, proxy in self.proxies.items():
if sync_id != self.sync_id:
proxy.source_item_deleted(self, key)
if self.sync_id:
ModelItemDeleted(self.sync_id, index)

20
src/backend/models/model_item.py
View File

@@ -42,19 +42,23 @@ class ModelItem:
super().__setattr__(name, value)
model = self.parent_model
parent = self.parent_model
if not model.sync_id:
if not parent.sync_id:
return
with model._write_lock:
index_then = model._sorted_data.index(self)
model._sorted_data.sort()
index_now = model._sorted_data.index(self)
fields = {name: self.serialize_field(name)}
fields = {name: self.serialize_field(name)}
with parent._write_lock:
index_then = parent._sorted_data.index(self)
parent._sorted_data.sort()
index_now = parent._sorted_data.index(self)
ModelItemSet(model.sync_id, index_then, index_now, fields)
ModelItemSet(parent.sync_id, index_then, index_now, fields)
for sync_id, proxy in parent.proxies.items():
if sync_id != parent.sync_id:
proxy.source_item_set(parent, self.id, self, fields)
def __delattr__(self, name: str) -> None:

16
src/backend/models/proxy.py
View File

@@ -1,6 +1,6 @@
# SPDX-License-Identifier: LGPL-3.0-or-later
from typing import TYPE_CHECKING
from typing import TYPE_CHECKING, Any, Dict, Optional
from . import SyncId
from .model import Model
@@ -12,14 +12,12 @@ if TYPE_CHECKING:
class ModelProxy(Model):
def __init__(self, sync_id: SyncId) -> None:
super().__init__(sync_id)
self.take_items_ownership = False
Model.proxies[sync_id] = self
for sync_id, model in Model.instances.items():
if sync_id != self.sync_id and self.accept_source(model):
for key, item in model.items():
# if isinstance(model, ModelProxy):
# key = key[1]
self.source_item_set(model, key, item)
@@ -27,9 +25,15 @@ class ModelProxy(Model):
return True
def source_item_set(self, source: Model, key, value: "ModelItem") -> None:
def source_item_set(
self,
source: Model,
key,
value: "ModelItem",
_changed_fields: Optional[Dict[str, Any]] = None,
) -> None:
if self.accept_source(source):
self[source.sync_id, key] = value
self.__setitem__((source.sync_id, key), value, _changed_fields)
def source_item_deleted(self, source: Model, key) -> None: