U
    {hF                     @   s    d dl mZ G dd deZdS )    )Sitemapc                   @   s>   e Zd ZdddZdd Zdd Zdd	 Zd
d Zdd ZdS )r   Nc                 C   s
   || _ d S N)request)selfr    r   r/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/contrib/sitemaps/sitemap_generator.py__init__
   s    zSitemap.__init__c                 C   s   | | jS r   )get_full_urlr   r   objr   r   r   location   s    zSitemap.locationc                 C   s   |j p
|jS r   )Zlast_published_atZlatest_revision_created_atr
   r   r   r   lastmod   s    zSitemap.lastmodc                 C   s8   ddl m} || j}|d kr4|jdjddS |S )Nr   )Site	root_pageT)Zis_default_site)Zwagtail.modelsr   Zfind_for_requestr   objectsZselect_relatedget)r   r   siter   r   r   get_wagtail_site   s
    zSitemap.get_wagtail_sitec                 C   s(   |   jjdd  d  S )NT)Z	inclusivepath)r   r   Zget_descendantslivepublicZorder_byZdefer_streamfieldsZspecific)r   r   r   r   items   s    

zSitemap.itemsc           	      C   sl   g }t  }| j|j D ]4}|| j}|D ]}|| ||	d q0q|rhd |krht
|| _|S )Nr   )setZ	paginatorpageZobject_listiteratorZget_sitemap_urlsr   appendaddr   maxZlatest_lastmod)	r   r   protocoldomainurlsZ	last_modsitemZurl_info_itemsZurl_infor   r   r   _urls(   s    

zSitemap._urls)N)	__name__
__module____qualname__r   r   r   r   r   r"   r   r   r   r   r   	   s   
r   N)Zdjango.contrib.sitemapsr   ZDjangoSitemapr   r   r   r   <module>   s   