52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
import os, sys, ConfigParser, shutil, glob
|
|
venus_base = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
sys.path.insert(0,venus_base)
|
|
|
|
if __name__ == "__main__":
|
|
|
|
if sys.argv[1] == '-v' or sys.argv[1] == '--verbose':
|
|
import planet
|
|
planet.getLogger('DEBUG',None)
|
|
del sys.argv[1]
|
|
|
|
from planet import config
|
|
config.parser = ConfigParser.ConfigParser()
|
|
config.parser.add_section('Planet')
|
|
config.parser.add_section(sys.argv[1])
|
|
work = reduce(os.path.join, ['tests','work','reconsititute'], venus_base)
|
|
output = os.path.join(work, 'output')
|
|
config.parser.set('Planet','cache_directory',work)
|
|
config.parser.set('Planet','output_dir',output)
|
|
config.parser.set('Planet','template_files','themes/common/atom.xml.xslt')
|
|
|
|
for name, value in zip(sys.argv[2::2],sys.argv[3::2]):
|
|
config.parser.set(sys.argv[1], name.lstrip('-'), value)
|
|
|
|
from planet import spider
|
|
spider.spiderPlanet(only_if_new=False)
|
|
|
|
from planet import feedparser
|
|
for source in glob.glob(os.path.join(work, 'sources/*')):
|
|
feed = feedparser.parse(source).feed
|
|
if feed.has_key('title'):
|
|
config.parser.set('Planet','name',feed.title_detail.value)
|
|
if feed.has_key('link'):
|
|
config.parser.set('Planet','link',feed.link)
|
|
if feed.has_key('author_detail'):
|
|
if feed.author_detail.has_key('name'):
|
|
config.parser.set('Planet','owner_name',feed.author_detail.name)
|
|
if feed.author_detail.has_key('email'):
|
|
config.parser.set('Planet','owner_email',feed.author_detail.email)
|
|
|
|
from planet import splice
|
|
doc = splice.splice()
|
|
splice.apply(doc.toxml('utf-8'))
|
|
|
|
atom = open(os.path.join(output,'atom.xml')).read()
|
|
|
|
shutil.rmtree(work)
|
|
os.removedirs(os.path.dirname(work))
|
|
|
|
print atom
|