From 6f39b11ff22d819cfe61746630da5ada281250f8 Mon Sep 17 00:00:00 2001 From: levina <82658782+levina-lab@users.noreply.github.com> Date: Thu, 10 Feb 2022 12:10:08 +0700 Subject: [PATCH] function --- driver/utils.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/driver/utils.py b/driver/utils.py index e3d8038..6480752 100644 --- a/driver/utils.py +++ b/driver/utils.py @@ -1,4 +1,6 @@ import asyncio +import socket +from config import HEROKU_API_KEY from driver.core import bot, calls from driver.database.dbqueue import remove_active_chat from driver.queues import QUEUE, clear_queue, get_queue, pop_an_item @@ -144,3 +146,13 @@ async def bash(cmd): err = stderr.decode().strip() out = stdout.decode().strip() return out, err + + +async def is_heroku(): + return "heroku" in socket.getfqdn() + + +async def user_input(input): + if " " in input or "\n" in input: + return str(input.split(maxsplit=1)[1].strip()) + return ""