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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#!/usr/bin/python
REMOTE_LIST="http://weechat.org/files/plugins.xml"
MIN_VERSION="0.3.0"
import sys, os.path, urllib2
from xml.etree.cElementTree import fromstring as xmlParser
class WeechatScripts:
def get_xmldom(self):
xmldom = None
try:
fd = urllib2.urlopen(REMOTE_LIST)
xmlstr = fd.read()
except:
xmlstr = None
print "ERR: unable to retrieve '%s'" % (REMOTE_LIST)
if xmlstr:
try:
xmldom = xmlParser(xmlstr)
except:
print "ERR: unable to parse XML"
xmldom = None
return xmldom
def get_datas(self):
datas = {}
xmldom = self.get_xmldom()
if xmldom:
file_len = license_len = date_len = 0
for el in xmldom.findall('plugin'):
if el.find('min_weechat').text < MIN_VERSION:
continue
lang = el.find('language').text
filename = el.find('url').text.split('/')[-1]
file_len = max(len(filename), file_len)
license_len = max(len(el.find('license').text), license_len)
author = "%s <%s>" % (el.find('author').text, el.find('mail').text)
author = author.replace(' [at] ', '@')
author = author.replace(' [dot] ', '.')
date = ''
date_add = int(el.find('added').text.split('-')[0])
if el.find('updated').text == None:
date = "(%d)" % (date_add)
else:
date_upd = int(el.find('updated').text.split('-')[0])
if date_upd > date_add:
date = "(%d-%d)" % (date_add, date_upd)
else:
date = "(%d)" % (date_add)
date_len = max(len(date), date_len)
if not datas.has_key(lang):
datas[lang] = []
datas[lang].append({
'language': el.find('language').text,
'url': el.find('url').text,
'file': filename,
'version': el.find('version').text,
'license': el.find('license').text,
'author': author,
'date': date,
})
return datas, date_len, file_len, license_len
def copyright(self):
scriptlist, date_len, file_len, license_len = self.get_datas()
if scriptlist != {}:
for l in sorted(scriptlist.keys()):
print "%s%s:" % (l[0].upper(), l[1:])
print "=" * len(l)
for script in scriptlist[l]:
format = " Copyright (C) %%-%ds %%-%ds | %%-%ds | %%s" % (date_len, file_len, license_len)
print format % (script['date'], script['file'], script['license'], script['author'].encode('utf-8'))
print
def wget(self, url, path):
ret = False
content = None
try:
fd = urllib2.urlopen(url)
content = fd.read()
except:
print "ERR: unable to retrieve '%s'" % (url)
content = None
if content:
try:
f = open(path, 'w')
f.write(content)
f.close()
ret = True
except:
print "ERR: unable to write file '%s'" % (path)
return ret
def download(self):
scriptlist, _, _, _ = self.get_datas()
if scriptlist != {}:
for l in scriptlist:
for script in scriptlist[l]:
if not os.path.exists(script['language']):
os.mkdir(script['language'])
path = "%s/%s" % (script['language'], script['file'])
print "Saving '%s': " % (path),
if self.wget(script['url'], path):
print "OK"
else:
print "FAIL"
if __name__ == '__main__':
args = ['copyright', 'download']
if len(sys.argv) != 2:
print "%s: %s" % (sys.argv[0], "|".join(args))
elif sys.argv[1] not in args:
print "%s: %s" % (sys.argv[0], "|".join(args))
else:
ws = WeechatScripts()
if sys.argv[1] == 'copyright':
ws.copyright()
elif sys.argv[1] == 'download':
ws.download()
|
