diff options
| -rwxr-xr-x | src/featherweight.py | 73 |
1 files changed, 66 insertions, 7 deletions
diff --git a/src/featherweight.py b/src/featherweight.py index 776fa49..80f0643 100755 --- a/src/featherweight.py +++ b/src/featherweight.py @@ -22,19 +22,76 @@ import xml.parsers.expat import sys -rss = sys.argv[1] -with open(rss, 'r') as file: - rss = file.read() +feed = sys.argv[1] +with open(feed, 'r') as file: + feed = file.read() parser = xml.parsers.expat.ParserCreate() + +is_rss = False +feeds = [] +rss_root = None +item = None +text = None + + def start_element(name, attributes): - print('Start element:', name, attributes) + global is_rss, feeds, rss_root, item, text + name = name.lower() + if rss_root is None: + if name == 'rss': + is_rss = True + elif is_rss: + if name == 'channel': + rss_root = {'items' : []} + return + else: + if item is None: + if name == 'item': + item = {} + return + text = '' + + def end_element(name): - print('End element:', name) + global is_rss, feeds, rss_root, item, text + if rss_root is not None: + if item is not None: + if name == 'item': + rss_root['items'].append(item) + item = None + elif name == 'title': + item['title'] = text + elif name == 'description': + item['description'] = text + elif name == 'link': + item['link'] = text + elif name == 'guid': + item['guid'] = text + elif name == 'pubdate': + item['pubdate'] = text + else: + if name == 'title': + rss_root['title'] = text + elif name == 'description': + rss_root['description'] = text + elif name == 'link': + rss_root['link'] = text + elif name == 'channel': + feeds.append(rss_root) + rss_root = None + elif name == 'rss': + is_rss = False + text = None + + def char_data(data): - print('Character data:', repr(data)) + global text + if text is not None: + text += data + parser.StartElementHandler = start_element @@ -42,5 +99,7 @@ parser.EndElementHandler = end_element parser.CharacterDataHandler = char_data -parser.Parse(rss, True) +parser.Parse(feed, True) + +print(feeds) |