import requests
import json
import pandas as pd
from configparser import ConfigParser
from .log import logger
from .exceptions import QualitubeException


config = ConfigParser()
config.read("config.ini")
API_KEY = config['credentials']['api_key']


class Videos:
    """
    Wrapper Class to the YouTube Data API v3's `Videos` endpoint with
    extra functionalities.
    """
    def __init__(self, videos_ids, api_key=API_KEY):
        self.videos_ids = videos_ids
        self.api_key = api_key
    
    def _try_parse(self, item, key):
        try:
            parsed = item[key]
        except KeyError:
            logger.warn(
                f"YouTube Data API v3 does not provide the `{key}` parameter fo"
                f"r the requested video. Setting it as `None`"
            )
            parsed = None
        return parsed

    def _parse_response(self, data):
        raw = json.loads(data)
        try:
            items = raw['items']
        except KeyError:
            if "error" in raw.keys():
                raise QualitubeException(
                    f"\nAre you sure you set qualitube's config.ini file correctly?"
                    f"\nYou are getting the following error from YouTube's API response:"
                    f"\n\t{raw}"
                )
            raise

        parsed = []
        for item in items:
            parsed.append({
                'channel_id': self._try_parse(item['snippet'], 'channelId'),
                'channel_title': self._try_parse(item['snippet'], 'channelTitle'),
                'video_id': self._try_parse(item, 'id'),
                'video_title': self._try_parse(item['snippet'], 'title'),
                'video_description': self._try_parse(item['snippet'], 'description'),
                'video_tags': self._try_parse(item['snippet'], 'tags'),
                'video_published_at': self._try_parse(item['snippet'], 'publishedAt'),
                'video_view_count': self._try_parse(item['statistics'], 'viewCount'),
                'video_like_count': self._try_parse(item['statistics'], 'likeCount'),
                'video_dislike_count': self._try_parse(item['statistics'], 'dislikeCount'),
                'video_favorite_count': self._try_parse(item['statistics'], 'favoriteCount'),
                'video_comment_count': self._try_parse(item['statistics'], 'commentCount')
            })
            logger.info(f"Got Video -> id: {item['id']} title: {item['snippet']['title']}")

        try:
            next_page_token = raw["nextPageToken"]
        except KeyError:
            next_page_token = False
        
        return parsed, next_page_token

    def _get_ids_parameter(self):
        ids = ','.join(self.videos_ids)
        return ids

    def get_data(self, page_token=None):
        headers = {
            'Accept': 'application/json',
        }

        params = [
            ('part', 'snippet,statistics'),
            ('id', self._get_ids_parameter()),
            ('key', self.api_key)
        ]

        if page_token:
            params.append(
                ('pageToken', page_token)
            )

        r = requests.get('https://youtube.googleapis.com/youtube/v3/videos', headers=headers, params=params)
        
        videos_data, next_page_token = self._parse_response(r.text)

        if next_page_token:
            return videos_data + self.get_data(next_page_token)

        return videos_data
    
    def to_df(self):
        df = pd.DataFrame(self.get_data())
        return df

{'avg_line_length': 33.75961538461539, 'max_line_length': 108, 'alphanum_fraction': 0.569068641412703, 'licenses': ['MIT'], 'repository_name': 'LABHDUFBA/qualitube', 'path': 'qualitube/videos.py', 'size': 3511, 'lang': 'Python'}