� �F�V~ � @ s� d d l Z d d l Z d d l Z d d l m Z d d l Z d d l m Z d d d d d � Z d d � Z d d � Z d d � Z d S)� N)� ElementTree)�open�en� c C sz d | d | d | d | } t j j | � j � } t | � sU t d � � n t j d d � } t j | d | �S) z3Get closed captions from YouTube and parse the XML.z#http://youtube.com/api/timedtext?v=z&lang=z&name=z&tlang=zNo CC available�encodingzutf-8�parser) �urllib�request�urlopen�read�len� RuntimeError�ET� XMLParser� fromstring)�video_id�lang�trackZtlangZcc_urlZ xml_sourceZ utf8_parser� r �../code/getyoutubecc.py�get_cc s " r c C sx | j � } d d � | D� } d d � | D� } i d d � | D� d 6d d � | D� d 6d d � | D� d 6} | S) z5Translate a single language YouTube XML to EdX sjson.c S sH g | ]> } t | j d � t | j d � | j j d d � f � q S)�start�dur� � )�float�attrib�text�replace)�.0�ir r r � <listcomp> s z xml2sjson.<locals>.<listcomp>c S s@ g | ]6 \ } } } t | d � t | | d � | f � q S)g @�@)�int)r r �j�kr r r r! s c S s g | ] } | d � q S)r r )r r r r r r! s r c S s g | ] } | d � q S)� r )r r r r r r! s �endc S s g | ] } | d � q S)� r )r r r r r r! s r )� getchildren)�cc�entries�resultr r r � xml2sjson s r, c C s/ t | d d � � } t j | | � Wd QXd S)N�wzutf-8)r �json�dump)�sjson�filename�fr r r � write_sjson! s r3 c C s) t | � } t | � } t | | � d S)N)r r, r3 )r r1 �feedr0 r r r �save_youtube_cc& s r5 )�urllib.requestr � urllib.error� urllib.parseZ xml.etreer r r. �codecsr r r, r3 r5 r r r r �<module> s $
