Compare commits
34 Commits
v1.0
...
d3eb82f1bd
Author | SHA1 | Date | |
---|---|---|---|
d3eb82f1bd | |||
353284b2c3 | |||
82561f050f | |||
39b821ff7f | |||
d9ac0cc36e | |||
d707398c26 | |||
0cb5f8fa38 | |||
d70584d4f8 | |||
49831aeb47 | |||
4226ce86c8 | |||
b4ca6b778d | |||
9d5765c492 | |||
8216af5f2a | |||
219b1745da | |||
d570069e6b | |||
f267526411 | |||
86fa47d259 | |||
888e647e1f | |||
c1beb8374e | |||
48a7033ba6 | |||
9e436d8d78 | |||
d8c286bb2d | |||
a081edfa97 | |||
716e30e2b1 | |||
73499a9586 | |||
e3a762fe10 | |||
842b592bfc | |||
f9f88be8af | |||
5c8696216e | |||
aa16064d17 | |||
ffd642fd2a | |||
3945eb6763 | |||
0438b55ce6 | |||
8fb4e089be |
28
.gitea/workflows/deploy.yaml
Normal file
28
.gitea/workflows/deploy.yaml
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
name: Garfbot CI/CD Deployment
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches: [ main ]
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
Deploy:
|
||||||
|
container:
|
||||||
|
volumes:
|
||||||
|
- /home/crate/garfbot:/workspace/crate/garfbot/deploy
|
||||||
|
steps:
|
||||||
|
- name: Pull Garfbot and restart container
|
||||||
|
run: |
|
||||||
|
cd /workspace/crate/garfbot/deploy
|
||||||
|
|
||||||
|
git pull origin main
|
||||||
|
|
||||||
|
CHANGED=$(git diff --name-only HEAD~1 HEAD)
|
||||||
|
|
||||||
|
if echo "$CHANGED" | grep -qE "(Dockerfile|requirements\.txt|docker-compose\.yml)"; then
|
||||||
|
docker stop garfbot
|
||||||
|
docker rm garfbot
|
||||||
|
docker build -t git.crate.zip/crate/garfbot:latest .
|
||||||
|
docker run -d --restart always -v $PWD:/usr/src/app --name garfbot git.crate.zip/crate/garfbot:latest
|
||||||
|
else
|
||||||
|
docker restart garfbot
|
||||||
|
fi
|
3
.gitignore
vendored
3
.gitignore
vendored
@ -6,3 +6,6 @@ garfpy/__pycache__/
|
|||||||
*.old
|
*.old
|
||||||
*.log*
|
*.log*
|
||||||
meows.py
|
meows.py
|
||||||
|
meow_counts.json
|
||||||
|
user_stats.json
|
||||||
|
responses.json
|
@ -12,5 +12,8 @@ RUN pip3 install discord
|
|||||||
RUN pip3 install openai
|
RUN pip3 install openai
|
||||||
RUN pip3 install aiohttp
|
RUN pip3 install aiohttp
|
||||||
RUN pip3 install requests
|
RUN pip3 install requests
|
||||||
|
RUN pip3 install wikipedia
|
||||||
|
RUN pip3 install pillow
|
||||||
|
RUN pip3 install qrcode
|
||||||
|
|
||||||
CMD [ "python", "garfmain.py" ]
|
CMD [ "python", "garfmain.py" ]
|
||||||
|
@ -16,7 +16,7 @@ Responds with iputils-ping result from target.
|
|||||||
`garfpic {target}`
|
`garfpic {target}`
|
||||||
Responds with dns lookup result from target.
|
Responds with dns lookup result from target.
|
||||||
|
|
||||||
`garfpic {target}`
|
`garfhack {target}`
|
||||||
Responds with nmap scan result from target.
|
Responds with nmap scan result from target.
|
||||||
|
|
||||||
`garfshop {item} {zip}`
|
`garfshop {item} {zip}`
|
||||||
@ -36,7 +36,6 @@ Add your various API tokens:
|
|||||||
```python
|
```python
|
||||||
GARFBOT_TOKEN = "Discord API token"
|
GARFBOT_TOKEN = "Discord API token"
|
||||||
OPENAI_TOKEN = "OpenAI API token"
|
OPENAI_TOKEN = "OpenAI API token"
|
||||||
GIF_TOKEN = "tenor.com API token"
|
|
||||||
```
|
```
|
||||||
|
|
||||||
I recommend building a docker image using the included DockerFile as a template. Run the container binding /usr/src/app to GarfBot's CWD:
|
I recommend building a docker image using the included DockerFile as a template. Run the container binding /usr/src/app to GarfBot's CWD:
|
||||||
|
144
garfmain.py
144
garfmain.py
@ -1,14 +1,12 @@
|
|||||||
import config
|
import config
|
||||||
# import random
|
|
||||||
import asyncio
|
import asyncio
|
||||||
import discord
|
import discord
|
||||||
import subprocess
|
import subprocess
|
||||||
|
|
||||||
from garfpy import(
|
from garfpy import(
|
||||||
logger, is_private,
|
logger, is_private,
|
||||||
kroger_token, find_store, search_product,
|
aod_message, generate_qr,
|
||||||
picture_time, process_image_requests, generate_chat,
|
Kroger, GarfAI, GarfbotRespond)
|
||||||
aod_message)
|
|
||||||
|
|
||||||
|
|
||||||
gapikey = config.GIF_TOKEN
|
gapikey = config.GIF_TOKEN
|
||||||
@ -22,11 +20,16 @@ intents.messages = True
|
|||||||
intents.message_content = True
|
intents.message_content = True
|
||||||
garfbot = discord.Client(intents=intents)
|
garfbot = discord.Client(intents=intents)
|
||||||
|
|
||||||
|
garf_respond = GarfbotRespond()
|
||||||
|
garfield = GarfAI()
|
||||||
|
kroger = Kroger()
|
||||||
|
|
||||||
|
|
||||||
@garfbot.event
|
@garfbot.event
|
||||||
async def on_ready():
|
async def on_ready():
|
||||||
try:
|
try:
|
||||||
asyncio.create_task(process_image_requests())
|
garf_respond.load_responses()
|
||||||
|
asyncio.create_task(garfield.process_image_requests())
|
||||||
logger.info(f"Logged in as {garfbot.user.name} running {txtmodel} and {imgmodel}.")
|
logger.info(f"Logged in as {garfbot.user.name} running {txtmodel} and {imgmodel}.")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(e)
|
logger.error(e)
|
||||||
@ -34,41 +37,58 @@ async def on_ready():
|
|||||||
|
|
||||||
@garfbot.event
|
@garfbot.event
|
||||||
async def on_message(message):
|
async def on_message(message):
|
||||||
|
|
||||||
|
content = message.content.strip()
|
||||||
|
lower = content.lower()
|
||||||
|
user = message.author.name
|
||||||
|
guild = message.guild.name if message.guild else "Direct Message"
|
||||||
|
guild_id = message.guild.id
|
||||||
|
|
||||||
|
# Chats & pics
|
||||||
if message.author == garfbot.user:
|
if message.author == garfbot.user:
|
||||||
return
|
return
|
||||||
|
|
||||||
if message.content.lower().startswith("hey garfield") or isinstance(message.channel, discord.DMChannel):
|
if lower.startswith("hey garfield") or isinstance(message.channel, discord.DMChannel):
|
||||||
question = message.content[12:] if message.content.lower().startswith("hey garfield") else message.content
|
prompt = content[12:] if lower.startswith("hey garfield") else message.content
|
||||||
answer = await generate_chat(question)
|
answer = await garfield.generate_chat(prompt)
|
||||||
|
logger.info(f"Chat Request - User: {user}, Server: {guild}, Prompt: {prompt}")
|
||||||
await message.channel.send(answer)
|
await message.channel.send(answer)
|
||||||
|
|
||||||
if message.content.lower().startswith('garfpic '):
|
if lower.startswith('garfpic '):
|
||||||
user = message.author.name
|
prompt = content[8:]
|
||||||
server = message.guild.name if message.guild else "Direct Message"
|
logger.info(f"Image Request - User: {user}, Server: {guild}, Prompt: {prompt}")
|
||||||
prompt = message.content[8:]
|
|
||||||
logger.info(f"Image Request - User: {user}, Server: {server}, Prompt: {prompt}")
|
|
||||||
await message.channel.send(f"`Please wait... image generation queued: {prompt}`")
|
await message.channel.send(f"`Please wait... image generation queued: {prompt}`")
|
||||||
await picture_time(message, prompt)
|
await garfield.garfpic(message, prompt)
|
||||||
|
|
||||||
# if message.content.lower() == "lasagna":
|
# Wikipedia
|
||||||
# await send_gif(message, "garfield lasagna")
|
if lower.startswith('garfwiki '):
|
||||||
|
query = message.content[9:]
|
||||||
|
summary = await garfield.wikisum(query)
|
||||||
|
await message.channel.send(summary)
|
||||||
|
|
||||||
# if message.content.lower() == "monday":
|
# QR codes
|
||||||
# await send_gif(message, "garfield monday")
|
if lower.startswith('garfqr '):
|
||||||
|
text = message.content[7:]
|
||||||
|
if len(text) > 1000:
|
||||||
|
await message.channel.send("❌ Text too long! Maximum 1000 characters.")
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
qr_code = await generate_qr(text)
|
||||||
|
sendfile = discord.File(fp=qr_code, filename="qrcode.png")
|
||||||
|
await message.channel.send(file=sendfile)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(e)
|
||||||
|
await message.channel.send(e)
|
||||||
|
|
||||||
# if message.content.lower().startswith("garfgif "):
|
# IP utils
|
||||||
# search_term = message.content[8:]
|
query = message.content.split()
|
||||||
# await send_gif(message, search_term)
|
target = query[-1]
|
||||||
|
|
||||||
if message.content.lower().startswith("garfping "):
|
if lower.startswith("garfping "):
|
||||||
try:
|
try:
|
||||||
query = message.content.split()
|
logger.info(f"Ping Request - User: {user}, Server: {guild}, Target: {target}")
|
||||||
user = message.author.name
|
|
||||||
server = message.guild.name if message.guild else "Direct Message"
|
|
||||||
target = query[-1]
|
|
||||||
logger.info(f"Ping Request - User: {user}, Server: {server}, Target: {target}")
|
|
||||||
if is_private(target):
|
if is_private(target):
|
||||||
rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
|
rejection = await garfield.generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
|
||||||
await message.channel.send(rejection)
|
await message.channel.send(rejection)
|
||||||
else:
|
else:
|
||||||
result = subprocess.run(['ping', '-c', '4', target], capture_output=True, text=True)
|
result = subprocess.run(['ping', '-c', '4', target], capture_output=True, text=True)
|
||||||
@ -76,15 +96,11 @@ async def on_message(message):
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
await message.channel.send(f"`GarfBot Error: {str(e)}`")
|
await message.channel.send(f"`GarfBot Error: {str(e)}`")
|
||||||
|
|
||||||
if message.content.lower().startswith("garfdns "):
|
if lower.startswith("garfdns "):
|
||||||
try:
|
try:
|
||||||
query = message.content.split()
|
logger.info(f"NSLookup Request - User: {user}, Server: {guild}, Target: {target}")
|
||||||
user = message.author.name
|
|
||||||
server = message.guild.name if message.guild else "Direct Message"
|
|
||||||
target = query[-1]
|
|
||||||
logger.info(f"NSLookup Request - User: {user}, Server: {server}, Target: {target}")
|
|
||||||
if is_private(target):
|
if is_private(target):
|
||||||
rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
|
rejection = await garfield.generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
|
||||||
await message.channel.send(rejection)
|
await message.channel.send(rejection)
|
||||||
else:
|
else:
|
||||||
result = subprocess.run(['nslookup', target], capture_output=True, text=True)
|
result = subprocess.run(['nslookup', target], capture_output=True, text=True)
|
||||||
@ -92,15 +108,11 @@ async def on_message(message):
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
await message.channel.send(f"`GarfBot Error: {str(e)}`")
|
await message.channel.send(f"`GarfBot Error: {str(e)}`")
|
||||||
|
|
||||||
if message.content.lower().startswith("garfhack "):
|
if lower.startswith("garfhack "):
|
||||||
try:
|
try:
|
||||||
query = message.content.split()
|
logger.info(f"Nmap Request - User: {user}, Server: {guild}, Target: {target}")
|
||||||
user = message.author.name
|
|
||||||
server = message.guild.name if message.guild else "Direct Message"
|
|
||||||
target = query[-1]
|
|
||||||
logger.info(f"Nmap Request - User: {user}, Server: {server}, Target: {target}")
|
|
||||||
if is_private(target):
|
if is_private(target):
|
||||||
rejection = await generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
|
rejection = await garfield.generate_chat("Hey Garfield, explain to me why I am dumb for trying to hack your private computer network.")
|
||||||
await message.channel.send(rejection)
|
await message.channel.send(rejection)
|
||||||
else:
|
else:
|
||||||
await message.channel.send(f"`Scanning {target}...`")
|
await message.channel.send(f"`Scanning {target}...`")
|
||||||
@ -110,52 +122,32 @@ async def on_message(message):
|
|||||||
await message.channel.send(f"`GarfBot Error: {str(e)}`")
|
await message.channel.send(f"`GarfBot Error: {str(e)}`")
|
||||||
|
|
||||||
# Kroger Shopping
|
# Kroger Shopping
|
||||||
if message.content.lower().startswith("garfshop "):
|
if lower.startswith("garfshop "):
|
||||||
try:
|
try:
|
||||||
kroken = kroger_token()
|
query = message.content[9:]
|
||||||
kroger_query = message.content.split()
|
response = kroger.garfshop(query)
|
||||||
product = " ".join(kroger_query[1:-1])
|
|
||||||
zipcode = kroger_query[-1]
|
|
||||||
loc_data = find_store(zipcode, kroken)
|
|
||||||
loc_id = loc_data['data'][0]['locationId']
|
|
||||||
store_name = loc_data['data'][0]['name']
|
|
||||||
product_query = search_product(product, loc_id, kroken)
|
|
||||||
products = product_query['data']
|
|
||||||
sorted_products = sorted(products, key=lambda item: item['items'][0]['price']['regular'])
|
|
||||||
response = f"Prices for `{product}` at `{store_name}` near `{zipcode}`:\n"
|
|
||||||
for item in sorted_products:
|
|
||||||
product_name = item['description']
|
|
||||||
price = item['items'][0]['price']['regular']
|
|
||||||
response += f"- `${price}`: {product_name} \n"
|
|
||||||
await message.channel.send(response)
|
await message.channel.send(response)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
await message.channel.send(f"`GarfBot Error: {str(e)}`")
|
await message.channel.send(f"`GarfBot Error: {str(e)}`")
|
||||||
|
|
||||||
# Army of Dawn Server only!!
|
# Army of Dawn Server only!!
|
||||||
if message.guild and message.guild.id == 719605634772893757:
|
if message.guild and message.guild.id == 719605634772893757:
|
||||||
|
|
||||||
await aod_message(garfbot, message)
|
await aod_message(garfbot, message)
|
||||||
|
|
||||||
|
# Auto-responses
|
||||||
|
if message.guild:
|
||||||
|
responses = garf_respond.get_responses(guild_id)
|
||||||
|
|
||||||
# # GarfGifs
|
if lower.startswith('garfbot response '):
|
||||||
# @garfbot.event
|
await garf_respond.garfbot_response(message, content)
|
||||||
# async def send_gif(message, search_term):
|
return
|
||||||
# lmt = 50
|
|
||||||
# ckey = "garfbot"
|
|
||||||
# r = requests.get(f"https://tenor.googleapis.com/v2/search?q={search_term}&key={gapikey}&client_key={ckey}&limit={lmt}")
|
|
||||||
# if r.status_code == 200:
|
|
||||||
# top_50gifs = json.loads(r.content)
|
|
||||||
# gif_url = random.choice(top_50gifs["results"])["itemurl"]
|
|
||||||
# logger.info(gif_url)
|
|
||||||
# # logger.info(gif_url)
|
|
||||||
# try:
|
|
||||||
# await message.channel.send(gif_url)
|
|
||||||
# except KeyError:
|
|
||||||
# await message.channel.send("Oops, something went wrong.")
|
|
||||||
# else:
|
|
||||||
# await message.channel.send(f"`Oops, something went wrong. Error code: {r.status_code}`")
|
|
||||||
|
|
||||||
|
for trigger, response in responses.items():
|
||||||
|
if trigger.lower() in lower:
|
||||||
|
await message.channel.send(response)
|
||||||
|
break
|
||||||
|
|
||||||
|
# Run Garfbot
|
||||||
async def garfbot_connect():
|
async def garfbot_connect():
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
|
@ -1,12 +1,10 @@
|
|||||||
# garfpy/__init__.py
|
# garfpy/__init__.py
|
||||||
|
|
||||||
from .log import logger
|
from .log import logger
|
||||||
from .kroger import(
|
from .kroger import Kroger
|
||||||
kroger_token, find_store, search_product
|
|
||||||
)
|
|
||||||
from .garfai import(
|
|
||||||
picture_time,
|
|
||||||
process_image_requests,
|
|
||||||
generate_chat
|
|
||||||
)
|
|
||||||
from .iputils import is_private
|
from .iputils import is_private
|
||||||
from .aod import aod_message
|
from .aod import aod_message
|
||||||
|
from .qr import generate_qr
|
||||||
|
from .kroger import Kroger
|
||||||
|
from .garfai import GarfAI
|
||||||
|
from .respond import GarfbotRespond
|
158
garfpy/garfai.py
158
garfpy/garfai.py
@ -4,87 +4,95 @@ import config
|
|||||||
import aiohttp
|
import aiohttp
|
||||||
import asyncio
|
import asyncio
|
||||||
import discord
|
import discord
|
||||||
|
import wikipedia
|
||||||
from openai import AsyncOpenAI
|
from openai import AsyncOpenAI
|
||||||
from garfpy import logger
|
from garfpy import logger
|
||||||
|
|
||||||
openaikey = config.OPENAI_TOKEN
|
|
||||||
|
|
||||||
txtmodel = config.TXT_MODEL
|
class GarfAI:
|
||||||
imgmodel = config.IMG_MODEL
|
def __init__(self):
|
||||||
|
self.openaikey = config.OPENAI_TOKEN
|
||||||
|
self.txtmodel = config.TXT_MODEL
|
||||||
|
self.imgmodel = config.IMG_MODEL
|
||||||
|
self.image_request_queue = asyncio.Queue()
|
||||||
|
|
||||||
# GarfPics
|
async def garfpic(self, message, prompt):
|
||||||
image_request_queue = asyncio.Queue()
|
await self.image_request_queue.put({'message': message, 'prompt': prompt})
|
||||||
|
|
||||||
async def picture_time(message, prompt):
|
async def generate_image(self, prompt):
|
||||||
await image_request_queue.put({'message': message, 'prompt': prompt})
|
try:
|
||||||
|
client = AsyncOpenAI(api_key = self.openaikey)
|
||||||
|
response = await client.images.generate(
|
||||||
|
model=self.imgmodel,
|
||||||
|
prompt=prompt,
|
||||||
|
n=1,
|
||||||
|
size="1024x1024"
|
||||||
|
)
|
||||||
|
image_url = response.data[0].url
|
||||||
|
return image_url
|
||||||
|
except openai.BadRequestError as e:
|
||||||
|
return f"`GarfBot Error: ({e.status_code}) - Your request was rejected as a result of our safety system.`"
|
||||||
|
except openai.InternalServerError as e:
|
||||||
|
logger.error(e)
|
||||||
|
return f"`GarfBot Error: ({e.status_code}) - Monday`"
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(e)
|
||||||
|
return f"`GarfBot Error: Lasagna`"
|
||||||
|
|
||||||
async def generate_image(prompt):
|
async def process_image_requests(self):
|
||||||
try:
|
async with aiohttp.ClientSession() as session:
|
||||||
client = AsyncOpenAI(api_key = openaikey)
|
while True:
|
||||||
response = await client.images.generate(
|
request = await self.image_request_queue.get()
|
||||||
model=imgmodel,
|
message = request['message']
|
||||||
prompt=prompt,
|
prompt = request['prompt']
|
||||||
n=1,
|
image_url = await self.generate_image(prompt)
|
||||||
size="1024x1024"
|
if "GarfBot Error" not in image_url:
|
||||||
)
|
logger.info("Downloading & sending image...")
|
||||||
image_url = response.data[0].url
|
async with session.get(image_url) as resp:
|
||||||
return image_url
|
if resp.status == 200:
|
||||||
except openai.BadRequestError as e:
|
image_data = await resp.read()
|
||||||
return f"`GarfBot Error: ({e.status_code}) - Your request was rejected as a result of our safety system.`"
|
ram_image = io.BytesIO(image_data)
|
||||||
except openai.InternalServerError as e:
|
ram_image.seek(0)
|
||||||
logger.error(e)
|
timestamp = message.created_at.strftime('%Y%m%d%H%M%S')
|
||||||
return f"`GarfBot Error: ({e.status_code}) - Monday`"
|
filename = f"{timestamp}_generated_image.png"
|
||||||
except Exception as e:
|
sendfile = discord.File(fp=ram_image, filename=filename)
|
||||||
logger.error(e)
|
try:
|
||||||
return f"`GarfBot Error: Lasagna`"
|
await message.channel.send(file=sendfile)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(e)
|
||||||
|
else:
|
||||||
|
await message.channel.send("`GarfBot Error: Odie`")
|
||||||
|
else:
|
||||||
|
await message.channel.send(image_url)
|
||||||
|
self.image_request_queue.task_done()
|
||||||
|
await asyncio.sleep(2)
|
||||||
|
|
||||||
async def process_image_requests():
|
async def generate_chat(self, question):
|
||||||
async with aiohttp.ClientSession() as session:
|
try:
|
||||||
while True:
|
client = AsyncOpenAI(api_key = self.openaikey)
|
||||||
request = await image_request_queue.get()
|
response = await client.chat.completions.create(
|
||||||
message = request['message']
|
model=self.txtmodel,
|
||||||
prompt = request['prompt']
|
messages=[
|
||||||
image_url = await generate_image(prompt)
|
{"role": "system", "content": "Pretend you are sarcastic Garfield."},
|
||||||
if "GarfBot Error" not in image_url:
|
{"role": "user", "content": f"{question}"}
|
||||||
logger.info("Downloading & sending image...")
|
],
|
||||||
async with session.get(image_url) as resp:
|
max_tokens=400
|
||||||
if resp.status == 200:
|
)
|
||||||
image_data = await resp.read()
|
answer = response.choices[0].message.content
|
||||||
ram_image = io.BytesIO(image_data)
|
return answer.replace("an AI language model", "a cartoon animal")
|
||||||
ram_image.seek(0)
|
except openai.BadRequestError as e:
|
||||||
timestamp = message.created_at.strftime('%Y%m%d%H%M%S')
|
return f"`GarfBot Error: {e}`"
|
||||||
filename = f"{timestamp}_generated_image.png"
|
except openai.APIError as e:
|
||||||
sendfile = discord.File(fp=ram_image, filename=filename)
|
logger.info(e, flush=True)
|
||||||
try:
|
return f"`GarfBot Error: Monday`"
|
||||||
await message.channel.send(file=sendfile)
|
except Exception as e:
|
||||||
except Exception as e:
|
logger.info(e, flush=True)
|
||||||
logger.error(e)
|
return f"`GarfBot Error: Lasagna`"
|
||||||
else:
|
|
||||||
await message.channel.send("`GarfBot Error: Odie`")
|
|
||||||
else:
|
|
||||||
await message.channel.send(image_url)
|
|
||||||
image_request_queue.task_done()
|
|
||||||
await asyncio.sleep(2)
|
|
||||||
|
|
||||||
# GarfChats
|
async def wikisum(self, query):
|
||||||
async def generate_chat(question):
|
try:
|
||||||
try:
|
summary = wikipedia.summary(query)
|
||||||
client = AsyncOpenAI(api_key = openaikey)
|
garfsum = await self.generate_chat(f"Please summarize in your own words: {summary}")
|
||||||
response = await client.chat.completions.create(
|
return garfsum
|
||||||
model=txtmodel,
|
except Exception as e:
|
||||||
messages=[
|
return e
|
||||||
{"role": "system", "content": "Pretend you are sarcastic Garfield."},
|
|
||||||
{"role": "user", "content": f"{question}"}
|
|
||||||
],
|
|
||||||
max_tokens=400
|
|
||||||
)
|
|
||||||
answer = response.choices[0].message.content
|
|
||||||
return answer.replace("an AI language model", "a cartoon animal")
|
|
||||||
except openai.BadRequestError as e:
|
|
||||||
return f"`GarfBot Error: {e}`"
|
|
||||||
except openai.APIError as e:
|
|
||||||
logger.info(e, flush=True)
|
|
||||||
return f"`GarfBot Error: Monday`"
|
|
||||||
except Exception as e:
|
|
||||||
logger.info(e, flush=True)
|
|
||||||
return f"`GarfBot Error: Lasagna`"
|
|
@ -4,45 +4,67 @@ from base64 import b64encode
|
|||||||
from garfpy import logger
|
from garfpy import logger
|
||||||
|
|
||||||
|
|
||||||
client_id = config.CLIENT_ID
|
class Kroger:
|
||||||
client_secret = config.CLIENT_SECRET
|
def __init__(self):
|
||||||
|
self.client_id = config.CLIENT_ID
|
||||||
|
self.client_secret = config.CLIENT_SECRET
|
||||||
|
self.auth = b64encode(f"{self.client_id}:{self.client_secret}".encode()).decode()
|
||||||
|
|
||||||
auth = b64encode(f"{client_id}:{client_secret}".encode()).decode()
|
def kroger_token(self):
|
||||||
|
headers = {
|
||||||
|
'Content-Type': 'application/x-www-form-urlencoded',
|
||||||
|
'Authorization': f'Basic {self.auth}'
|
||||||
|
}
|
||||||
|
|
||||||
def kroger_token():
|
response = requests.post('https://api.kroger.com/v1/connect/oauth2/token', headers=headers, data={
|
||||||
headers = {
|
'grant_type': 'client_credentials',
|
||||||
'Content-Type': 'application/x-www-form-urlencoded',
|
'scope': 'product.compact'
|
||||||
'Authorization': f'Basic {auth}'
|
})
|
||||||
}
|
|
||||||
|
|
||||||
response = requests.post('https://api.kroger.com/v1/connect/oauth2/token', headers=headers, data={
|
response.raise_for_status()
|
||||||
'grant_type': 'client_credentials',
|
return response.json()['access_token']
|
||||||
'scope': 'product.compact'
|
|
||||||
})
|
|
||||||
|
|
||||||
response.raise_for_status()
|
def find_store(self, zipcode, kroken):
|
||||||
return response.json()['access_token']
|
headers = {
|
||||||
|
'Authorization': f'Bearer {kroken}',
|
||||||
|
}
|
||||||
|
params = {
|
||||||
|
'filter.zipCode.near': zipcode,
|
||||||
|
'filter.limit': 1,
|
||||||
|
}
|
||||||
|
response = requests.get('https://api.kroger.com/v1/locations', headers=headers, params=params)
|
||||||
|
return response.json()
|
||||||
|
|
||||||
def find_store(zipcode, kroken):
|
def search_product(self, product, loc_id, kroken):
|
||||||
headers = {
|
logger.info(f"Searching for {product}...")
|
||||||
'Authorization': f'Bearer {kroken}',
|
headers = {
|
||||||
}
|
'Authorization': f'Bearer {kroken}',
|
||||||
params = {
|
}
|
||||||
'filter.zipCode.near': zipcode,
|
params = {
|
||||||
'filter.limit': 1,
|
'filter.term': product,
|
||||||
}
|
'filter.locationId': loc_id,
|
||||||
response = requests.get('https://api.kroger.com/v1/locations', headers=headers, params=params)
|
'filter.limit': 10
|
||||||
return response.json()
|
}
|
||||||
|
response = requests.get('https://api.kroger.com/v1/products', headers=headers, params=params)
|
||||||
|
return response.json()
|
||||||
|
|
||||||
def search_product(product, loc_id, kroken):
|
def garfshop(self, query):
|
||||||
logger.info(f"Searching for {product}...")
|
try:
|
||||||
headers = {
|
query = query.split()
|
||||||
'Authorization': f'Bearer {kroken}',
|
kroken = self.kroger_token()
|
||||||
}
|
product = query[-2]
|
||||||
params = {
|
zipcode = query[-1]
|
||||||
'filter.term': product,
|
loc_data = self.find_store(zipcode, kroken)
|
||||||
'filter.locationId': loc_id,
|
loc_id = loc_data['data'][0]['locationId']
|
||||||
'filter.limit': 10
|
store_name = loc_data['data'][0]['name']
|
||||||
}
|
product_query = self.search_product(product, loc_id, kroken)
|
||||||
response = requests.get('https://api.kroger.com/v1/products', headers=headers, params=params)
|
products = product_query['data']
|
||||||
return response.json()
|
sorted_products = sorted(products, key=lambda item: item['items'][0]['price']['regular'])
|
||||||
|
response = f"Prices for `{product}` at `{store_name}` near `{zipcode}`:\n"
|
||||||
|
for item in sorted_products:
|
||||||
|
product_name = item['description']
|
||||||
|
price = item['items'][0]['price']['regular']
|
||||||
|
response += f"- `${price}`: {product_name} \n"
|
||||||
|
return response
|
||||||
|
except Exception as e:
|
||||||
|
return e
|
||||||
|
63
garfpy/qr.py
Normal file
63
garfpy/qr.py
Normal file
@ -0,0 +1,63 @@
|
|||||||
|
import qrcode
|
||||||
|
from io import BytesIO
|
||||||
|
|
||||||
|
|
||||||
|
def calculate_qr_settings(text):
|
||||||
|
text_length = len(text)
|
||||||
|
|
||||||
|
if text_length <= 25:
|
||||||
|
version = 1
|
||||||
|
box_size = 12
|
||||||
|
elif text_length <= 47:
|
||||||
|
version = 2
|
||||||
|
box_size = 10
|
||||||
|
elif text_length <= 77:
|
||||||
|
version = 3
|
||||||
|
box_size = 8
|
||||||
|
elif text_length <= 114:
|
||||||
|
version = 4
|
||||||
|
box_size = 7
|
||||||
|
elif text_length <= 154:
|
||||||
|
version = 5
|
||||||
|
box_size = 6
|
||||||
|
elif text_length <= 195:
|
||||||
|
version = 6
|
||||||
|
box_size = 5
|
||||||
|
elif text_length <= 224:
|
||||||
|
version = 7
|
||||||
|
box_size = 5
|
||||||
|
elif text_length <= 279:
|
||||||
|
version = 8
|
||||||
|
box_size = 4
|
||||||
|
elif text_length <= 335:
|
||||||
|
version = 9
|
||||||
|
box_size = 4
|
||||||
|
elif text_length <= 395:
|
||||||
|
version = 10
|
||||||
|
box_size = 3
|
||||||
|
else:
|
||||||
|
version = None
|
||||||
|
box_size = 3
|
||||||
|
|
||||||
|
return version, box_size
|
||||||
|
|
||||||
|
async def generate_qr(text):
|
||||||
|
version, box_size = calculate_qr_settings(text)
|
||||||
|
|
||||||
|
qr = qrcode.QRCode(
|
||||||
|
version=version,
|
||||||
|
error_correction=qrcode.constants.ERROR_CORRECT_L,
|
||||||
|
box_size=box_size,
|
||||||
|
border=4,
|
||||||
|
)
|
||||||
|
|
||||||
|
qr.add_data(text)
|
||||||
|
qr.make(fit=True)
|
||||||
|
|
||||||
|
qr_image = qr.make_image(fill_color="black", back_color="white")
|
||||||
|
|
||||||
|
img_buffer = BytesIO()
|
||||||
|
qr_image.save(img_buffer, format='PNG')
|
||||||
|
img_buffer.seek(0)
|
||||||
|
|
||||||
|
return img_buffer
|
163
garfpy/respond.py
Normal file
163
garfpy/respond.py
Normal file
@ -0,0 +1,163 @@
|
|||||||
|
from garfpy import logger
|
||||||
|
import discord
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
|
||||||
|
|
||||||
|
class GarfbotRespond:
|
||||||
|
def __init__(self):
|
||||||
|
self.guild_responses = {}
|
||||||
|
self.responses_file = 'responses.json'
|
||||||
|
|
||||||
|
def load_responses(self):
|
||||||
|
if os.path.exists(self.responses_file):
|
||||||
|
try:
|
||||||
|
with open(self.responses_file, 'r', encoding='utf-8') as f:
|
||||||
|
self.guild_responses = json.load(f)
|
||||||
|
self.guild_responses = {int(k): v for k, v in self.guild_responses.items()}
|
||||||
|
total_responses = sum(len(responses) for responses in self.guild_responses.values())
|
||||||
|
logger.info(f"Loaded responses for {len(self.guild_responses)} server(s), ({total_responses} total responses)")
|
||||||
|
except Exception as e:
|
||||||
|
logger.info(f"Error loading responses: {e}")
|
||||||
|
self.guild_responses = {}
|
||||||
|
else:
|
||||||
|
self.guild_responses = {}
|
||||||
|
|
||||||
|
def save_responses(self):
|
||||||
|
try:
|
||||||
|
save_data = {str(k): v for k, v in self.guild_responses.items()}
|
||||||
|
with open(self.responses_file, 'w', encoding='utf-8') as f:
|
||||||
|
json.dump(save_data, f, indent=2, ensure_ascii=False)
|
||||||
|
total_responses = sum(len(responses) for responses in self.guild_responses.values())
|
||||||
|
logger.info(f"Saved responses for {len(self.guild_responses)} servers ({total_responses} total responses)")
|
||||||
|
except Exception as e:
|
||||||
|
logger.info(f"Error saving responses: {e}")
|
||||||
|
|
||||||
|
def get_responses(self, guild_id):
|
||||||
|
if guild_id not in self.guild_responses:
|
||||||
|
self.guild_responses[guild_id] = {}
|
||||||
|
return self.guild_responses[guild_id]
|
||||||
|
|
||||||
|
async def garfbot_response(self, message, content):
|
||||||
|
guild_id = message.guild.id
|
||||||
|
|
||||||
|
logger.info(message.content)
|
||||||
|
|
||||||
|
match = re.search(r'garfbot response add "(.+)" "(.+)"', content, re.IGNORECASE)
|
||||||
|
if match:
|
||||||
|
trigger = match.group(1)
|
||||||
|
response_text = match.group(2)
|
||||||
|
await self.add_response(message, guild_id, trigger, response_text)
|
||||||
|
return
|
||||||
|
|
||||||
|
match = re.search(r'garfbot response add (\S+) (.+)', content, re.IGNORECASE)
|
||||||
|
if match:
|
||||||
|
trigger = match.group(1)
|
||||||
|
response_text = match.group(2)
|
||||||
|
await self.add_response(message, guild_id, trigger, response_text)
|
||||||
|
return
|
||||||
|
|
||||||
|
match = re.search(r'garfbot\s+response\s+remove\s+(.+)', content, re.IGNORECASE)
|
||||||
|
if match:
|
||||||
|
trigger = match.group(1).strip()
|
||||||
|
await self.remove_response(message, guild_id, trigger)
|
||||||
|
return
|
||||||
|
|
||||||
|
if content.lower() == "garfbot response list":
|
||||||
|
await self.list_responses(message, guild_id)
|
||||||
|
return
|
||||||
|
|
||||||
|
await message.channel.send(
|
||||||
|
"**Garfbot Auto-Response Commands:**\n"
|
||||||
|
"`garfbot response add \"trigger\" \"response\"`\n"
|
||||||
|
"`garfbot response remove \"trigger\"`\n"
|
||||||
|
"`garfbot response list`\n\n"
|
||||||
|
"**Examples:**\n"
|
||||||
|
"`garfbot response add \"hi\" \"Hello there!\"`\n"
|
||||||
|
"`garfbot response add \"that's what\" \"That's what she said!\"`\n"
|
||||||
|
"`garfbot response remove \"hi\"`"
|
||||||
|
)
|
||||||
|
|
||||||
|
async def add_response(self, message, guild_id, trigger, response_text):
|
||||||
|
if not response_text or not trigger:
|
||||||
|
await message.channel.send("❌ Trigger and response must not be null.")
|
||||||
|
return
|
||||||
|
|
||||||
|
responses = self.get_responses(guild_id)
|
||||||
|
responses[trigger] = response_text
|
||||||
|
self.guild_responses[guild_id] = responses
|
||||||
|
self.save_responses()
|
||||||
|
|
||||||
|
embed = discord.Embed(
|
||||||
|
title="✅ Auto-response Added.",
|
||||||
|
color=0x00ff00
|
||||||
|
)
|
||||||
|
embed.add_field(name="Trigger", value=f"`{trigger}`", inline=True)
|
||||||
|
embed.add_field(name="Response", value=f"`{response_text}`", inline=True)
|
||||||
|
embed.set_footer(text=f"Server: {message.guild.name}")
|
||||||
|
|
||||||
|
await message.channel.send(embed=embed)
|
||||||
|
|
||||||
|
async def remove_response(self, message, guild_id, trigger):
|
||||||
|
responses = self.get_responses(guild_id)
|
||||||
|
|
||||||
|
if trigger in responses:
|
||||||
|
removed_response = responses[trigger]
|
||||||
|
del responses[trigger]
|
||||||
|
self.guild_responses[guild_id] = responses
|
||||||
|
self.save_responses()
|
||||||
|
|
||||||
|
embed = discord.Embed(
|
||||||
|
title="✅ Auto-response Removed.",
|
||||||
|
color=0xff6b6b
|
||||||
|
)
|
||||||
|
embed.add_field(name="Trigger", value=f"`{trigger}`", inline=True)
|
||||||
|
embed.add_field(name="Response", value=f"`{removed_response}`", inline=True)
|
||||||
|
embed.set_footer(text=f"Server: {message.guild.name}")
|
||||||
|
|
||||||
|
await message.channel.send(embed=embed)
|
||||||
|
return
|
||||||
|
|
||||||
|
for key in responses.keys():
|
||||||
|
if key.lower() == trigger.lower():
|
||||||
|
removed_response = responses[key]
|
||||||
|
del responses[key]
|
||||||
|
self.guild_responses[guild_id] = responses
|
||||||
|
self.save_responses()
|
||||||
|
|
||||||
|
embed = discord.Embed(
|
||||||
|
title="✅ Auto-response Removed.",
|
||||||
|
color=0xff6b6b
|
||||||
|
)
|
||||||
|
embed.add_field(name="Trigger", value=f"`{key}`", inline=True)
|
||||||
|
embed.add_field(name="Response", value=f"`{removed_response}`", inline=True)
|
||||||
|
embed.set_footer(text=f"Server: {message.guild.name}")
|
||||||
|
|
||||||
|
await message.channel.send(embed=embed)
|
||||||
|
return
|
||||||
|
|
||||||
|
await message.channel.send(f"❌ No auto-response found for trigger: `{trigger}`")
|
||||||
|
|
||||||
|
async def list_responses(self, message, guild_id):
|
||||||
|
responses = self.get_responses(guild_id)
|
||||||
|
|
||||||
|
if not responses:
|
||||||
|
await message.channel.send("No auto-responses configured for this server.")
|
||||||
|
return
|
||||||
|
|
||||||
|
embed = discord.Embed(
|
||||||
|
title=f"Auto-Responses for {message.guild.name}",
|
||||||
|
color=0x3498db
|
||||||
|
)
|
||||||
|
|
||||||
|
for i, (trigger, response) in enumerate(responses.items(), 1):
|
||||||
|
display_response = response[:50] + "..." if len(response) > 50 else response
|
||||||
|
embed.add_field(
|
||||||
|
name=f"{i}. `{trigger}`",
|
||||||
|
value=f"→ {display_response}",
|
||||||
|
inline=False
|
||||||
|
)
|
||||||
|
|
||||||
|
embed.set_footer(text=f"Total responses: {len(responses)}")
|
||||||
|
await message.channel.send(embed=embed)
|
@ -2,3 +2,6 @@ discord.py
|
|||||||
openai
|
openai
|
||||||
aiohttp
|
aiohttp
|
||||||
requests
|
requests
|
||||||
|
wikipedia
|
||||||
|
pillow
|
||||||
|
qrcode
|
Reference in New Issue
Block a user