water_thread_wrapup.py

# Tally the votes in the daily water thread

import os
import praw
import re
import time
import sys
from datetime import datetime as dt

import gpio_out as g
import config as c
import post_templates as posts

# Set-up
r = c.getReddit()
sr = c.getSubReddit(r)

path = c.pathPrefix()

if c.checkKillSwitch() == 1:
    sys.exit()

with open(path+'daily_thread.txt', 'r+') as f:
    thread = f.read()
    f.close()

s = r.get_submission(submission_id = thread)
s.lock()
s.unsticky()

# pull ALL top-level comments and calculate score
# Takes FOREVVVVEEEERRR

print "more comments starting"
s.replace_more_comments(limit = None, threshold = 0)
comments = s.comments

def get_comment_score(comment):
    yes = re.search(r'\byes\b', x.body, re.IGNORECASE) or re.search(r'\baye\b', x.body, re.IGNORECASE)
    no = re.search(r'\bno\b', x.body, re.IGNORECASE) or re.search(r'\bnot on your nelly\b', x.body, re.IGNORECASE)
    if yes and no:
        return 0
    elif yes: 
        return 1
    elif no:
        return -1
    else:
        return 0

def reply_to_vote(comment, score=None):
    score = get_comment_score(comment) if (not score) else score
    if score == 0:
        return None
    elif score == 1:
        comment.reply("thanks, I've recorded your vote to water!")
    elif score == -1:
        comment.reply("thanks, I've recorded your vote to not water!")

# Record comments
recorded_yes = {}
recorded_no ={}

for x in comments:
    try:
	    score = get_comment_score(x)
	    tuple_log = (s.id, x.name, x.author.name, x.body, score)
	    if score == 1:
		d = recorded_yes
	    elif score == -1:
		d = recorded_no
	    else:
		continue
	    if not d.has_key(tuple_log[2]):
		d[tuple_log[2]] = tuple_log
		reply_to_vote(x, tuple_log[4])
    except:
            continue

total = len(recorded_yes) - len(recorded_no)

self_text = s.selftext
wrapup = posts.body_edit.format(posts.now_formatted, len(recorded_yes), len(recorded_no))

wrapup = self_text + "\n **** \n" + wrapup

s.edit(wrapup)

with open(path+'yes_votes.txt', 'a+') as f:
    f.write(str(recorded_yes))
    f.close()

with open(path+'no_votes.txt', 'a+') as f:
    f.write(str(recorded_no))
    f.close()

if total > 0:
    with open(path+'topup.txt', 'w') as f:
      f.write('1')
      f.close()
      g.on_off(15)
else:
    with open(path+'topup.txt', 'w') as f:
      f.write('0') 
      f.close()

outcome = 'yes' if total > 1 else 'no'
dow = posts.now_pst.strftime('%A')
history_record = ','.join([dow, outcome, s.id])

with open(path+'history.txt', 'a+') as f:
    new_row = str(history_record+"\n")
    f.write(new_row)    
    f.close()