source: products/quintagroup.plonegooglesitemaps/branches/sitemap_date/quintagroup/plonegooglesitemaps/browser/configletview.py @ 3565

Last change on this file since 3565 was 3506, checked in by potar, 12 years ago

fixed pep8

  • Property svn:eol-style set to native
File size: 6.4 KB
Line 
1from zope.component import queryMultiAdapter
2from zope.interface import implements, Interface, Attribute
3
4from OFS.Image import cookId
5from OFS.ObjectManager import BadRequestException
6from Products.Five import BrowserView
7
8import urlparse
9
10
11def 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
22class 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
60class 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
Note: See TracBrowser for help on using the repository browser.