class Quote(models.Model):
    id = models.IntegerField(primary_key=True)
    author = models.CharField(max_length=255)
    body = models.CharField(max_length=2047)
    likes = models.IntegerField()

    def __str__(self):
        return self.body

class Story(models.Model):
    title = models.CharField(max_length=200)
    story = models.CharField(max_length=1000)
    name = models.CharField(max_length=200)
    longitude = models.FloatField()
    latitude = models.FloatField()
    location = models.CharField(max_length=200)
    tag = models.CharField(max_length=200)
    date = models.DateTimeField(auto_now_add=True)
    notifyAdmin = models.BooleanField(default=False)

    def __str__(self):
        return self.title

def no_quote_helper(param):

    url ="https://favqs.com/api/qotd"
    headers = {"Authorization": 'Token token="{}"'.format(QUOTE_API_KEY)}

    try:
        response = requests.get(url, headers=headers, params=param)
        quote = response.json()
        q = quote['quote']
        return q
    except: 
        return "No quotes found"

def most_like_quote_helper(quote):
    likemax = -1
    likeselect = -1

    for i in range(len(quote['quotes'])):
        if quote['quotes'][i]['favorites_count'] > likemax: # get the quote with the most likes tagged with the story's tag in the Favqs API
            likemax = quote['quotes'][i]['favorites_count']
            likeselect = i

    return quote['quotes'][likeselect]

class GetQuoteTag(APIView):    
    """
    Get quotes that are tagged with the tag of the given story. 
    """    
    def get(self, request, pk):

        try:
            story=Story.objects.get(pk=pk)
        except: 
            return Response(status=status.HTTP_400_BAD_REQUEST)
        tag = story.tag

        param = {'filter': tag.lower(), 'type': "tag"}
        url = "https://favqs.com/api/quotes/"
        headers = {"Authorization": 'Token token="{}"'.format(QUOTE_API_KEY)}

        try:
            response = requests.get(url, headers=headers, params=param)
            quote = response.json()

            q = most_like_quote_helper(quote)

            if q['body'] == 'No quotes found': # If a quote with that tag doesn't exist return random quote
                q = no_quote_helper(param)
                if q == "No quotes found": # If no random quote is returned
                    raise Exception(q)
            
            qselect = {'id': q['id'],'Quote': q['body'], 'Author': q['author'], 'Likes': q['favorites_count']}
            return Response(qselect, status=status.HTTP_200_OK)
        except:
            return Response(response.status)
        

lass GetQuoteLoc(APIView):
    """
    Get quotes that contain the location of the given story. 
    """    
    def get(self, request, pk):
        likemax = -1
        likeselect = -1
        try:
            story=Story.objects.get(id=pk)
        except:
            return Response(status=status.HTTP_400_BAD_REQUEST)
        location = story.location
        param = {'filter': location}
        url = "https://favqs.com/api/quotes/"
        headers = {"Authorization": 'Token token="{}"'.format(QUOTE_API_KEY)}
        try:
            response = requests.get(url, headers=headers, params=param)
            quote = response.json()
            for i in range(len(quote['quotes'])):
                if quote['quotes'][i]['favorites_count'] > likemax: # get the quote with the most likes containing the location of the story
                    likemax = quote['quotes'][i]['favorites_count']
                    likeselect = i
            q = quote['quotes'][likeselect]
            if q['body'] == 'No quotes found': # if no quote contains the location name
                return Response(q['body']+" with location " + location, status=status.HTTP_400_BAD_REQUEST)
            
            qselect = {'id': q['id'],'Quote': q['body'], 'Author': q['author'], 'Likes': q['favorites_count']}
            return Response(qselect, status=status.HTTP_200_OK)
        except:
            return Response(response.status)

class FavQuote(APIView):
    """
    Add the quote to the database if it is liked. Only the quotes that are retrieved from the Favqs API can be posted.
    """
    def post(self, request, pk):
        quote = {}
        try:
            quote = Quote.objects.get(id=pk)
            quote.likes += 1  # if it is already in the database, increase the number of likes after post request
            quote.save()
            serializer = QuoteSerializer(quote)
            return Response(serializer.data, status=status.HTTP_200_OK)
        except Quote.DoesNotExist:
            url = "https://favqs.com/api/quotes/"+str(pk)
            headers={"Authorization": 'Token token="{}"'.format(QUOTE_API_KEY)}
            response = requests.get(url, headers=headers).json()

            if 'status' in response and response["status"] == 404:
                return Response(response, status=status.HTTP_404_NOT_FOUND)

            quote = {"id":response['id'], "author": response['author'], "body": response['body'], "likes": response['favorites_count']+1}
        
            serializer = QuoteSerializer(data=quote)
            if serializer.is_valid():
                serializer.save()
                return Response(serializer.data, status=status.HTTP_201_CREATED)
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

self.client = Client()
        self.story1 = Story.objects.create(
                id = 1, 
                title = "A Wonderful Day in Istanbul", 
                story = "It was my first time visiting Istanbul. It is full of places to see and enjoy. Can't wait to visit Istanbul again!",
                name = "Leyla", 
                longitude = 41,
                latitude = 28,
                location = "Istanbul",
                tag = "Joy"
        )
        
        self.story2 = Story.objects.create(
                id = 2,
                title = "A Wonderful Day in Istanbul", 
                story = "It was my first time visiting Istanbul. It is full of places to see and enjoy. Can't wait to visit Istanbul again!",
                name = "Leyla", 
                longitude = 41,
                latitude = 28,
                location = "",
                tag = "Joy"
        )

        self.quote = Quote.objects.create(
                id = 700,
                author = "John Keats",
                body = "A thing of beauty is a joy forever: its loveliness increases it will never pass into nothingness.",
                likes = 0
        )

def test_get_quote_tag(self):
        response = self.client.get(reverse('get_quote_tag', args=(self.story.id,)))
        self.assertNotEqual(response.status_code, 400)

def test_get_quote_tag_success(self):
        response = self.client.get(reverse('get_quote_tag', args=(self.story.id,)))
        self.assertEqual(response.status_code, 200)

    def test_get_quote_loc(self):
        response = self.client.get(reverse('get_quote_loc', args=(self.story.id,)))
        self.assertNotEqual(response.status_code, 400)
    

    def test_get_quote_loc_success(self):
        response = self.client.get(reverse('get_quote_loc', args=(self.story.id,)))
        self.assertEqual(response.status_code, 200)

    def test_quote_loc_exist(self):
        response = self.client.get(reverse('get_quote_loc', args=(self.story.id,)))
        self.assertIsNotNone(response.data['Quote'])

    def test_quote_response(self):
        response = self.client.post(reverse('fav_quote', args=(self.quote.id,)))
        self.assertNotEqual(response.status_code, 400)

    def test_post_success(self): 
        response = self.client.post(reverse('fav_quote', args=(self.quote.id,)))
        self.assertEqual(response.status_code, 200)

def test_post_like(self):
        likes = self.quote.likes
        response = self.client.post(reverse('fav_quote', args=(self.quote.id,)))
        self.assertEquals(likes+1, response.data['likes'])