# -*- coding: utf-8 -*-

import requests.sessions
from BeautifulSoup import BeautifulSoup


def get_source_info(url):
    source_info = {}
    if 'thevideo' in url:
        source_info['source'] = 'thevideo.me'
        with requests.session() as s:
            p = s.get(url)
            soup = BeautifulSoup(p.text, 'html.parser')
            title = soup.findAll('script', src=False, type=False)
            for i in title:
                if "title" in i.prettify():
                    for line in i.prettify().split('\n'):
                        if " title" in line:
                            line = line.replace("title: '", '').replace("',", '')
                            if "720" in line:
                                source_info['qual'] = "720p"
                            elif "1080" in line:
                                source_info['qual'] = "1080p"
                            else:
                                source_info['qual'] = "SD"
        return source_info
    elif 'vidzi' in url:
        return "SD"
