3
u|e;  ã               @   sÌ   d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl	m
Z
 ddlmZ dd	„ Zd
d„ Zdd„ ZG dd„ dƒZG dd„ dƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZeZdS )aî  
Syndication feed generation library -- used for generating RSS, etc.

Sample usage:

>>> from django.utils import feedgenerator
>>> feed = feedgenerator.Rss201rev2Feed(
...     title="Poynter E-Media Tidbits",
...     link="http://www.poynter.org/column.asp?id=31",
...     description="A group Weblog by the sharpest minds in online media/journalism/publishing.",
...     language="en",
... )
>>> feed.add_item(
...     title="Hello",
...     link="http://www.holovaty.com/test/",
...     description="Testing."
... )
>>> with open('test.rss', 'w') as fp:
...     feed.write(fp, 'utf-8')

For definitions of the different versions of RSS, see:
https://web.archive.org/web/20110718035220/http://diveintomark.org/archives/2004/02/04/incompatible-rss
é    N)ÚStringIO)Úurlparse)Ú
iri_to_uri)Úutc)ÚSimplerXMLGeneratorc             C   s*   t | tjƒstjj| tjƒ ƒ} tjj| ƒS )N)Ú
isinstanceÚdatetimeÚcombineÚtimeÚemailÚutilsÚformat_datetime)Údate© r   ú^/var/www/html/StaffProfile/staffvenv/lib/python3.6/site-packages/django/utils/feedgenerator.pyÚrfc2822_date"   s    r   c             C   s:   t | tjƒstjj| tjƒ ƒ} | jƒ | jƒ d kr4dnd S )NÚZÚ )r   r   r	   r
   Ú	isoformatÚ	utcoffset)r   r   r   r   Úrfc3339_date(   s    r   c             C   s8   t | ƒ}d}|dk	r"d|jdƒ }d|j||j|jf S )zˆ
    Create a TagURI.

    See https://web.archive.org/web/20110514113830/http://diveintomark.org/archives/2004/05/28/howto-atom-id
    r   Nz,%sz%Y-%m-%dztag:%s%s:%s/%s)r   ÚstrftimeÚhostnameÚpathÚfragment)Úurlr   ÚbitsÚdr   r   r   Úget_tag_uri.   s
    r   c               @   s|   e Zd ZdZddd„Zdddddddf ddddfdd„Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ ZdS )ÚSyndicationFeedzGBase class for all syndication feeds. Subclasses should provide write()Nc             K   s~   dd„ }|	odd„ |	D ƒ}	||ƒt |ƒ||ƒ||ƒ||ƒ||ƒt |ƒ||ƒ|	pPf t |
ƒ||ƒ|pb|||ƒdœ|–| _g | _d S )Nc             S   s   | d k	rt | ƒS | S )N)Ústr)Úsr   r   r   Úto_str@   s    z(SyndicationFeed.__init__.<locals>.to_strc             S   s   g | ]}t |ƒ‘qS r   )r    )Ú.0Úcr   r   r   ú
<listcomp>B   s    z,SyndicationFeed.__init__.<locals>.<listcomp>)ÚtitleÚlinkÚdescriptionÚlanguageÚauthor_emailÚauthor_nameÚauthor_linkÚsubtitleÚ
categoriesÚfeed_urlÚfeed_copyrightÚidÚttl)r   ÚfeedÚitems)Úselfr&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   Z	feed_guidr2   Úkwargsr"   r   r   r   Ú__init__=   s"    
zSyndicationFeed.__init__c                s‚   dd„ ‰ |o‡ fdd„|D ƒ}| j jˆ |ƒt|ƒˆ |ƒˆ |ƒˆ |ƒt|ƒ||ˆ |ƒˆ |	ƒ|
|p`f |pff ˆ |ƒˆ |ƒdœ|–ƒ dS )zí
        Add an item to the feed. All args are expected to be strings except
        pubdate and updateddate, which are datetime.datetime objects, and
        enclosures, which is an iterable of instances of the Enclosure class.
        c             S   s   | d k	rt | ƒS | S )N)r    )r!   r   r   r   r"   ^   s    z(SyndicationFeed.add_item.<locals>.to_strc                s   g | ]}ˆ |ƒ‘qS r   r   )r#   r$   )r"   r   r   r%   `   s    z,SyndicationFeed.add_item.<locals>.<listcomp>)r&   r'   r(   r*   r+   r,   ÚpubdateÚupdateddateÚcommentsÚ	unique_idÚunique_id_is_permalinkÚ
