Merge pull request #3 from nm17/afk_func

Afk func

Author: nm17

nicevk/plugins/afk_handler.py

@@ -0,0 +1,27 @@
+import time
+from datetime import timedelta
+
+from vkbottle import Message
+from nicevk.api import state, save_state, user
+
+
+@user.middleware.middleware_handler()
+async def answer(ans: Message):
+    print("Asd")
+    domain = str((await user.api.users.get(fields="domain"))[0].domain)
+    if (
+        (len(domain) > 1 and domain in ans.text)
+        or str(user.user_id) in ans.text
+    ):
+        if not state["afk"] or not state["afk"]["status"] or str(ans.chat_id) in state["afk"]["mentioned"]:
+            return True
+        diff = str(timedelta(seconds=(time.time() - state["afk"]["time"])))
+        if state["afk"]["reason"]:
+            await ans("I am afk (for {}): {}".format(diff, state["afk"]["reason"]))
+        else:
+            await ans("I am afk (for {}), contact me later".format(diff))
+        state["afk"]["mentioned"].append(str(ans.chat_id))
+        save_state()
+        return True
+    print(ans.text, str((await user.api.users.get(fields="domain"))[0].domain))
+    return True

nicevk/plugins/afk_plugin.py

@@ -0,0 +1,32 @@
+from vkbottle import Message
+from nicevk.api import user, commands, state, save_state
+import time
+
+commands.extend([".afk [reason] - makes you afk", ".unafk - seems like you are here"])
+
+
+@user.on.message_handler(text=".afk <reason>")
+async def afk(ans: Message, reason: str):
+    if "afk" not in state.keys():
+        state["afk"] = {}
+    state["afk"]["status"] = True
+    if reason is not None:
+        state["afk"]["reason"] = reason
+    state["afk"]["time"] = time.time()
+    state["afk"]["mentioned"] = []
+    save_state()
+    await ans.api.messages.edit(ans.peer_id, ans.id, "I will be back")
+
+
+@user.on.message_handler(text=".unafk")
+async def afk(ans: Message):
+    if "afk" not in state.keys():  # Using unafk without afk
+        state["afk"] = {}
+    state["afk"]["status"] = False
+    try:
+        del state["afk"]["time"]
+        del state["afk"]["mentioned"]
+    except KeyError:
+        pass
+    save_state()
+    await ans.api.messages.edit(ans.peer_id, ans.id, "I am back")

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nicevk"
-version = "0.5.1"
+version = "0.6.0"
 description = ""
 authors = ["nm17 <[email protected]>"]