121 lines
4.0 KiB
Python
121 lines
4.0 KiB
Python
"""
|
|
This file is part of the web2py Web Framework
|
|
Copyrighted by Massimo Di Pierro <mdipierro@cs.depaul.edu>
|
|
License: LGPLv3 (http://www.gnu.org/licenses/lgpl.html)
|
|
"""
|
|
import datetime
|
|
import decimal
|
|
from storage import Storage
|
|
from html import TAG, XmlComponent
|
|
from html import xmlescape
|
|
from languages import lazyT
|
|
import contrib.rss2 as rss2
|
|
|
|
try:
|
|
import json as json_parser # try stdlib (Python 2.6)
|
|
except ImportError:
|
|
try:
|
|
import simplejson as json_parser # try external module
|
|
except:
|
|
import contrib.simplejson as json_parser # fallback to pure-Python module
|
|
|
|
def custom_json(o):
|
|
if hasattr(o,'custom_json') and callable(o.custom_json):
|
|
return o.custom_json()
|
|
if isinstance(o, (datetime.date,
|
|
datetime.datetime,
|
|
datetime.time)):
|
|
return o.isoformat()[:19].replace('T',' ')
|
|
elif isinstance(o, (int, long)):
|
|
return int(o)
|
|
elif isinstance(o, decimal.Decimal):
|
|
return str(o)
|
|
elif isinstance(o, lazyT):
|
|
return str(o)
|
|
elif isinstance(o,XmlComponent):
|
|
return str(o)
|
|
elif hasattr(o,'as_list') and callable(o.as_list):
|
|
return o.as_list()
|
|
elif hasattr(o,'as_dict') and callable(o.as_dict):
|
|
return o.as_dict()
|
|
else:
|
|
raise TypeError(repr(o) + " is not JSON serializable")
|
|
|
|
|
|
def xml_rec(value, key, quote=True):
|
|
if hasattr(value,'custom_xml') and callable(value.custom_xml):
|
|
return value.custom_xml()
|
|
elif isinstance(value, (dict, Storage)):
|
|
return TAG[key](*[TAG[k](xml_rec(v, '',quote)) \
|
|
for k, v in value.items()])
|
|
elif isinstance(value, list):
|
|
return TAG[key](*[TAG.item(xml_rec(item, '',quote)) for item in value])
|
|
elif hasattr(value,'as_list') and callable(value.as_list):
|
|
return str(xml_rec(value.as_list(),'',quote))
|
|
elif hasattr(value,'as_dict') and callable(value.as_dict):
|
|
return str(xml_rec(value.as_dict(),'',quote))
|
|
else:
|
|
return xmlescape(value,quote)
|
|
|
|
|
|
def xml(value, encoding='UTF-8', key='document', quote=True):
|
|
return ('<?xml version="1.0" encoding="%s"?>' % encoding) + str(xml_rec(value,key,quote))
|
|
|
|
|
|
def json(value,default=custom_json):
|
|
return json_parser.dumps(value,default=default)
|
|
|
|
|
|
def csv(value):
|
|
return ''
|
|
|
|
def ics(events, title=None, link=None, timeshift=0, **ignored):
|
|
import datetime
|
|
title = title or '(unkown)'
|
|
if link and not callable(link):
|
|
link = lambda item,prefix=link: prefix.replace('[id]',str(item['id']))
|
|
s = 'BEGIN:VCALENDAR'
|
|
s += '\nVERSION:2.0'
|
|
s += '\nX-WR-CALNAME:%s' % title
|
|
s += '\nSUMMARY:%s' % title
|
|
s += '\nPRODID:Generated by web2py'
|
|
s += '\nCALSCALE:GREGORIAN'
|
|
s += '\nMETHOD:PUBLISH'
|
|
for item in events:
|
|
s += '\nBEGIN:VEVENT'
|
|
s += '\nUID:%s' % item['id']
|
|
if link:
|
|
s += '\nURL:%s' % link(item)
|
|
shift = datetime.timedelta(seconds=3600*timeshift)
|
|
start = item['start_datetime']+shift
|
|
stop = item['stop_datetime']+shift
|
|
s += '\nDTSTART:%s' % start.strftime('%Y%m%dT%H%M%S')
|
|
s += '\nDTEND:%s' % stop.strftime('%Y%m%dT%H%M%S')
|
|
s += '\nSUMMARY:%s' % item['title']
|
|
s += '\nEND:VEVENT'
|
|
s += '\nEND:VCALENDAR'
|
|
return s
|
|
|
|
def rss(feed):
|
|
if not 'entries' in feed and 'items' in feed:
|
|
feed['entries'] = feed['items']
|
|
now=datetime.datetime.now()
|
|
rss = rss2.RSS2(title = str(feed.get('title','(notitle)')),
|
|
link = str(feed.get('link',None)),
|
|
description = str(feed.get('description','')),
|
|
lastBuildDate = feed.get('created_on', now),
|
|
items = [rss2.RSSItem(
|
|
title=str(entry.get('title','(notitle)')),
|
|
link=str(entry.get('link',None)),
|
|
description=str(entry.get('description','')),
|
|
pubDate=entry.get('created_on', now)
|
|
) for entry in feed.get('entries',[])])
|
|
return rss2.dumps(rss)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|