enclosuresr.   Úitem_copyrightr2   N)r4   Úappendr   )r5   r&   r'   r(   r*   r+   r,   r8   r:   r;   r<   r.   r>   r2   r9   r=   r6   r   )r"   r   Úadd_itemU   s&    	
zSyndicationFeed.add_itemc             C   s
   t | jƒS )N)Úlenr4   )r5   r   r   r   Ú	num_itemst   s    zSyndicationFeed.num_itemsc             C   s   i S )zx
        Return extra attributes to place on the root (i.e. feed/channel) element.
        Called from write().
        r   )r5   r   r   r   Úroot_attributesw   s    zSyndicationFeed.root_attributesc             C   s   dS )zd
        Add elements in the root (i.e. feed/channel) element. Called
        from write().
        Nr   )r5   Úhandlerr   r   r   Úadd_root_elements~   s    z!SyndicationFeed.add_root_elementsc             C   s   i S )zZ
        Return extra attributes to place on each item (i.e. item/entry) element.
        r   )r5   Úitemr   r   r   Úitem_attributes…   s    zSyndicationFeed.item_attributesc             C   s   dS )zF
        Add elements on each item (i.e. item/entry) element.
        Nr   )r5   rD   rF   r   r   r   Úadd_item_elements‹   s    z!SyndicationFeed.add_item_elementsc             C   s   t dƒ‚dS )z‰
        Output the feed in the given encoding to outfile, which is a file-like
        object. Subclasses should override this.
        z;subclasses of SyndicationFeed must provide a write() methodN)ÚNotImplementedError)r5   ÚoutfileÚencodingr   r   r   Úwrite‘   s    zSyndicationFeed.writec             C   s   t ƒ }| j||ƒ |jƒ S )zD
        Return the feed in the given encoding as a string.
        )r   rL   Úgetvalue)r5   rK   r!   r   r   r   ÚwriteString˜   s    zSyndicationFeed.writeStringc             C   s^   d}d}x>| j D ]4}x.|D ]&}|j|ƒ}|r|dks<||kr|}qW qW |p\tjjƒ jtdS )z
        Return the latest item's pubdate or updateddate. If no items
        have either of these attributes this return the current UTC date/time.
        Nr9   r8   )Útzinfo)r9   r8   )r4   Úgetr   ÚutcnowÚreplacer   )r5   Zlatest_dateZ	date_keysrF   Zdate_keyZ	item_dater   r   r   Úlatest_post_date    s    

z SyndicationFeed.latest_post_date)
NNNNNNNNNN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r7   r@   rB   rC   rE   rG   rH   rL   rN   rS   r   r   r   r   r   ;   s      
r   c               @   s   e Zd ZdZdd„ ZdS )Ú	EnclosurezAn RSS enclosurec             C   s   || | _ | _t|ƒ| _dS )z#All args are expected to be stringsN)ÚlengthÚ	mime_typer   r   )r5   r   rY   rZ   r   r   r   r7   µ   s    zEnclosure.__init__N)rT   rU   rV   rW   r7   r   r   r   r   rX   ³   s   rX   c               @   s8   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ ZdS )ÚRssFeedz"application/rss+xml; charset=utf-8c             C   s^   t ||ƒ}|jƒ  |jd| jƒ ƒ |jd| jƒ ƒ | j|ƒ | j|ƒ | j|ƒ |jdƒ d S )NZrssÚchannel)	r   ÚstartDocumentÚstartElementÚrss_attributesrC   rE   Úwrite_itemsÚendChannelElementÚ
endElement)r5   rJ   rK   rD   r   r   r   rL   ¾   s    



