import sys
from django.conf import settings
from django.core.exceptions import ObjectDoesNotExist
from flockwithme.app.scheduler.models import TwitterUser, TwitterStatus



class addtoDatabase(object):
	def __init__(self,
			 socialprofile = None,
			 twitterUser_ids=None,
			 twitterUser_screen_names=None,
			 twitterStatus_ids=None,
			 twitterStatus_texts=None,
			 *args, **kwargs
			 ):
		self.socialprofile = socialprofile
		self.twitterUser_ids = twitterUser_ids
		self.twitterUser_screen_names = twitterUser_screen_names
		self.twitterStatus_ids =twitterStatus_ids
		self.twitterStatus_texts = twitterStatus_texts
		return super(addtoDatabase, self).__init__(*args, **kwargs)


	def add_friends(self):
		if not self.socialprofile:
			return "No socialprofile"
		for twitter_id in self.twitterUser_ids:
			try:
				tuser = TwitterUser.objects.get(twitter_id=twitter_id)
			except ObjectDoesNotExist:
				print "Object does not exist"
				pass

			self.socialprofile.add_friend(tuser)
			self.socialprofile.save()

	def add_followers(self):
		if not self.socialprofile:
			return "No socialprofile"
		for twitter_id in self.twitterUser_ids:
			try:
				tuser = TwitterUser.objects.get_or_create(twitter_id=twitter_id)
			except ObjectDoesNotExist:
				print "object does not exist"
				pass

			self.socialprofile.add_follower(tuser)
			self.socialprofile.save()

	def add_favorites(self):
		if not self.socialprofile:
			return "No socialprofile"
		for twitter_id in self.twitterStatus_ids:
			try:
				tstatus = TwitterStatus.objects.get(twitter_id=twitter_id)
			except ObjectDoesNotExist:
				print "object does not exist"
				pass
			self.socialprofile.add_favorite(tstatus)
			self.socialprofile.save()

	def add_unfriended(self):
		if not self.socialprofile:
			return "No socialprofile"
		for twitter_id in self.twitterUser_ids:
			try:
				tuser = TwitterUser.objects.get(twitter_id = twitter_id)
			except ObjectDoesNotExist:
				return "Object does not exist"

			self.socialprofile.add_unfriended(tuser)
			self.socialprofile.save()

	def add_tweeted(self):
		if not self.socialprofile:
			return "No socialprofile"
		if not self.twitterUser_screen_names:
			return "No twitter user screen names"

		for s_name in self.twitterUser_screen_names:
			try:
				tuser = TwitterUser.objects.get(screen_name=s_name)
			except ObjectDoesNotExist:
				return "Object does not exist"

			self.socialprofile.add_tweet(tuser)
			self.socialprofile.save()

	def add_dmed(self):
		if not self.socialprofile:
			return "No socialprofile"
		if not self.twitterUser_ids:
			return "No Twitter Ids"
		for twitter_id in self.twitterUser_ids:
			try:
				tuser = TwitterUser.objects.get(twitter_id=twitter_id)
			except ObjectDoesNotExist:
				return "Object Does not exist"
			self.socialprofile.add_dm(tuser)
			self.socialprofile.save()

	def run(self):
		if self.twitterStatus_ids:
			for status in self.twitterStatus_ids:
				try:
					status, created = TwitterStatus.objects.get_or_create(twitter_id=status)
				except Exception, e:
					print e
				status.save()
		if self.twitterStatus_texts:
			for status in self.twitterStatus_texts:
				try:
					status, created = TwitterStatus.objects.get_or_create(text=status)
				except Exception, e:
					print e

				status.save()

		if self.twitterUser_screen_names:
			for screen_name in self.twitterUser_screen_names:
				try:
					tuser, created = TwitterUser.objects.get_or_create(screen_name=screen_name)

				except Exception, e:
					print e
				tuser.save()

		if self.twitterUser_ids:
			print "ids"
			for twitter_id in self.twitterUser_ids:
				try:
					tuser, created = TwitterUser.objects.get_or_create(twitter_id=twitter_id)
				except Exception, e:
					print e
				tuser.save()