Compare commits
3 Commits
4f6a0017d3
...
9a35c7c430
| Author | SHA1 | Date | |
|---|---|---|---|
| 9a35c7c430 | |||
| 28d2e4efab | |||
| 224c0517ae |
23
main.py
23
main.py
|
|
@ -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)
|
||||
Loading…
Reference in New Issue
Block a user