zRssFeed.writec             C   s   | j ddœS )Nzhttp://www.w3.org/2005/Atom)Úversionz
xmlns:atom)Ú_version)r5   r   r   r   r_   È   s    zRssFeed.rss_attributesc             C   s<   x6| j D ],}|jd| j|ƒƒ | j||ƒ |jdƒ qW d S )NrF   )r4   r^   rG   rH   rb   )r5   rD   rF   r   r   r   r`   Î   s    zRssFeed.write_itemsc             C   sö   |j d| jd ƒ |j d| jd ƒ |j d| jd ƒ | jd d k	r^|j dd d| jd dœƒ | jd d k	r~|j d| jd ƒ x| jd	 D ]}|j d
|ƒ qŠW | jd d k	r¾|j d| jd ƒ |j dt| jƒ ƒƒ | jd d k	rò|j d| jd ƒ d S )Nr&   r'   r(   r/   z	atom:linkr5   )ÚrelÚhrefr)   r.   Úcategoryr0   Ú	copyrightZlastBuildDater2   )ÚaddQuickElementr3   r   rS   )r5   rD   Úcatr   r   r   rE   Ô   s    zRssFeed.add_root_elementsc             C   s   |j dƒ d S )Nr\   )rb   )r5   rD   r   r   r   ra   ä   s    zRssFeed.endChannelElementN)	rT   rU   rV   Úcontent_typerL   r_   r`   rE   ra   r   r   r   r   r[   »   s   
r[   c               @   s   e Zd ZdZdd„ ZdS )ÚRssUserland091Feedz0.91c             C   s@   |j d|d ƒ |j d|d ƒ |d d k	r<|j d|d ƒ d S )Nr&   r'   r(   )ri   )r5   rD   rF   r   r   r   rH   ë   s    z$RssUserland091Feed.add_item_elementsN)rT   rU   rV   rd   rH   r   r   r   r   rl   è   s   rl   c               @   s   e Zd ZdZdd„ ZdS )ÚRss201rev2Feedz2.0c             C   sº  |j d|d ƒ |j d|d ƒ |d d k	r<|j d|d ƒ |d rj|d rj|j dd|d |d f ƒ n8|d r„|j d|d ƒ n|d r¢|j d|d d	d
iƒ |d d k	rÂ|j dt|d ƒƒ |d d k	rÞ|j d|d ƒ |d d k	r(i }t|jdƒtƒrt|d ƒjƒ |d< |j d|d |ƒ |d d k	rF|j d|d ƒ |d r–t|d ƒ}t|ƒdkrrt	dƒ‚|d }|j dd|j
|j|jdœƒ x|d D ]}|j d|ƒ q W d S )Nr&   r'   r(   r+   r*   Úauthorz%s (%s)z
dc:creatorzxmlns:dcz http://purl.org/dc/elements/1.1/r8   ZpubDater:   r;   r<   ZisPermaLinkZguidr2   r=   é   zrRSS feed items may only have one enclosure, see http://www.rssboard.org/rss-profile#element-channel-item-enclosurer   Ú	enclosurer   )r   rY   Útyper.   rg   )ri   r   r   rP   Úboolr    ÚlowerÚlistrA   Ú
ValueErrorr   rY   rZ   )r5   rD   rF   Z
guid_attrsr=   rp   rj   r   r   r   rH   ö   sD    
z Rss201rev2Feed.add_item_elementsN)rT   rU   rV   rd   rH   r   r   r   r   rm   ò   s   rm   c               @   s<   e Zd ZdZdZdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dS )Ú	Atom1Feedz#application/atom+xml; charset=utf-8zhttp://www.w3.org/2005/Atomc             C   sD   t ||ƒ}|jƒ  |jd| jƒ ƒ | j|ƒ | j|ƒ |jdƒ d S )Nr3   )r   r]   r^   rC   rE   r`   rb   )r5   rJ   rK   rD   r   r   r   rL   +  s    


zAtom1Feed.writec             C   s.   | j d d k	r | j| j d dœS d| jiS d S )Nr)   )Úxmlnszxml:langrw   )r3   Úns)r5   r   r   r   rC   3  s    zAtom1Feed.root_attributesc             C   s`  |j d| jd ƒ |j ddd| jd dœƒ | jd d k	rT|j ddd| jd dœƒ |j d| jd ƒ |j d	t| jƒ ƒƒ | jd
 d k	rð|jdi ƒ |j d| jd
 ƒ | jd d k	rÆ|j d| jd ƒ | jd d k	ræ|j d| jd ƒ |jdƒ | jd d k	r|j d| jd ƒ x&| jd D ]}|j ddd|iƒ qW | jd d k	r\|j d| jd ƒ d S )Nr&   r'   r   Ú	alternate)re   rf   r/   r5   r1   Úupdatedr+   rn   Únamer*   r   r,   Úurir-   r.   rg   Útermr0   Úrights)ri   r3   r   rS   r^   rb   )r5   rD   rj   r   r   r   rE   9  s(    
zAtom1Feed.add_root_elementsc             C   s<   x6| j D ],}|jd| j|ƒƒ | j||ƒ |jdƒ qW d S )NÚentry)r4   r^   rG   rH   rb   )r5   rD   rF   r   r   r   r`   O  s    zAtom1Feed.write_itemsc          	   C   s¤  |j d|d ƒ |j dd|d ddœƒ |d d k	rH|j dt|d ƒƒ |d d k	rh|j d	t|d ƒƒ |d
 d k	rÒ|jdi ƒ |j d|d
 ƒ |d d k	r¬|j d|d ƒ |d d k	rÈ|j d|d ƒ |jdƒ |d d k	rè|d }nt|d |d ƒ}|j d|ƒ |d d k	r*|j d|d ddiƒ x0|d D ]$}|j ddd|j|j|jdœƒ q4W x$|d D ]}|j ddd|iƒ qfW |d d k	r |j d|d ƒ d S )Nr&   r'   r   ry   )rf   re   r8   Z	publishedr9   rz   r+   rn   r{   r*   r   r,   r|   r;   r1   r(   Úsummaryrq   Úhtmlr=   rp   )re   rf   rY   rq   r.   rg   r}   r>   r~   )ri   r   r^   rb   r   r   rY   rZ   )r5   rD   rF   r;   rp   rj   r   r   r   rH   U  s<    

zAtom1Feed.add_item_elementsN)
rT   rU   rV   rk   rx   rL   rC   rE   r`   rH   r   r   r   r   rv   &  s   rv   )rW   r   r   Úior   Úurllib.parser   Zdjango.utils.encodingr   Zdjango.utils.timezoner   Zdjango.utils.xmlutilsr   r   r   r   r   rX   r[   rl   rm   rv   ZDefaultFeedr   r   r   r   Ú<module>   s"   x-
4b