Support a order
key for accounts in config
This commit is contained in:
parent
fd3fe06d15
commit
9b5b3a9f06
|
@ -106,6 +106,7 @@ class Backend:
|
||||||
password: str,
|
password: str,
|
||||||
device_id: Optional[str] = None,
|
device_id: Optional[str] = None,
|
||||||
homeserver: str = "https://matrix.org",
|
homeserver: str = "https://matrix.org",
|
||||||
|
order: int = -1,
|
||||||
) -> str:
|
) -> str:
|
||||||
"""Create and register a `MatrixClient`, login and return a user ID."""
|
"""Create and register a `MatrixClient`, login and return a user ID."""
|
||||||
|
|
||||||
|
@ -120,15 +121,18 @@ class Backend:
|
||||||
raise
|
raise
|
||||||
|
|
||||||
self.clients[client.user_id] = client
|
self.clients[client.user_id] = client
|
||||||
self.models["accounts"][client.user_id] = Account(client.user_id)
|
self.models["accounts"][client.user_id] = Account(client.user_id,order)
|
||||||
return client.user_id
|
return client.user_id
|
||||||
|
|
||||||
|
|
||||||
async def resume_client(self,
|
async def resume_client(
|
||||||
user_id: str,
|
self,
|
||||||
token: str,
|
user_id: str,
|
||||||
device_id: str,
|
token: str,
|
||||||
homeserver: str = "https://matrix.org") -> None:
|
device_id: str,
|
||||||
|
homeserver: str = "https://matrix.org",
|
||||||
|
order: int = -1,
|
||||||
|
) -> None:
|
||||||
"""Create and register a `MatrixClient` with known account details."""
|
"""Create and register a `MatrixClient` with known account details."""
|
||||||
|
|
||||||
client = MatrixClient(
|
client = MatrixClient(
|
||||||
|
@ -137,7 +141,7 @@ class Backend:
|
||||||
)
|
)
|
||||||
|
|
||||||
self.clients[user_id] = client
|
self.clients[user_id] = client
|
||||||
self.models["accounts"][user_id] = Account(user_id)
|
self.models["accounts"][user_id] = Account(user_id, order)
|
||||||
|
|
||||||
await client.resume(user_id=user_id, token=token, device_id=device_id)
|
await client.resume(user_id=user_id, token=token, device_id=device_id)
|
||||||
|
|
||||||
|
@ -145,12 +149,13 @@ class Backend:
|
||||||
async def load_saved_accounts(self) -> List[str]:
|
async def load_saved_accounts(self) -> List[str]:
|
||||||
"""Call `resume_client` for all saved accounts in user config."""
|
"""Call `resume_client` for all saved accounts in user config."""
|
||||||
|
|
||||||
async def resume(user_id: str, info: Dict[str, str]) -> str:
|
async def resume(user_id: str, info: Dict[str, Any]) -> str:
|
||||||
await self.resume_client(
|
await self.resume_client(
|
||||||
user_id = user_id,
|
user_id = user_id,
|
||||||
token = info["token"],
|
token = info["token"],
|
||||||
device_id = info["device_id"],
|
device_id = info["device_id"],
|
||||||
homeserver = info["homeserver"],
|
homeserver = info["homeserver"],
|
||||||
|
order = info.get("order", -1),
|
||||||
)
|
)
|
||||||
return user_id
|
return user_id
|
||||||
|
|
||||||
|
|
|
@ -33,6 +33,7 @@ class Account(ModelItem):
|
||||||
"""A logged in matrix account."""
|
"""A logged in matrix account."""
|
||||||
|
|
||||||
id: str = field()
|
id: str = field()
|
||||||
|
order: int = -1
|
||||||
display_name: str = ""
|
display_name: str = ""
|
||||||
avatar_url: str = ""
|
avatar_url: str = ""
|
||||||
max_upload_size: int = 0
|
max_upload_size: int = 0
|
||||||
|
@ -42,10 +43,8 @@ class Account(ModelItem):
|
||||||
total_mentions: int = 0
|
total_mentions: int = 0
|
||||||
|
|
||||||
def __lt__(self, other: "Account") -> bool:
|
def __lt__(self, other: "Account") -> bool:
|
||||||
"""Sort by user ID."""
|
"""Sort by order, then by user ID."""
|
||||||
name = self.id[1:]
|
return (self.order, self.id.lower()) < (other.order, other.id.lower())
|
||||||
other_name = other.id[1:]
|
|
||||||
return name.lower() < other_name.lower()
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
|
@ -120,10 +119,12 @@ class Room(ModelItem):
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class AccountOrRoom(Account, Room):
|
class AccountOrRoom(Account, Room):
|
||||||
type: Union[Type[Account], Type[Room]] = Account
|
type: Union[Type[Account], Type[Room]] = Account
|
||||||
|
account_order: int = -1
|
||||||
|
|
||||||
def __lt__(self, other: "AccountOrRoom") -> bool: # type: ignore
|
def __lt__(self, other: "AccountOrRoom") -> bool: # type: ignore
|
||||||
return (
|
return (
|
||||||
|
self.account_order,
|
||||||
self.id if self.type is Account else self.for_account,
|
self.id if self.type is Account else self.for_account,
|
||||||
other.type is Account,
|
other.type is Account,
|
||||||
self.left,
|
self.left,
|
||||||
|
@ -134,6 +135,7 @@ class AccountOrRoom(Account, Room):
|
||||||
(self.display_name or self.id).lower(),
|
(self.display_name or self.id).lower(),
|
||||||
|
|
||||||
) < (
|
) < (
|
||||||
|
other.account_order,
|
||||||
other.id if other.type is Account else other.for_account,
|
other.id if other.type is Account else other.for_account,
|
||||||
self.type is Account,
|
self.type is Account,
|
||||||
other.left,
|
other.left,
|
||||||
|
|
|
@ -29,7 +29,7 @@ class ModelStore(UserDict):
|
||||||
model: Model
|
model: Model
|
||||||
|
|
||||||
if key == "all_rooms":
|
if key == "all_rooms":
|
||||||
model = AllRooms()
|
model = AllRooms(self["accounts"])
|
||||||
elif key == "matching_accounts":
|
elif key == "matching_accounts":
|
||||||
model = MatchingAccounts(self["all_rooms"])
|
model = MatchingAccounts(self["all_rooms"])
|
||||||
elif is_tuple and len(key) == 3 and key[2] == "filtered_members":
|
elif is_tuple and len(key) == 3 and key[2] == "filtered_members":
|
||||||
|
|
|
@ -10,10 +10,12 @@ from .model_item import ModelItem
|
||||||
|
|
||||||
|
|
||||||
class AllRooms(FieldSubstringFilter):
|
class AllRooms(FieldSubstringFilter):
|
||||||
def __init__(self) -> None:
|
def __init__(self, accounts: Model) -> None:
|
||||||
super().__init__(sync_id="all_rooms", fields=("display_name",))
|
super().__init__(sync_id="all_rooms", fields=("display_name",))
|
||||||
self.items_changed_callbacks.append(self.refilter_accounts)
|
self.items_changed_callbacks.append(self.refilter_accounts)
|
||||||
|
|
||||||
|
self.accounts = accounts
|
||||||
|
|
||||||
self._collapsed: Set[str] = set()
|
self._collapsed: Set[str] = set()
|
||||||
|
|
||||||
|
|
||||||
|
@ -39,7 +41,14 @@ class AllRooms(FieldSubstringFilter):
|
||||||
|
|
||||||
|
|
||||||
def convert_item(self, item: ModelItem) -> AccountOrRoom:
|
def convert_item(self, item: ModelItem) -> AccountOrRoom:
|
||||||
return AccountOrRoom(**asdict(item), type=type(item)) # type: ignore
|
return AccountOrRoom(
|
||||||
|
**asdict(item),
|
||||||
|
type = type(item), # type: ignore
|
||||||
|
|
||||||
|
account_order =
|
||||||
|
item.order if isinstance(item, Account) else
|
||||||
|
self.accounts[item.for_account].order, # type: ignore
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def accept_item(self, item: ModelItem) -> bool:
|
def accept_item(self, item: ModelItem) -> bool:
|
||||||
|
|
|
@ -183,14 +183,19 @@ class Accounts(JSONDataFile):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
client = self.backend.clients[user_id]
|
client = self.backend.clients[user_id]
|
||||||
|
saved = await self.read()
|
||||||
|
|
||||||
await self.write({
|
await self.write({
|
||||||
**await self.read(),
|
**saved,
|
||||||
client.user_id: {
|
client.user_id: {
|
||||||
"homeserver": client.homeserver,
|
"homeserver": client.homeserver,
|
||||||
"token": client.access_token,
|
"token": client.access_token,
|
||||||
"device_id": client.device_id,
|
"device_id": client.device_id,
|
||||||
"enabled": True,
|
"enabled": True,
|
||||||
|
"order": max(
|
||||||
|
account.get("order", i)
|
||||||
|
for i, account in enumerate(saved.values())
|
||||||
|
) + 1,
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user