1 | from zope.component import queryMultiAdapter |
---|
2 | from zope.interface import implements, Interface, Attribute |
---|
3 | |
---|
4 | from OFS.Image import cookId |
---|
5 | from OFS.ObjectManager import BadRequestException |
---|
6 | from Products.Five import BrowserView |
---|
7 | |
---|
8 | import urlparse |
---|
9 | |
---|
10 | |
---|
11 | def splitNum(num): |
---|
12 | res = [] |
---|
13 | prefn = 3 |
---|
14 | for c in str(num)[::-1]: |
---|
15 | res.insert(0, c) |
---|
16 | if not len(res) % prefn: |
---|
17 | res.insert(0, ',') |
---|
18 | prefn += 4 |
---|
19 | return "".join(res[0] == ',' and res[1:] or res) |
---|
20 | |
---|
21 | |
---|
22 | class IConfigletSettingsView(Interface): |
---|
23 | """ |
---|
24 | Sitemap view interface |
---|
25 | """ |
---|
26 | |
---|
27 | sitemaps = Attribute("Returns mapping of sitemap's type to list of " |
---|
28 | "appropriate objects") |
---|
29 | hasContentSM = Attribute("Returns boolean about existance of content " |
---|
30 | "sitemap") |
---|
31 | hasMobileSM = Attribute("Returns boolean about existance of mobile " |
---|
32 | "sitemap") |
---|
33 | hasNewsSM = Attribute("Returns boolean about existance of news sitemap") |
---|
34 | sm_types = Attribute("List of sitemap types") |
---|
35 | |
---|
36 | def sitemapsDict(): |
---|
37 | """ Return dictionary like object with data for table |
---|
38 | """ |
---|
39 | def sitemapsURLByType(): |
---|
40 | """ Return dictionary like object with sitemap_type as key |
---|
41 | and sitemap object(s) as value |
---|
42 | """ |
---|
43 | def getVerificationFiles(): |
---|
44 | """ Return list of existent verification files on site. |
---|
45 | Update googlesitemap_properties.verification_file |
---|
46 | property for only existent files |
---|
47 | """ |
---|
48 | |
---|
49 | def uploadVerificationFile(vfile): |
---|
50 | """ Upload passed site verification file to the site. |
---|
51 | On success - update googlesitemaps verification files list. |
---|
52 | Return tuple where : |
---|
53 | 1. boolean value - is verification file successfully created. |
---|
54 | 2. string value: |
---|
55 | 2.1. if successfull - id of created verification file |
---|
56 | 2.2. if failure - error descirption |
---|
57 | """ |
---|
58 | |
---|
59 | |
---|
60 | class ConfigletSettingsView(BrowserView): |
---|
61 | """ |
---|
62 | Configlet settings browser view |
---|
63 | """ |
---|
64 | implements(IConfigletSettingsView) |
---|
65 | sitemaps = [] |
---|
66 | |
---|
67 | def __init__(self, context, request): |
---|
68 | self.context = context |
---|
69 | self.request = request |
---|
70 | |
---|
71 | self.tools = queryMultiAdapter((self.context, self.request), |
---|
72 | name="plone_tools") |
---|
73 | self.pps = queryMultiAdapter((self.context, self.request), |
---|
74 | name="plone_portal_state") |
---|
75 | self.sitemaps = [i.getObject() for i in |
---|
76 | self.tools.catalog()(portal_type='Sitemap')] |
---|
77 | |
---|
78 | @property |
---|
79 | def sm_types(self): |
---|
80 | return [i.getSitemapType() for i in self.sitemaps] |
---|
81 | |
---|
82 | @property |
---|
83 | def hasContentSM(self): |
---|
84 | return 'content' in self.sm_types |
---|
85 | |
---|
86 | @property |
---|
87 | def hasMobileSM(self): |
---|
88 | return 'mobile' in self.sm_types |
---|
89 | |
---|
90 | @property |
---|
91 | def hasNewsSM(self): |
---|
92 | return 'news' in self.sm_types |
---|
93 | |
---|
94 | def sitemapsURLByType(self): |
---|
95 | sitemaps = {} |
---|
96 | for sm in self.sitemaps: |
---|
97 | smlist = sitemaps.setdefault(sm.getSitemapType(), []) |
---|
98 | smlist.append({'url': sm.absolute_url(), 'id': sm.id}) |
---|
99 | |
---|
100 | sitemaps['all'] = sitemaps.setdefault('content', []) + \ |
---|
101 | sitemaps.setdefault('mobile', []) + sitemaps.setdefault('news', []) |
---|
102 | return sitemaps |
---|
103 | |
---|
104 | def sitemapsURLs(self): |
---|
105 | sitemaps = {} |
---|
106 | for sm in self.sitemaps: |
---|
107 | smlist = sitemaps.setdefault(sm.getSitemapType(), []) |
---|
108 | smlist.append(sm.absolute_url()) |
---|
109 | return sitemaps |
---|
110 | |
---|
111 | def sitemapsDict(self): |
---|
112 | content, mobile, news = [], [], [] |
---|
113 | for sm in self.sitemaps: |
---|
114 | data = self.getSMData(sm) |
---|
115 | if data['sm_type'] == 'Content': |
---|
116 | content.append(data) |
---|
117 | elif data['sm_type'] == 'Mobile': |
---|
118 | mobile.append(data) |
---|
119 | elif data['sm_type'] == 'News': |
---|
120 | news.append(data) |
---|
121 | return content + mobile + news |
---|
122 | |
---|
123 | def getSMData(self, ob): |
---|
124 | size, entries = self.getSitemapData(ob) |
---|
125 | return {'sm_type': ob.getSitemapType().capitalize(), |
---|
126 | 'sm_id': ob.id, |
---|
127 | 'sm_url': ob.absolute_url(), |
---|
128 | 'sm_size': size and splitNum(size) or '', |
---|
129 | 'sm_entries': entries and splitNum(entries) or '', |
---|
130 | } |
---|
131 | |
---|
132 | def getSitemapData(self, ob): |
---|
133 | size, entries = (0, 0) |
---|
134 | view = ob and ob.defaultView() or None |
---|
135 | if view: |
---|
136 | self.request.RESPONSE |
---|
137 | bview = queryMultiAdapter((ob, self.request), name=view) |
---|
138 | if bview: |
---|
139 | try: |
---|
140 | size = len(bview()) |
---|
141 | entries = bview.numEntries |
---|
142 | self.request.RESPONSE.setHeader('Content-Type', |
---|
143 | 'text/html') |
---|
144 | except: |
---|
145 | pass |
---|
146 | return (size, entries) |
---|
147 | |
---|
148 | def deleteGSMVerificationFile(self): |
---|
149 | portal = self.pps.portal() |
---|
150 | portal.manage_delObjects([self.request.id, ]) |
---|
151 | self.request.RESPONSE.redirect( |
---|
152 | urlparse.urljoin(self.context.absolute_url, |
---|
153 | 'prefs_gsm_verification')) |
---|
154 | |
---|
155 | def getVerificationFiles(self): |
---|
156 | vfs = [] |
---|
157 | props = getattr(self.tools.properties(), 'googlesitemap_properties') |
---|
158 | portal = self.pps.portal() |
---|
159 | if props: |
---|
160 | portal_ids = portal.objectIds() |
---|
161 | props_vfs = list(props.getProperty('verification_filenames', [])) |
---|
162 | |
---|
163 | vfs = [vf for vf in props_vfs if vf in portal_ids] |
---|
164 | if not props_vfs == vfs: |
---|
165 | props._updateProperty('verification_filenames', vfs) |
---|
166 | |
---|
167 | return [{'id': x, 'title': portal[x].title} for x in vfs] |
---|
168 | |
---|
169 | def uploadVerificationFile(self, request): |
---|
170 | vfilename = "" |
---|
171 | portal = self.pps.portal() |
---|
172 | try: |
---|
173 | comment = request.get("comment") |
---|
174 | vfile = request.get("verification_file") |
---|
175 | vfilename, vftitle = cookId("", "", vfile) |
---|
176 | portal.manage_addFile(id="", file=vfile, title=comment) |
---|
177 | portal[vfilename].manage_addProperty( |
---|
178 | 'CreatedBy', 'quintagroupt.plonegooglesitemaps', 'string') |
---|
179 | except BadRequestException, e: |
---|
180 | return False, str(e) |
---|
181 | else: |
---|
182 | props = self.tools.properties().googlesitemap_properties |
---|
183 | vfilenames = list(props.getProperty('verification_filenames', [])) |
---|
184 | vfilenames.append(vfilename) |
---|
185 | props.manage_changeProperties(verification_filenames=vfilenames) |
---|
186 | return True, vfilename |
---|