[1] | 1 | from urllib2 import urlopen |
---|
| 2 | from urllib import quote as urlquote |
---|
| 3 | from Products.CMFCore.utils import getToolByName |
---|
| 4 | import re |
---|
| 5 | import Products.qPloneGoogleSitemaps.config as config |
---|
| 6 | from OFS.ObjectManager import BadRequestException |
---|
| 7 | def ping_google(url): |
---|
| 8 | """Ping sitemap to Google""" |
---|
| 9 | sitemap_url = urlquote(url + "/google-sitemaps") |
---|
| 10 | g = urlopen('http://www.google.com/webmasters/sitemaps/ping?sitemap='+sitemap_url) |
---|
| 11 | result = g.read() |
---|
| 12 | g.close() |
---|
| 13 | return 0 |
---|
| 14 | |
---|
| 15 | def searchAndReplace(string, what, with): |
---|
| 16 | """Emulate sed command s/""" |
---|
| 17 | res = re.sub(what,with,string) |
---|
| 18 | return res |
---|
| 19 | |
---|
| 20 | OPERATORS = {'s': searchAndReplace,} |
---|
| 21 | |
---|
| 22 | def applyOperations(objects, operations): |
---|
| 23 | """Parse Operations """ |
---|
| 24 | parse = re.compile(r"(.?[^\\])/(.*[^\\]|)/(.*[^\\]|)/") |
---|
| 25 | operations=[parse.match(op).groups() for op in operations] |
---|
| 26 | result={} |
---|
| 27 | for ob in objects: |
---|
| 28 | url = ob.getURL() |
---|
| 29 | for operator, what, with in operations: |
---|
| 30 | url = OPERATORS[operator](url, what, with.replace("\\", "")) |
---|
| 31 | #TODO: Remove or replace following condition |
---|
| 32 | #it is senseless in the case we need intelligent |
---|
| 33 | #result set. Better condition would be to place |
---|
| 34 | #freshest brain into result |
---|
| 35 | if url in result.keys(): |
---|
| 36 | continue |
---|
| 37 | #TODO: replace brain with only data necessary to |
---|
| 38 | #generate sitemap |
---|
| 39 | result[url]=ob |
---|
| 40 | return result |
---|
| 41 | |
---|
| 42 | def additionalURLs(self): |
---|
| 43 | """Add URLs to sitemap that arn't objects""" |
---|
| 44 | res = [] |
---|
| 45 | plone_home = getToolByName(self, 'portal_url').getPortalObject().absolute_url() |
---|
| 46 | root = self.getPhysicalRoot().absolute_url() |
---|
| 47 | |
---|
| 48 | props = getToolByName(self,'portal_properties') |
---|
| 49 | try: |
---|
| 50 | URLs = props.googlesitemap_properties.urls |
---|
| 51 | except AttributeError: |
---|
| 52 | URLs = [] |
---|
| 53 | |
---|
| 54 | add_zope = re.compile('^/') |
---|
| 55 | add_plone= re.compile('^[^http://|https://|\\\]') |
---|
| 56 | for url in URLs: |
---|
| 57 | if add_zope.match(url): |
---|
| 58 | res.append(root+url) |
---|
| 59 | elif add_plone.match(url): |
---|
| 60 | res.append(plone_home+'/'+url) |
---|
| 61 | else: |
---|
| 62 | res.append(url) |
---|
| 63 | return res |
---|
| 64 | |
---|
| 65 | """workflows & co""" |
---|
| 66 | def getWorkflowTransitions(self,workflow_id): |
---|
| 67 | pw = getToolByName(self,'portal_workflow') |
---|
| 68 | wf = pw.getWorkflowById(workflow_id) |
---|
| 69 | if not wf: |
---|
| 70 | return None |
---|
| 71 | return wf.transitions.values() |
---|
| 72 | |
---|
| 73 | def setWorkflowTransitions(self,transitions): |
---|
| 74 | """set workflow transitions properties""" |
---|
| 75 | portal_workflow = getToolByName(self, 'portal_workflow') |
---|
| 76 | transmap = {} |
---|
| 77 | for key in transitions: |
---|
| 78 | if key.find('#')>0: |
---|
| 79 | ids = key.split('#') |
---|
| 80 | wfid = ids[0] |
---|
| 81 | if not wfid in transmap.keys(): |
---|
| 82 | transmap[wfid]=[] |
---|
| 83 | transmap[wfid].append(ids[1]) |
---|
| 84 | for wfid in transmap.keys(): |
---|
| 85 | workflow = portal_workflow.getWorkflowById(wfid) |
---|
| 86 | if config.ping_googlesitemap not in workflow.scripts.objectIds(): |
---|
| 87 | workflow.scripts.manage_addProduct['ExternalMethod'].manage_addExternalMethod( |
---|
| 88 | config.ping_googlesitemap, |
---|
| 89 | 'Ping sitemap', |
---|
| 90 | 'qPloneGoogleSitemaps.ping_googlesitemap', |
---|
| 91 | config.ping_googlesitemap) |
---|
| 92 | transitions_set = transmap[wfid] |
---|
| 93 | for transition in workflow.transitions.values(): |
---|
| 94 | trid = transition.id |
---|
| 95 | tras = transition.after_script_name |
---|
| 96 | if (tras == '') and (trid in transitions_set): |
---|
| 97 | #set |
---|
| 98 | after_script = config.ping_googlesitemap |
---|
| 99 | elif (tras == config.ping_googlesitemap) and not (trid in transitions_set): |
---|
| 100 | #reset |
---|
| 101 | after_script = '' |
---|
| 102 | else: |
---|
| 103 | #avoid properties set |
---|
| 104 | continue |
---|
| 105 | transition.setProperties(title=transition.title, |
---|
| 106 | new_state_id=transition.new_state_id, |
---|
| 107 | after_script_name=after_script, |
---|
| 108 | actbox_name=transition.actbox_name) |
---|