Compare commits

...

3 Commits

23
main.py
View File

@ -3,6 +3,7 @@ from discord.ext import commands
import requests
import json
from datetime import datetime as dt
import aiohttp
### reads token files
@ -11,6 +12,16 @@ def readToken(filePath):
return f.readline()
### takes a json, returns a http code
async def postToSheet(data, gsheetToken):
async with aiohttp.ClientSession() as session:
async with session.post(gsheetToken, json=data) as response:
print("Status:", response.status)
return response.status
bot = commands.Bot()
# various tokens
@ -51,7 +62,7 @@ async def delivery(ctx,
"Water",
"Water Purifiers"
]),
quantity: discord.Option(discord.SlashCommandOptionType.integer, description="Please be nice and input a value between 1 and 1238 (unless fdev touched something again)") ,
quantity: discord.Option(discord.SlashCommandOptionType.integer, description="Please be nice and input a value between 1 and 1238 (unless fdev touched something again)", min_value=1, max_value=1238),
target: discord.Option(str, choices=['Station', 'Carrier'])
):
@ -72,12 +83,12 @@ async def delivery(ctx,
}
try:
response = requests.post(gsheetToken, json.dumps(data), headers={"Content-Type": "application/json"})
if response.status_code == 200:
response = await postToSheet(data, gsheetToken)
if response == 200:
await ctx.followup.send(f"your delivery of {quantity} of {commodity} to a {target} has been added to the sheet!")
print(f"[{dt.isoformat(dt.now())}]{author} delivery of {quantity} of {commodity} to a {target}")
else:
await ctx.followup.send(f"Failed to log delivery (HTTP {response.status_code}). Please contact the yellow people if that keeps happening")
await ctx.followup.send(f"Failed to log delivery (HTTP {response}). Please contact the yellow people if that keeps happening")
except Exception as e:
await ctx.followup.send(f"help : Error: {e}")
@ -99,7 +110,7 @@ async def changeSheetUrl(ctx,
with open("sheet.token",'w') as f:
f.write(url)
await ctx.followup.send(f"URL has been set")
await ctx.followup.send(f"URL has been set to {url}")
gsheetToken = readToken("sheet.token")
except Exception as e:
await ctx.followup.send(f"something shat the bed")
@ -108,4 +119,4 @@ async def changeSheetUrl(ctx,
#run the damn thing
bot.run(discordToken)
bot.run(discordToken)