Prise en charge du mode d'enregistrement anonyme avec mot de passe.
[photoprint.git] / _utils / import_printing_list.py
1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3 #######################################################################################
4 # Copyright © 2009 Benoît Pin <pin@cri.ensmp.fr> #
5 # Plinn - http://plinn.org #
6 # #
7 # #
8 # This program is free software; you can redistribute it and/or #
9 # modify it under the terms of the GNU General Public License #
10 # as published by the Free Software Foundation; either version 2 #
11 # of the License, or (at your option) any later version. #
12 # #
13 # This program is distributed in the hope that it will be useful, #
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of #
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
16 # GNU General Public License for more details. #
17 # #
18 # You should have received a copy of the GNU General Public License #
19 # along with this program; if not, write to the Free Software #
20 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. #
21 #######################################################################################
22 """
23 Downloads RSS based order description and make a local human readable file tree
24 to facilitate printing tasks.
25
26 """
27
28
29 from urllib2 import HTTPBasicAuthHandler
30 from urllib2 import build_opener
31 from urllib2 import urlopen
32 from xml.dom.minidom import parseString
33 from xml.dom import Node
34 from os import mkdir, chdir
35 from os.path import abspath, join, expanduser, exists
36 from getpass import getpass
37
38 ELEMENT_NODE = Node.ELEMENT_NODE
39
40 def getHttpOpener(url, login, password) :
41 auth_handler = HTTPBasicAuthHandler()
42 host = '/'.join(url.split('/', 3)[:3])
43 auth_handler.add_password('Zope', host, login, password)
44 opener = build_opener(auth_handler)
45 return opener
46
47 def getXml(url, opener) :
48 url = '%s?disable_cookie_login__=1' % url
49 xml = opener.open(url).read()
50 return xml
51
52 def genFileTree(url, login, password, dest) :
53 opener = getHttpOpener(url, login, password)
54 xml = getXml(url, opener)
55 d = parseString(xml)
56 doc = d.documentElement
57
58 channel = doc.getElementsByTagName('channel')[0]
59 orderName = getContentOf(channel, 'title')
60
61 chdir(dest)
62 mkdir(orderName)
63
64 for item in iterElementChildsByTagName(d.documentElement, 'item') :
65 ppTitle = getContentOf(item, 'pp:title')
66 ppQuantity = getContentOf(item, 'pp:quantity')
67
68 printTypePath = join(orderName, ppTitle)
69 printQuantityPath = join(orderName, ppTitle, ppQuantity)
70
71 if not exists(printTypePath) :
72 mkdir(printTypePath)
73 infoFile = open(join(printTypePath, 'info.txt'), 'w')
74 infoFile.write(getContentOf(item, 'pp:title'))
75 infoFile.write('\n\n')
76 infoFile.write(getContentOf(item, 'pp:description'))
77 infoFile.close()
78
79 if not exists(printQuantityPath) :
80 mkdir(printQuantityPath)
81
82 hdUrl = '%s?disable_cookie_login__=1' % getContentOf(item, 'link')
83 localFileName = getContentOf(item, 'title')
84 print localFileName
85 localFile = open(join(printQuantityPath, localFileName), 'w')
86 localFile.write(opener.open(hdUrl).read())
87 localFile.close()
88
89 def iterElementChildsByTagName(parent, tagName) :
90 child = parent.firstChild
91 while child :
92 if child.nodeType == ELEMENT_NODE and child.tagName == tagName :
93 yield child
94 child = child.nextSibling
95
96 def getContentOf(parent, tagName) :
97 child = parent.firstChild
98 while child :
99 if child.nodeType == ELEMENT_NODE and child.tagName == tagName :
100 return child.firstChild.nodeValue.encode('utf-8')
101 child = child.nextSibling
102
103 raise ValueError("%r tag not found" % tagName)
104
105
106 def main() :
107 url = raw_input('url flux xml de la commande : ')
108 login = raw_input('login : ')
109 password = getpass('mot de passe : ')
110 dest = raw_input('cible [~/Desktop]')
111 if not dest :
112 dest = expanduser('~/Desktop')
113
114 genFileTree(url, login, password, dest)
115
116 if __name__ == '__main__' :
117 main()
118
119