deathbybandaid/SpiceBot-migrated
1
0
Fork 0
mirror of https://github.com/SpiceBot/SpiceBot.git synced 2026-11-13 15:09:54 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

test

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
sopel_modules/SpiceBot/Reddit.py
View File

@ -6,20 +6,45 @@ This is the SpiceBot Reddit system.
import spicemanip import spicemanip
import requests
from fake_useragent import UserAgent
class BotReddit(): class BotReddit():
"""This Logs all channels known to the server""" """This Logs all channels known to the server"""
def __init__(self): def __init__(self):
self.dict = {} self.dict = {}
self.header = {'User-Agent': str(UserAgent().chrome)}
def prerun(self, trigger): 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_args = spicemanip.main(trigger.args[1], 'create')
trigger.sb = { trigger_argsdict = {
"slashcomm": spicemanip.main(trigger_args, 1).lower()[:1], "slashcomm": spicemanip.main(trigger_args, 1).lower()[:1],
"command": spicemanip.main(trigger_args, 1).lower()[2:], "command": spicemanip.main(trigger_args, 1).lower()[2:],
"args": spicemanip.main(trigger_args, "2+", 'list') "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() reddit = BotReddit()

8
sopel_modules/SpiceBot_Core_Commands/Reddit.py
View File

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