|
|
|
@@ -1,7 +1,9 @@
|
|
|
|
|
import aiohttp
|
|
|
|
|
import fluxer
|
|
|
|
|
import requests
|
|
|
|
|
import json
|
|
|
|
|
import asyncio
|
|
|
|
|
import base64
|
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
|
import sqlite3
|
|
|
|
|
import os
|
|
|
|
@@ -118,7 +120,7 @@ async def setChannel(message):
|
|
|
|
|
|
|
|
|
|
@bot.command()
|
|
|
|
|
async def version(message):
|
|
|
|
|
await message.channel.send("Version 0.4.1 is running\nSource code: https://github.com/michelleDeko/ich_iel-bot")
|
|
|
|
|
await message.channel.send("Version 0.5.1 is running\nSource code: https://github.com/michelleDeko/ich_iel-bot")
|
|
|
|
|
|
|
|
|
|
# the cat bot died, so i wanted to add this command to this bot
|
|
|
|
|
@bot.command()
|
|
|
|
@@ -133,6 +135,46 @@ async def cat(message):
|
|
|
|
|
else:
|
|
|
|
|
await message.channel.send("Failed to fetch cat image")
|
|
|
|
|
|
|
|
|
|
# i thought dogs and foxes would be nice to have too
|
|
|
|
|
@bot.command()
|
|
|
|
|
async def dog(message):
|
|
|
|
|
response = requests.get("https://dog.ceo/api/breeds/image/random")
|
|
|
|
|
if response.status_code == 200:
|
|
|
|
|
data = response.json()
|
|
|
|
|
if data and isinstance(data, dict) and "message" in data:
|
|
|
|
|
await message.channel.send(data["message"])
|
|
|
|
|
else:
|
|
|
|
|
await message.channel.send("Could not fetch dog image")
|
|
|
|
|
else:
|
|
|
|
|
await message.channel.send("Failed to fetch dog image")
|
|
|
|
|
|
|
|
|
|
@bot.command()
|
|
|
|
|
async def fox(message):
|
|
|
|
|
response = requests.get("https://randomfox.ca/floof/")
|
|
|
|
|
if response.status_code == 200:
|
|
|
|
|
data = response.json()
|
|
|
|
|
if data and isinstance(data, dict) and "image" in data:
|
|
|
|
|
await message.channel.send(data["image"])
|
|
|
|
|
else:
|
|
|
|
|
await message.channel.send("Could not fetch fox image")
|
|
|
|
|
else:
|
|
|
|
|
await message.channel.send("Failed to fetch fox image")
|
|
|
|
|
|
|
|
|
|
# time for racoons
|
|
|
|
|
@bot.command()
|
|
|
|
|
async def racoon(message):
|
|
|
|
|
urls = random.choice([
|
|
|
|
|
"https://api.mapach.es/v1/meme",
|
|
|
|
|
"https://api.mapach.es/v1/coon"
|
|
|
|
|
])
|
|
|
|
|
async with aiohttp.ClientSession() as session:
|
|
|
|
|
async with session.get(urls) as response:
|
|
|
|
|
image_bytes = await response.read()
|
|
|
|
|
if response.status != 200:
|
|
|
|
|
await message.channel.send("Failed to fetch racoon image")
|
|
|
|
|
return
|
|
|
|
|
await message.channel.send(file=fluxer.File(image_bytes, filename="racoon.jpg"))
|
|
|
|
|
|
|
|
|
|
async def post_reddit():
|
|
|
|
|
subreddit = os.getenv("SUBREDDIT", "ich_iel")
|
|
|
|
|
posts = await get_latest_post(subreddit)
|
|
|
|
@@ -170,6 +212,7 @@ async def post_reddit():
|
|
|
|
|
else:
|
|
|
|
|
logging.warning(f"Bot is not in guild {guild_id}, removing guild from database")
|
|
|
|
|
cur.execute("DELETE FROM channels WHERE guild_id = ?", (guild_id,))
|
|
|
|
|
cur.execute("DELETE FROM posted WHERE guild_id = ?", (guild_id,))
|
|
|
|
|
con.commit()
|
|
|
|
|
break
|
|
|
|
|
else:
|
|
|
|
@@ -201,27 +244,6 @@ async def check_guild(guild_id):
|
|
|
|
|
logging.warning(f"Bot doesn't have access to guild {guild_id}")
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
@bot.event
|
|
|
|
|
async def on_guild_remove(guild):
|
|
|
|
|
if isinstance(guild, fluxer.Guild):
|
|
|
|
|
guild_id = guild.id
|
|
|
|
|
logging.info(f"Removed from guild: {guild.name} (ID: {guild_id}), removing from database")
|
|
|
|
|
else:
|
|
|
|
|
if guild.get("unavailable"):
|
|
|
|
|
logging.info(f"Guild {guild.get('id')} is temporarily unavailable, ignoring")
|
|
|
|
|
return
|
|
|
|
|
guild_id = int(guild["id"])
|
|
|
|
|
logging.info(f"Removed from guild {guild_id}, removing from database")
|
|
|
|
|
try:
|
|
|
|
|
con = sqlite3.connect('data/ich_iel-bot.db')
|
|
|
|
|
cur = con.cursor()
|
|
|
|
|
cur.execute("DELETE FROM channels WHERE guild_id = ?", (guild_id,))
|
|
|
|
|
cur.execute("DELETE FROM posted WHERE guild_id = ?", (guild_id,))
|
|
|
|
|
con.commit()
|
|
|
|
|
logging.info(f"Guild {guild_id} removed from database successfully")
|
|
|
|
|
except sqlite3.Error as e:
|
|
|
|
|
logging.error(f"Database error while removing guild {guild_id}: {e}")
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|
logging.info("Starting bot...")
|
|
|
|
|
asyncio.run(init_db())
|
|
|
|
|