import praw
import time
import datetime
import sqlite3
import re
import sys


data = sqlite3.connect('findareddit.db')
cur = data.cursor()
cur.execute('CREATE TABLE IF NOT EXISTS checked (id)')
data.commit()

total = 0
log = {}

cur.execute('CREATE TABLE IF NOT EXISTS logs(session, flaired)')
data.commit()

class submissionSearch(object):

	def __init__(self, submission):
		self.submission = submission
		self.created = submission.created_utc
		self.day_previous = time.mktime(datetime.datetime.now().timetuple()) - (24*60*60)
		self.id = submission.id
		self.author = submission.author
		self.submission.flair_class = None
		self.run()

	def check_database(self):
		cur.execute('SELECT * FROM checked WHERE id=?',[self.id])
		if not cur.fetchone():
			return True
		else:
			return False

	def submit_to_database(self):
		cur.execute('INSERT INTO checked VALUES(?)',[self.id])
		data.commit()
		print('Saved to database!')

	def check_age(self):
		if int(self.created - self.day_previous) > 0:
			return True
		else:
			return False

	def find_suggested(self):
		subreddit_list = []
		for comment in self.submission.comments:
			try:
				subreddit_start = str(comment.body)[re.search('r/',str(comment.body)).span()[1]:]
				subreddit = subreddit_start[subreddit_start.find(' ')]
				subreddit_list.append(subreddit)
			except:
				pass
		return subreddit_list

	def set_flair(self):
		global log, total
		subreddit_list = self.find_suggested()
		for comment in reddit.redditor(str(self.author)).new(limit=100):
			if any(subreddit.lower() in str(comment.subreddit).lower() for subreddit in subreddit_list if str(subreddit).lower() != 'findareddit') and int(comment.created_utc - self.created) > 0 and self.check_database() is True:
				self.submission.mod.flair(text='Found!',css_class='foundflair')
				print('Flaired submission by /u/{}'.format(self.author))
				self.submit_to_database()
				total += 1


	def run(self):
		try:
			if self.check_database() is True and self.check_age() is True:
				self.set_flair()
			else:
				pass
		except Exception as e:
			time.sleep(10)
			if e is KeyboardInterrupt:
				log[str(time.strftime('%G-%m-%d'))] = total
				with open('log.txt','a') as f:
					f.write('\n\n{}'.format(log[str(time.strftime('%G-%m-%d'))]))
					f.close()
				sys.exit()
			else:
				print(e)

def main():
	global reddit
	reddit = praw.Reddit(username='RhodoMod',password='ThunderingMisusingWesterlyDescriptive@123',client_id='4QhscarQGdK50w',client_secret='wA7JnUwWNWdH3-htRR_bToqHIHg',user_agent='Moderator for /r/findareddit')
	while True:
		print('Scanning subreddit...')
		for submission in reddit.subreddit('findareddit').new(limit=100):
			submissionSearch(submission)
		print('Finished scanning, will begin new set momentarily...')
		time.sleep(1800)

if __name__ == '__main__':
	main()