#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# feed.py generate rss feed
#
# author: observer
# email: jingchaohu@gmail.com
# blog: http://obmem.com
# last edit @ 2009.12.22
import sqlite3
import time
import os,sys

def feed(path=None,conn=None,cat1='',cat2=''):

    if (not path) or (not conn):
        path = "/var/www/simplecd.old"
        conn = sqlite3.connect(path+'/verycd.sqlite3.db')
        conn.text_factory = str
    c=conn.cursor()

    for _ in range(3):
        try: 
            if not cat1:
                c.execute('select * from verycd order by updtime desc limit 20');
            elif not cat2:
                c.execute('select * from verycd where category1=? order by updtime desc limit 20',(cat1,));
            else:
                c.execute('select * from verycd where category1=? and category2=? order by updtime desc limit 20',(cat1,cat2));
            break
        except:
            time.sleep(0.1)
            continue

    data = None

    try:
        data = c.fetchall()
    except:
        c.close()
        conn.commit()
        return
    
    c.close()
    conn.commit()

    pubdate = time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.gmtime())

    feed = '''<?xml version="1.0" encoding="utf-8"?>
<rss version="2.0" xmlns:atom="http://www.w3.org/2005/Atom" xmlns:dc="http://purl.org/dc/elements/1.1/">
<channel>
<title> SimpleCD - 最新电驴%s%s资源 </title>
<atom:link href="http://www.simplecd.org/feed" rel="self" type="application/rss+xml" />
<link>http://www.simplecd.org</link>
<description><![CDATA[ SimpleCD - 最新电驴%s%s资源 ]]></description>
<language>zh-cn</language>\n''' % (cat1,cat2,cat1,cat2)
    feed += '<pubDate>%s</pubDate>\n' % pubdate
    feed += '<lastBuildDate>%s</lastBuildDate>\n' % pubdate
    feed += '''<docs>http://blogs.law.harvard.edu/tech/rss</docs>
<generator>SimpleCD.com</generator>
<managingEditor>jingchaohu@gmail.com (webmaster)</managingEditor>
<webMaster>jingchaohu@gmail.com (webmaster)</webMaster>
<ttl>4</ttl>
'''

    for d in data:
        # data:0  1   2      3     4       5       6    7    8    9
        #      id ttl status brief pubtime pudtime cat1 cat2 ed2k content
        title = d[1]
        link = 'http://www.simplecd.org/id/%s' % d[0]
        rss = '摘要信息：'+d[3]+'<br>\n类别：'+d[6]+'<br>\n子类别：'+d[7]+'<br>\n'
        rss += d[9] 
        feed +='''
    <item>    
        <title><![CDATA[%s]]></title>
        <link>%s</link>
        <description><![CDATA[
        %s        
        ]]>    </description>
        <pubDate>%s</pubDate>
        <dc:creator>observer</dc:creator>
    </item>
'''% (title,link,rss,pubdate)

    feed +='''</channel>
</rss>'''
    return feed

if __name__ == '__main__':
    path = os.path.dirname(os.path.realpath(sys.argv[0]))
    conn = sqlite3.connect(path+'/verycd.sqlite3.db')
    conn.text_factory = str
    print feed(path,conn,cat1="音乐")