paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-14 10:57:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_service_bot_system: Fix type: ignore issues.

Browse Source 
Signed-off-by: Anders Kaseorg <anders@zulip.com>
This commit is contained in:
Anders Kaseorg
2020-06-30 16:50:08 -07:00
parent 01e24ccd84
commit ec70aab138
3 changed files with 13 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
zerver/lib/bot_lib.py
View File

@@ -1,7 +1,7 @@
import importlib
import json
import os
from typing import Any, Dict
from typing import Any, Callable, Dict
from django.utils.translation import ugettext as _
@@ -45,13 +45,13 @@ class StateHandler:
def __init__(self, user_profile: UserProfile) -> None:
self.user_profile = user_profile
self.marshal = lambda obj: json.dumps(obj)
self.demarshal = lambda obj: json.loads(obj)
self.marshal: Callable[[object], str] = lambda obj: json.dumps(obj)
self.demarshal: Callable[[str], object] = lambda obj: json.loads(obj)
def get(self, key: str) -> str:
def get(self, key: str) -> object:
return self.demarshal(get_bot_storage(self.user_profile, key))
def put(self, key: str, value: str) -> None:
def put(self, key: str, value: object) -> None:
set_bot_storage(self.user_profile, [(key, self.marshal(value))])
def remove(self, key: str) -> None: