import os
import praw
import sqlite3
import config
import time
import urllib.request as request
from PIL import Image


reddit = praw.Reddit(username=config.USERNAME, password=config.PASSWORD, client_id=config.CLIENT_ID, client_secret=config.CLIENT_SECRET,user_agent=config.USER_AGENT)

data = sqlite3.connect('pics.db')
cur = data.cursor()
cur.execute('CREATE TABLE IF NOT EXISTS hasRemoved(id, author, image)')
data.commit()

class Cancer(object):

	def __init__(self, submission, size):
		self.submission = submission
		self.size = size
		self.url = submission.url
		self.id = submission.id
		self.author = str(submission.author)
		self.query = 'SELECT * FROM hasRemoved WHERE id=?'
		self.entry = 'INSERT INTO hasRemoved VALUES(?,?,?)'
		self.outfile = ['.jpg','.png']
		self.cancer()

	def checkDatabase(self):
		cur.execute(self.query, [self.id])
		if cur.fetchone():
			return True
		else:
			return False

	def submitToDatabase(self):
		cur.execute(self.entry, [self.id, self.author, self.url])
		data.commit()
		return True

	def checkSize(self):
		try:
			if '/a/' in self.url:
				self.submission.mod.remove()
				pass
			try:
				image = request.urlretrieve(self.url, config.PATH+self.outfile[0])
				image = Image.open(config.PATH+self.outfile[0])
				image_size = image.size
			except Exception as e:
				print('Exception:',e)
				image = request.urlretrieve(self.url, config.PATH+self.outfile[0])
				image = Image.open(config.PATH+self.outfile[1])
				image_size = image.size
			print(image_size)
			if self.size[0] < image_size[0] or self.size[1] < image_size[1]:
				print(True)
				return True
			else:
				return False
		except Exception as e:
			print(e)
			#self.submission.mod.remove()
			return None

	def cancer(self):
		try:
			if self.checkDatabase() is False and self.checkSize() is True:
				self.submission.reply(config.REMOVAL_MESSAGE).mod.distinguish(sticky=True)
				self.submission.mod.remove()
				self.submitToDatabase()
				return True
		except Exception as e:
			print(e)
			self.submitToDatabase()


def main():
	while True:
		for submission in reddit.subreddit(config.SUBREDDIT).new(limit=config.LIMIT):
			current_day = int(time.strftime('%d')); current_month = int(time.strftime('%m'))
			if int(time.strftime('%d'), time.localtime(submission.created_utc)) in range(current_day-1, current_day+1) and int(time.strftime('%m'), time.localtime(submission.created_utc)) == current_month:
				Cancer(submission, config.SIZE)

if __name__ == '__main__':
	main()