winsparkle.py 2.88 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
"""
 *  Copyright (C) 2016 Savoir-faire Linux Inc.
 *
 *  Author: Olivier Soldano <olivier.soldano@savoirfairelinux.com>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.
"""

import sys
import xml.etree.ElementTree as ET
from xml.dom import minidom
import datetime
from email.utils import formatdate


def sameDate(timestamp, pub):
    date1 = timestamp.split()
    date2 = pub.text.split()
    return date1[:4] == date2[:4]

def insertNewPackage(parent_element, title, attrib):
    now = datetime.datetime.now()

    new_item = ET.Element("item")

    titre = ET.SubElement(new_item,"titre")
    titre.text = title + now.strftime("%Y/%m/%d %H:%M")

    pubDate = ET.SubElement(new_item, "pubDate")
    pubDate.text = formatdate()

    enclosure = ET.SubElement(new_item, "enclosure", attrib=attrib)

    parent_element.insert(4,new_item)


if __name__ == "__main__":
    now = datetime.datetime.now()
    now_timestamp = formatdate() # rfc 2822
    sparkle_file = sys.argv[1]
    title = sys.argv[2]
    url = sys.argv[3]
    os = sys.argv[4]
    length = sys.argv[5]
    ET.register_namespace('sparkle','http://www.andymatuschak.org/xml-namespaces/sparkle')
    namespace = {'sparkle' : 'http://www.andymatuschak.org/xml-namespaces/sparkle'}
    tree = ET.parse(sparkle_file)
    channel = tree.find("channel")
    attrib = {'url' : url,
62 63
              'sparkle:version' : now.strftime("%Y%m%d"),
              'sparkle:shortVersionString' : "nightly-" + now.strftime("%Y%m%d"),
64 65 66 67 68
              'sparkle:os' : os,
              'length' : length,
              'type' : "application/octet-stream"
    }

69
    # remove all publications of the same day (but not same os)
70
    for item in tree.findall(".//item"):
71
        if sameDate(now_timestamp, item.find("pubDate")) and not\
72 73 74 75 76 77 78 79 80 81 82
        item.find("./enclosure[@sparkle:os='%s']" % os, namespace) is None:
            channel.remove(item)

    insertNewPackage(channel, title, attrib)

    # Pretty printing with xml dom
    str_tree = ET.tostring(tree.getroot(),encoding='utf-8').decode('utf-8').replace('\n','').replace('\r','')
    reparsed_doc = minidom.parseString(str_tree)
    xml_out = open(sparkle_file,"wb")
    xml_out.write(reparsed_doc.toprettyxml(indent='  ', newl='\n',encoding="utf-8"))
    xml_out.close()