source: products/quintagroup.plonegooglesitemaps/trunk/quintagroup/plonegooglesitemaps/browser/configletview.py @ 2516

Last change on this file since 2516 was 2516, checked in by mylan, 14 years ago

#192: Added uploadVerificationFile method to configlet view

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