78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
import asyncio
|
||
|
|
import time
|
||
|
|
import os
|
||
|
|
|
||
|
|
import requests
|
||
|
|
import discord
|
||
|
|
from discord.ext import commands
|
||
|
|
|
||
|
|
import config
|
||
|
|
|
||
|
|
|
||
|
|
def login_required(fn):
|
||
|
|
async def decorated_fn(ctx, *args):
|
||
|
|
if str(ctx.author.id) not in config.allowed_users:
|
||
|
|
await ctx.reply("Vous n'êtes pas autorisé à exécuter cette fonction.")
|
||
|
|
return None
|
||
|
|
return await fn(ctx, *args)
|
||
|
|
|
||
|
|
return decorated_fn
|
||
|
|
|
||
|
|
def check_status(service):
|
||
|
|
"""Renvoie le statut d'un service"""
|
||
|
|
service_status = os.system(f"systemctl is-active --quiet {service}")
|
||
|
|
if service_status == 0:
|
||
|
|
return "✅ Running"
|
||
|
|
return "❌ Stopped"
|
||
|
|
|
||
|
|
bot = commands.Bot(command_prefix=config.command_prefix)
|
||
|
|
|
||
|
|
|
||
|
|
@bot.event
|
||
|
|
async def on_ready():
|
||
|
|
"""On ready"""
|
||
|
|
print("Connecté en tant que:")
|
||
|
|
print(bot.user.name)
|
||
|
|
print(bot.user.id)
|
||
|
|
print("------")
|
||
|
|
|
||
|
|
game = discord.Game(f"Disponible ✅ | {bot.command_prefix}help")
|
||
|
|
await bot.change_presence(status=discord.Status.idle, activity=game)
|
||
|
|
|
||
|
|
services_status = {service:check_status(service) for service in config.services}
|
||
|
|
while True:
|
||
|
|
new_services_status = {service:check_status(service) for service in config.services}
|
||
|
|
for service in new_services_status.keys():
|
||
|
|
if new_services_status[service] != services_status[service]:
|
||
|
|
|
||
|
|
embed = discord.Embed(colour=discord.Colour.blue())
|
||
|
|
embed.set_author(name=time.strftime("%m-%d-%Y %H:%M"))
|
||
|
|
embed.add_field(
|
||
|
|
name=service, value=new_services_status[service], inline=False
|
||
|
|
)
|
||
|
|
for discord_id in config.allowed_users:
|
||
|
|
user = await bot.fetch_user(discord_id)
|
||
|
|
await user.send(embed=embed)
|
||
|
|
|
||
|
|
services_status = new_services_status
|
||
|
|
await asyncio.sleep(300)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@bot.command(name="status")
|
||
|
|
@login_required
|
||
|
|
async def status(ctx):
|
||
|
|
"""Renvoie le statut des différents services"""
|
||
|
|
embed = discord.Embed(colour=discord.Colour.blue())
|
||
|
|
embed.set_author(name=time.strftime("%m-%d-%Y %H:%M"))
|
||
|
|
for service in config.services:
|
||
|
|
service_status = check_status(service)
|
||
|
|
embed.add_field(
|
||
|
|
name=service, value=service_status, inline=False
|
||
|
|
)
|
||
|
|
await ctx.reply(embed=embed)
|
||
|
|
|
||
|
|
|
||
|
|
bot.run(config.TOKEN)
|