This commit is contained in:
deathbybandaid 2019-06-19 10:10:31 -04:00
parent 3b102c5186
commit 1258fb38b9
2 changed files with 31 additions and 6 deletions

View File

@ -6,20 +6,45 @@ This is the SpiceBot Reddit system.
import spicemanip
import requests
from fake_useragent import UserAgent
class BotReddit():
"""This Logs all channels known to the server"""
def __init__(self):
self.dict = {}
self.header = {'User-Agent': str(UserAgent().chrome)}
def prerun(self, trigger):
# trigger_argsdict = self.trigger_handler(trigger)
page = self.check_reddit_up()
if not page:
message = "Reddit Appears to be down"
return message
elif str(page.status_code).startswith(tuple(["4", "5"])):
message = "Reddit is replying with http code " + str(page.status_code)
return message
message = "Reddit is replying with http code " + str(page.status_code)
return message
def trigger_handler(self, trigger):
trigger_args = spicemanip.main(trigger.args[1], 'create')
trigger.sb = {
trigger_argsdict = {
"slashcomm": spicemanip.main(trigger_args, 1).lower()[:1],
"command": spicemanip.main(trigger_args, 1).lower()[2:],
"args": spicemanip.main(trigger_args, "2+", 'list')
}
return trigger
return trigger_argsdict
def check_reddit_up(self):
try:
page = requests.get("https://www.reddit.com/", headers=self.header)
except Exception as e:
page = e
page = None
return page
reddit = BotReddit()

View File

@ -13,8 +13,8 @@ import sopel_modules.SpiceBot as SpiceBot
)
""")
def bot_command_reddit_user(bot, trigger):
trigger = SpiceBot.reddit.prerun(trigger)
bot.osd(str(trigger.sb))
message = SpiceBot.reddit.prerun(trigger)
bot.osd(str(message))
@sopel.module.rule(r"""(?:)r/
@ -23,5 +23,5 @@ def bot_command_reddit_user(bot, trigger):
)
""")
def bot_command_reddit_subreddit(bot, trigger):
trigger = SpiceBot.reddit.prerun(trigger)
bot.osd(str(trigger.sb))
message = SpiceBot.reddit.prerun(trigger)
bot.osd(str(message))