26 lines
992 B
Python
26 lines
992 B
Python
import heroku3
|
|
|
|
from functools import wraps
|
|
from bot import HEROKU_API_KEY, HEROKU_APP_NAME
|
|
|
|
|
|
heroku_client = heroku3.from_key(HEROKU_API_KEY) if HEROKU_API_KEY else None
|
|
|
|
def check_heroku(func):
|
|
@wraps(func)
|
|
async def heroku_cli(client, message):
|
|
heroku_app = None
|
|
if not heroku_client:
|
|
await message.reply_text("`Please Add HEROKU_API_KEY Key For This To Function To Work!`", parse_mode="markdown")
|
|
elif not HEROKU_APP_NAME:
|
|
await message.reply_text("`Please Add HEROKU_APP_NAME For This To Function To Work!`", parse_mode="markdown")
|
|
if HEROKU_APP_NAME and heroku_client:
|
|
try:
|
|
heroku_app = heroku_client.app(HEROKU_APP_NAME)
|
|
except:
|
|
await message.reply_text(message, "`Heroku Api Key And App Name Doesn't Match!`", parse_mode="markdown")
|
|
if heroku_app:
|
|
await func(client, message, heroku_app)
|
|
|
|
return heroku_cli
|