from imgurpython import ImgurClient from operator import itemgetter from database import Database import datetime import logging import ytapi import graph import time import praw import json import re import os with open("config.json", "r") as f: CONFIG = json.load(f) REDDIT = praw.Reddit(**CONFIG["redditapi"]) SUBREDDIT = REDDIT.subreddit(CONFIG["subreddit"]) COMMENT_TAIL = CONFIG["comment_tail"] FREE_FLAIRS = CONFIG["free_flairs"] IMGUR = ImgurClient(**CONFIG["imgurapi"]) db = Database() logging.basicConfig( format = "[%(asctime)s] %(message)s", level = logging.INFO, handlers=[ logging.FileHandler("actions.log"), logging.StreamHandler() ]) def get_time(): #this is not the correct way to do this but I don't care return time.strftime("%b %d %Y %H:%M:%S", time.gmtime()) def display(message): logging.info(message) def get_lambda_from_flair(s): result = re.search("\[(.*)\]", s) if result is not None and "λ" in result.group(1): return result.group(1) else: return "" def update_users_flair_from_comment(comment): #implemented only for legacy update_users_flair(str(comment.author)) def get_medal(actualscore): if actualscore >= 10 and actualscore < 25: return "🥉 Bronze " elif actualscore >= 25 and actualscore < 50: return "🥈 Silver " elif actualscore >= 50 and actualscore < 100: return "🥇 Gold " elif actualscore > 100: return "🏆 Platinum " else: return "" def update_users_flair(username): flairtext = next(SUBREDDIT.flair(redditor=username))["flair_text"] if flairtext is None: flairtext = "" else: flairscore = get_lambda_from_flair(flairtext) flairtext = str(flairtext.replace("[%s] " % flairscore, "")) if username in get_mods(): newflair = "[🏆 ∞λ] %s" % (flairtext) else: actualscore = db.get_lambda(username)[0] newflair = "[%s%iλ] %s" % (get_medal(actualscore), actualscore, flairtext) logging.info("/u/%s had their flair updated" % username) SUBREDDIT.flair.set(redditor = username, text = newflair) def get_mods(): return [str(i) for i in SUBREDDIT.moderator()] + ["AutoModerator"] def update_tables(scores, data): content = "" date = str(datetime.date.today()) mods = get_mods() imagepath = graph.make_graph(data) imageurl = upload_image(imagepath, date) bylambda = [i for i in sorted(scores, key = itemgetter(1), reverse = True) if i[0] not in mods][:10] byhelps = sorted(scores, key = itemgetter(2), reverse = True)[:10] SUBREDDIT.stylesheet.upload("wikigraph", imagepath) content += "\n\n##/r/SmallYTChannel lambda tables: %s" % date content += "\n\n###By lambda:" content += "\n\nUsername|Lambda|Help given\n:--|:--|:--" for line in bylambda: content += "\n/u/%s|%i|%i" % (line[0], line[1], line[2]) content += "\n\n###By Help given:" content += "\n\nUsername|Lambda|Help given\n:--|:--|:--" for line in byhelps: λ = str(line[1]) if line[0] in mods: λ = "∞" content += "\n/u/%s|%s|%i" % (line[0], λ, line[2]) content += "\n\n##Statistics from %s:\n\nIf you're looking at this through the wiki, not through the bot's profile, then" % (date) content += "the most up-to-date graph will be shown below. To see the graph at this date, follow [this link.](%s)" % (imageurl) content += "\n\n![](%%%%wikigraph%%%%)\n\nTotal λ in circulation|Useful advice given|Unique users\n:--|:--|:--\n%i|%i|%i" % (data[-1][1], data[-1][2], data[-1][3]) REDDIT.subreddit("u_SmallYTChannelBot").submit("/r/SmallYTChannel Statistics: %s" % date, url = imageurl).reply(content).mod.distinguish(sticky = True) SUBREDDIT.wiki["lambdatables"].edit(content, reason = "Update: %s" % date) SUBREDDIT.wiki[date].edit(content, reason = "Update: %s" % date) currentdata = SUBREDDIT.wiki["index"].content_md currentdata += "\n\n* [%s](/r/SmallYTChannel/wiki/%s)" % (date, date) SUBREDDIT.wiki["index"].edit(currentdata, reason = "Update: %s" % date) def upload_image(path, date): config = { 'album': None, 'name': 'SmallYTChannelBot Statistics graph: %s' % date, 'title': 'SmallYTChannelBot Statistics graph: %s' % date, 'description': 'SmallYTChannelBot Statistics graph: %s' % date } image = IMGUR.upload_from_path(path, config = config) return "https://i.imgur.com/%s.png" % image["id"] def every_day(): display("Starting every day program...") display("Updating database statistics...") db.update_stats() display("Posting and updating wiki...") update_tables(db.get_scores(), db.get_stats()) display("Formatting leaderboard...") leaderboard = format_monthly_leaderboard() display("Updating sidebar...") #it'd be cool to find a way to access this directly without iteration for widget in SUBREDDIT.widgets.sidebar: if widget.shortName == "Monthly Lambda Leaderboard": widget.mod.update(text = leaderboard) display("Updated in new reddit...") sidebar = SUBREDDIT.mod.settings()["description"] oldtable = sidebar.split("------")[-1] SUBREDDIT.mod.update(description = sidebar.replace(oldtable, "\n\n## Monthly Lambda Leaderboard\n\n" + leaderboard)) display("Updated in old reddit...") display("Completed.") def handle_mylambda(comment): author = str(comment.author) λ, links = db.get_lambda(author) if author in get_mods(): text = "/u/%s is a moderator, and therefore has ∞λ." % author else: text = "/u/%s currently has %iλ, and has helped helping the following posts:" % (author, λ) count = 0 for link in links: if "www.reddit.com" not in link: link = "https://www.reddit.com" + link #set a max limit on the number of times this will iterate to stop it #breaking because of Reddit's character limit. count += 1 text += "\n\n- [%s](%s)" % (REDDIT.submission(url = link).title, link) if count > 100: #experminent with this number text += "\n\n[%i more...]" % len(links) - count break update_users_flair_from_comment(comment) return text def handle_givelambda(comment): submission = comment.submission parentauthour = str(comment.parent().author) op = str(comment.author) if op == parentauthour: text = "You cannot give yourself λ." elif parentauthour == "SmallYTChannelBot": text = "Please only give lambda to humans." elif str(comment.author) in get_mods(): text = "The moderator /u/%s has given /u/%s 1λ. /u/%s now has %iλ." % (str(comment.author), parentauthour, parentauthour, db.get_lambda(parentauthour)[0] + 1) db.give_lambda(parentauthour, submission.permalink, timestamp = int(submission.created_utc)) display(text) elif submission.link_flair_text in FREE_FLAIRS: text = "You cannot give lambda in free posts anymore." elif op != str(submission.author): text = "Only the OP can give λ." elif db.user_given_lambda(parentauthour, str(submission.permalink)): text = "You have already given /u/%s λ for this submission. Why not give λ to another user instead?" % parentauthour else: display("'/u/%s' has given '/u/%s' lambda!" % (op, parentauthour)) text = "You have given /u/%s 1λ. /u/%s now has %iλ" % (parentauthour, parentauthour, db.get_lambda(parentauthour)[0] + 1) #bonus lambda giving was removed # if not db.link_in_db(submission.permalink) or not db.link_in_db(submission.permalink.replace("https://www.reddit.com", "")): # db.give_lambda(parentauthour, submission.permalink, op) # display("The OP received lambda too!") # else: db.give_lambda(parentauthour, submission.permalink, timestamp = int(submission.created_utc)) # update_users_flair_from_comment(comment) update_users_flair_from_comment(comment.parent()) return text def handle_takelambda(comment): try: splitted = comment.body.split() user = splitted[1].replace("/u/", "").replace("u/", "") toremove = int(splitted[2].replace("\\", "")) reason = " ".join(splitted[3:]) text = "/u/%s has had %iλ taken away from them for the reason '%s'. /u/%s now has %iλ" % (user, toremove, reason, user, db.get_lambda(user)[0] - toremove) db.change_lambda(user, -toremove) display("A moderator removed %i lambda from /u/%s for the reason '%s'" % (toremove, user, reason)) except Exception as e: display("{ERROR while removing λ} %s" % e) text = r"An error was encountered. Please use the syntax `!takelambda [user] [how much to remove {integer}] [reason]`" + "\n\nThe error was:\n\n" + str(e) update_users_flair(user) return text def handle_refundlambda(comment): try: splitted = comment.body.split() user = splitted[1].replace("/u/", "").replace("u/", "") toadd = int(splitted[2].replace("\\", "")) reason = " ".join(splitted[3:]) text = "/u/%s has had %iλ refunded for the reason '%s'. /u/%s now has %iλ" % (user, toadd, reason, user, db.get_lambda(user)[0] + toadd) db.change_lambda(user, toadd) display("A moderator refunded %i lambda from /u/%s for the reason '%s'" % (toadd, user, reason)) except Exception as e: display("{ERROR while refunding λ} %s" % e) text = r"An error was encountered. Please use the syntax `!refundlambda [user] [how much to add {integer}] [reason]`" + "\n\nThe error was:\n\n" + str(e) update_users_flair(user) return text def handle_submission(submission): score = db.get_lambda(str(submission.author))[0] if submission.link_flair_text in FREE_FLAIRS: if "youtube.com" in str(submission.url) or "youtu.be" in str(submission.url): text = "Your post has been removed because it has the wrong flair. [Discussion], [Meta] and [Collab] flairs are only for text submissions." submission.mod.remove() display("/u/%s had their submission removed for using the wrong flair." % submission.author) else: text = "Your post is a discussion, meta or collab post so it costs 0λ." else: if score < 3: text = """Thank you for submitting to /r/SmallYTChannel. Unfortunally, you submission has been removed since you do not have enough λ. You need 3λ to post. You currently have %iλ. For more information, read the [FAQ.](https://www.reddit.com/user/SmallYTChannelBot/comments/a4u7qj/smallytchannelbot_faq/)""" % score submission.mod.remove() display("/u/%s had their submission removed for insufficient lambda." % submission.author) else: text = """Thank you for submitting to /r/SmallYTChannel. You have spent 3λ to submit here, making your current balance %iλ. /u/%s, please comment `!givelambda` to the most helpful advice you are given. For more information, read the [FAQ.](https://www.reddit.com/user/SmallYTChannelBot/comments/a4u7qj/smallytchannelbot_faq/)""" % (score - 3, str(submission.author)) db.change_lambda(str(submission.author), -3) try: ytid = ytapi.get_videoId_from_url(submission.url) if "/" not in ytid: ytdata = ytapi.get_video_data(ytid) text += """ \n\n\n##Video data: **Field**|**Data** :-|:- Title|%s Thumbnail|[Link](%s) Views|%s Length|%s Likes/Dislikes|%s/%s Comments|%s Description|%s ##Channel Data: **Field**|**Data** :-|:- Name|%s Thumbnail|[Link](%s) Subscribers|%s Videos|%s Views|%s """ % ( ytdata["title"], ytdata["thumbnail"], ytdata["views"], ytdata["length"], ytdata["likes"], ytdata["dislikes"], ytdata["comments"], ytdata["description"], ytdata["channel"], ytdata["channelThumb"], ytdata["subscribers"], ytdata["videos"], ytdata["channelViews"] ) curflair = submission.link_flair_text if str(curflair) != "None": submission.mod.flair(" %s | %s | :youtube: %s" % (curflair, ytdata["length"], ytdata["channel"])) else: submission.mod.flair("%s | :youtube: %s" % (ytdata["length"], ytdata["channel"])) except: pass update_users_flair(str(submission.author)) return text def main(): comment_stream = SUBREDDIT.stream.comments(pause_after=-1) submission_stream = SUBREDDIT.stream.submissions(pause_after=-1) while True: try: for comment in comment_stream: if comment is None: break if not db.id_in_blacklist(comment.id): db.add_to_blacklist(comment.id) response = None if "!mylambda" in comment.body.lower() and str(comment.author) != "SmallYTChannelBot": response = handle_mylambda(comment) if "!givelambda" in comment.body.lower() and str(comment.author) != "SmallYTChannelBot": response = handle_givelambda(comment) if comment.body.startswith("!takelambda") and str(comment.author) in get_mods(): response = handle_takelambda(comment) if comment.body.startswith("!refundlambda") and str(comment.author) in get_mods(): response = handle_refundlambda(comment) if response is not None: reply = comment.reply(response + COMMENT_TAIL) reply.mod.distinguish(sticky = False) for submission in submission_stream: if submission is None: break if not db.id_in_blacklist(submission.id): db.add_to_blacklist(submission.id) display("There has been a new submission: '%s', with flair '%s'" % (submission.title, submission.link_flair_text)) response = None if str(submission.author) not in get_mods(): response = handle_submission(submission) reply = submission.reply(response + COMMENT_TAIL) reply.mod.distinguish(sticky = True) reply.mod.approve() except Exception as e: display("{ERROR} %s" % e) continue def get_submission_times(permalink): if not permalink.startswith("https://www.reddit.com"): permalink = "https://www.reddit.com" + permalink submission = REDDIT.submission(url = permalink) return submission.created_utc def add_times_to_lambdas(): updated_permalinks = [] for id_, permalink, user, created in db.get_all_lambdas(): if created is None and permalink not in updated_permalinks: db.add_date_to_permalink(permalink, get_submission_times(permalink)) updated_permalinks.append(permalink) logging.info("Added date for permalink %s" % permalink) def format_monthly_leaderboard(): leaderboard = db.get_lambda_leaderboard() out = "**Username**|**Medal**|**Times Helped**|**Lambda**\n:-|:-|:-|:-\n" for username, times_helped, λ in leaderboard: out += "/u/%s|%1s|%s|%sλ\n" % (username, get_medal(λ)[:-1], times_helped, λ) return out + "\nLast updated: %s" % get_time() if __name__ == "__main__": file = open("pid.txt", "w") file.write(str(os.getpid())) file.close() display("\n####################\n[%s] RESTARTED\n####################\n" % get_time()) main()